From 535aab75111b16db17758f1a062e9d563bcc4152 Mon Sep 17 00:00:00 2001 From: anakin87 Date: Wed, 18 Dec 2024 17:34:58 +0100 Subject: [PATCH] fmt --- .../components/generators/google_vertex/chat/gemini.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py b/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py index b9a1678bf..845e24f5f 100644 --- a/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py +++ b/integrations/google_vertex/src/haystack_integrations/components/generators/google_vertex/chat/gemini.py @@ -283,9 +283,7 @@ def _get_response(self, response_body: GenerationResponse) -> List[ChatMessage]: 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( - json.dumps(dict(part.function_call.args)), meta=metadata - ) + new_message = ChatMessage.from_assistant(json.dumps(dict(part.function_call.args)), meta=metadata) new_message.name = part.function_call.name replies.append(new_message) return replies