Bläddra i källkod

Bugfix: Vertex AI vision model not support image (#4853)

Pan, Wen-Ming 10 månader sedan
förälder
incheckning
cdbc260571
1 ändrade filer med 3 tillägg och 5 borttagningar
  1. 3 5
      api/core/model_runtime/model_providers/vertex_ai/llm/llm.py

+ 3 - 5
api/core/model_runtime/model_providers/vertex_ai/llm/llm.py

@@ -367,10 +367,8 @@ class VertexAiLargeLanguageModel(LargeLanguageModel):
                     else:
                         metadata, data = c.data.split(',', 1)
                         mime_type = metadata.split(';', 1)[0].split(':')[1]
-                        blob = {"inline_data":{"mime_type":mime_type,"data":data}}
-                        parts.append(blob)
-                
-                glm_content = glm.Content(role="user", parts=[parts])
+                        parts.append(glm.Part.from_data(mime_type=mime_type, data=data))
+                glm_content = glm.Content(role="user", parts=parts)
             return glm_content
         elif isinstance(message, AssistantPromptMessage):
             if message.content:
@@ -438,4 +436,4 @@ class VertexAiLargeLanguageModel(LargeLanguageModel):
                 exceptions.RequestRangeNotSatisfiable,
                 exceptions.Cancelled,
             ]
-        }
+        }