Fix signature for update_system_prompt

Capsize-Games · Feb 23, 2025 · daa4663 · daa4663
1 parent 087afff
commit daa4663
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/airunner/handlers/llm/agent/mistral_agent.py b/src/airunner/handlers/llm/agent/mistral_agent.py
@@ -714,7 +714,7 @@ def on_delete_messages_after_id(self, data: Dict):
             if self._chat_engine:
                 self._chat_engine.memory = self._chat_memory
 
-    def _update_system_prompt(self) -> Dict:
+    def _update_system_prompt(self):
         self.chat_engine_tool.update_system_prompt(self._system_prompt)
         self.rag_engine_tool.update_system_prompt(self._rag_system_prompt)