keep metadata var

deepset-ai · Oct 15, 2024 · 0fd8545 · 0fd8545
1 parent 0582d32
commit 0fd8545
Showing 1 changed file with 4 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
@@ -347,16 +347,16 @@ def _get_stream_response(
         replies: List[ChatMessage] = []
         for chunk in stream:
             content: Union[str, Dict[str, Any]] = ""
-            dict_chunk = chunk.to_dict()  # we store whole chunk as metadata in streaming calls
+            dict_chunk = chunk.to_dict()
+            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=dict_chunk, name=None)
+                            ChatMessage(content=content, role=ChatRole.ASSISTANT, meta=metadata, name=None)
                         )
                     elif "function_call" in part and len(part["function_call"]) > 0:
-                        metadata = dict(dict_chunk)
                         metadata["function_call"] = part["function_call"]
                         content = part["function_call"]["args"]
                         replies.append(
@@ -368,5 +368,5 @@ def _get_stream_response(
                             )
                         )
 
-                    streaming_callback(StreamingChunk(content=content, meta=dict_chunk))
+                    streaming_callback(StreamingChunk(content=content, meta=metadata))
         return replies