From daa4663d5eb146f8f5de9de738e9391021fd9535 Mon Sep 17 00:00:00 2001
From: w4ffl35 <25737761+w4ffl35@users.noreply.github.com>
Date: Sun, 23 Feb 2025 06:51:23 -0700
Subject: [PATCH] Fix signature for update_system_prompt

---
 src/airunner/handlers/llm/agent/mistral_agent.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/airunner/handlers/llm/agent/mistral_agent.py b/src/airunner/handlers/llm/agent/mistral_agent.py
index f8cc25de3..23639693d 100644
--- 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)