From 7483bc65577455b6929713ba39ee680604d9e304 Mon Sep 17 00:00:00 2001 From: Amna Mubashar Date: Tue, 27 Aug 2024 15:57:42 +0200 Subject: [PATCH] Replace system roles in responses --- .../components/generators/google_vertex/chat/gemini.py | 4 ++-- 1 file changed, 2 insertions(+), 2 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 21fa1f52f..d32703a35 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 @@ -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)]