From 88a10e723428de99047d4e271714c3e8b6cc6f43 Mon Sep 17 00:00:00 2001
From: Michael Panchenko <michael.panchenko@oraios-ai.de>
Date: Wed, 4 Jun 2025 18:46:14 +0200
Subject: [PATCH] Agent: explicitly save cache on shutdown and in each tool
 call

---
 src/serena/agent.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/serena/agent.py b/src/serena/agent.py
index 4240a342..cfececa1 100644
--- a/src/serena/agent.py
+++ b/src/serena/agent.py
@@ -835,6 +835,7 @@ class SerenaAgent:
         if self.is_language_server_running():
             log.info("Stopping the language server ...")
             assert self.language_server is not None
+            self.language_server.save_cache()
             self.language_server.stop()
         if self._gui_log_handler:
             log.info("Stopping the GUI log window ...")
@@ -1008,6 +1009,11 @@ class Tool(Component):
         if log_call:
             log.info(f"Result: {result}")
 
+        try:
+            self.language_server.save_cache()
+        except Exception as e:
+            log.error(f"Error saving language server cache: {e}")
+
         return result