Fix for metadata

deepset-ai · Sep 23, 2024 · b755b31 · b755b31
1 parent 4ee566d
commit b755b31
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -231,9 +231,9 @@ def _get_response(self, response_body: GenerationResponse) -> List[ChatMessage]:
         """
         replies: List[ChatMessage] = []
         for candidate in response_body.candidates:
+            metadata = candidate.to_dict()
             for part in candidate.content.parts:
                 # Remove content from metadata
-                metadata = part.to_dict()
                 metadata.pop("content", None)
                 if part._raw_part.text != "":
                     replies.append(