fmt

deepset-ai · Oct 15, 2024 · 0e93826 · 0e93826
1 parent 0fd8545
commit 0e93826
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/...ations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py b/...ations/google_ai/src/haystack_integrations/components/generators/google_ai/chat/gemini.py
@@ -348,14 +348,12 @@ def _get_stream_response(
         for chunk in stream:
             content: Union[str, Dict[str, Any]] = ""
             dict_chunk = chunk.to_dict()
-            metadata = dict(dict_chunk) # we copy and store the whole chunk as metadata in streaming calls
+            metadata = dict(dict_chunk)  # we copy and store the whole chunk as metadata in streaming calls
             for candidate in dict_chunk["candidates"]:
                 for part in candidate["content"]["parts"]:
                     if "text" in part and part["text"] != "":
                         content = part["text"]
-                        replies.append(
-                            ChatMessage(content=content, role=ChatRole.ASSISTANT, meta=metadata, name=None)
-                        )
+                        replies.append(ChatMessage(content=content, role=ChatRole.ASSISTANT, meta=metadata, name=None))
                     elif "function_call" in part and len(part["function_call"]) > 0:
                         metadata["function_call"] = part["function_call"]
                         content = part["function_call"]["args"]