Fix groq message handling on oracle side

galadriel-ai · Sep 5, 2024 · 26736d4 · 26736d4
1 parent e48c40d
commit 26736d4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/oracles/src/domain/llm/groq_llm.py b/oracles/src/domain/llm/groq_llm.py
@@ -20,8 +20,10 @@ async def execute(chat: Chat) -> Optional[GroqChatCompletion]:
         timeout=TIMEOUT,
     )
     for message in chat.messages:
-        if len(message.get("content")) and message.get("content")[0].get("text"):
-            message["content"] = message.get("content")[0].get("text")
+        if len(message.get("content")) and message.get("content"):
+            content = message.get("content")
+            if type(content) is not str:
+                message["content"] = message.get("content")[0].get("text")
     chat_completion: ChatCompletion = await client.chat.completions.create(
         messages=chat.messages,
         model=chat.config.model,