make Vertex compatible with new ChatMessage

deepset-ai · Dec 18, 2024 · 50e90a6 · 50e90a6
1 parent 7f62ca8
commit 50e90a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...oogle_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py b/...oogle_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py
@@ -280,11 +280,11 @@ def _get_response(self, response_body: GenerationResponse) -> List[ChatMessage]:
                 # Remove content from metadata
                 metadata.pop("content", None)
                 if part._raw_part.text != "":
-                    replies.append(ChatMessage.from_assistant(content=part._raw_part.text, meta=metadata))
+                    replies.append(ChatMessage.from_assistant(part._raw_part.text, meta=metadata))
                 elif part.function_call:
                     metadata["function_call"] = part.function_call
                     new_message = ChatMessage.from_assistant(
-                        content=json.dumps(dict(part.function_call.args)), meta=metadata
+                        json.dumps(dict(part.function_call.args)), meta=metadata
                     )
                     new_message.name = part.function_call.name
                     replies.append(new_message)