Merge pull request #130 from galadriel-ai/fix-groq

Fix groq message handling on oracle side
galadriel-ai · Sep 5, 2024 · fc6270b · fc6270b
2 parents e48c40d + 26736d4
commit fc6270b
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,