Skip to content

Commit 07ca7a6

Browse files
authored
fix: make Google Vertex Chat Generator compatible with new ChatMessage (#1255)
* make Vertex compatible with new ChatMessage * fmt
1 parent 4f067b9 commit 07ca7a6

File tree

1 file changed

+2
-4
lines changed
  • integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat

1 file changed

+2
-4
lines changed

integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -280,12 +280,10 @@ def _get_response(self, response_body: GenerationResponse) -> List[ChatMessage]:
280280
# Remove content from metadata
281281
metadata.pop("content", None)
282282
if part._raw_part.text != "":
283-
replies.append(ChatMessage.from_assistant(content=part._raw_part.text, meta=metadata))
283+
replies.append(ChatMessage.from_assistant(part._raw_part.text, meta=metadata))
284284
elif part.function_call:
285285
metadata["function_call"] = part.function_call
286-
new_message = ChatMessage.from_assistant(
287-
content=json.dumps(dict(part.function_call.args)), meta=metadata
288-
)
286+
new_message = ChatMessage.from_assistant(json.dumps(dict(part.function_call.args)), meta=metadata)
289287
new_message.name = part.function_call.name
290288
replies.append(new_message)
291289
return replies

0 commit comments

Comments
 (0)