Replace system roles in responses

deepset-ai · Aug 27, 2024 · 7483bc6 · 7483bc6
1 parent 611b05d
commit 7483bc6
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -236,7 +236,7 @@ def _get_response(self, response_body: GenerationResponse) -> List[ChatMessage]:
         for candidate in response_body.candidates:
             for part in candidate.content.parts:
                 if part._raw_part.text != "":
-                    replies.append(ChatMessage.from_system(part.text))
+                    replies.append(ChatMessage.from_assistant(part.text))
                 elif part.function_call is not None:
                     replies.append(
                         ChatMessage(
@@ -264,4 +264,4 @@ def _get_stream_response(
             responses.append(streaming_chunk.content)
 
         combined_response = "".join(responses).lstrip()
-        return [ChatMessage.from_system(content=combined_response)]
+        return [ChatMessage.from_assistant(content=combined_response)]