Fixed gemini role issue

Giskard-AI · Aug 29, 2024 · 4206755 · 4206755
1 parent 5d3730d
commit 4206755
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/giskard/llm/client/gemini.py b/giskard/llm/client/gemini.py
@@ -45,7 +45,7 @@ def _format(messages: Sequence[ChatMessage]) -> Sequence[ContentDict]:
             last_message["parts"].append(message.content)
             continue
 
-        content.append(ContentDict(role=message.role, parts=[message.content]))
+        content.append(ContentDict(role=role, parts=[message.content]))
 
     return content