small improvemtn

deepset-ai · Dec 18, 2024 · 55fcdda · 55fcdda
1 parent 01fcb80
commit 55fcdda
Showing 1 changed file with 6 additions and 5 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
@@ -280,13 +280,14 @@ def _message_to_content(self, message: ChatMessage) -> Content:
         else:
             msg = f"Unsupported message role {message.role}"
             raise ValueError(msg)
-        role = (
-            "user"
-            if message.is_from(ChatRole.USER)
+
+        role = "model"
+        if (
+            message.is_from(ChatRole.USER)
             or message.is_from(ChatRole.FUNCTION)
             or ("TOOL" in ChatRole._member_names_ and message.is_from(ChatRole.TOOL))
-            else "model"
-        )
+        ):
+            role = "user"
         return Content(parts=[part], role=role)
 
     @component.output_types(replies=List[ChatMessage])