From 6587479a5c8de2013168681516fe868b00652240 Mon Sep 17 00:00:00 2001
From: Artem-Dneprovskii <dneprartemeg@gmail.com>
Date: Sun, 28 Jun 2026 18:19:54 +0200
Subject: [PATCH] feat(interpreter): clean up leftover agent file in
 cleanup_session

---
 treesearch/interpreter.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/treesearch/interpreter.py b/treesearch/interpreter.py
index 979f238..d69cfe6 100755
--- a/treesearch/interpreter.py
+++ b/treesearch/interpreter.py
@@ -69,8 +69,12 @@ def __init__(
         self.agent_file_path = self.working_dir / agent_file_name
 
     def cleanup_session(self):
-        # TODO: Do some cleanup here if necessary
-        pass
+        try:
+            if self.agent_file_path.exists():
+                self.agent_file_path.unlink()
+                logger.debug(f"Removed leftover agent file: {self.agent_file_path}")
+        except OSError as e:
+            logger.warning(f"Failed to remove agent file {self.agent_file_path}: {e}")
 
     def run(
         self,