Stop passing list of dict, expects list of Messages

Capsize-Games · Feb 23, 2025 · 087afff · 087afff
1 parent 826b21e
commit 087afff
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/airunner/handlers/llm/agent/mistral_agent.py b/src/airunner/handlers/llm/agent/mistral_agent.py
@@ -701,9 +701,7 @@ def clear_history(self, data: Optional[Dict] = None):
             self._chat_memory.chat_store_key = self.conversation.key
             messages = self.chat_store.get_messages(self.conversation.key)
             if messages:
-                self._chat_memory.set([
-                    message.model_dump() for message in messages
-                ])
+                self._chat_memory.set(messages)
             if self._chat_engine:
                 self._chat_engine.memory = self._chat_memory
         self.reload_rag_engine()