From f685741481c129455b42dd67418c1f684597e71e Mon Sep 17 00:00:00 2001
From: alireza78a <alireza78.crypto@gmail.com>
Date: Wed, 11 Mar 2026 18:13:52 +0330
Subject: [PATCH] fix(agent): use atomic write in _save_session_log to prevent
 data loss

---
 run_agent.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/run_agent.py b/run_agent.py
index e91376b03..4bdbf7c31 100644
--- a/run_agent.py
+++ b/run_agent.py
@@ -28,6 +28,7 @@ import json
 import logging
 logger = logging.getLogger(__name__)
 import os
+import tempfile
 import random
 import re
 import sys
@@ -1398,8 +1399,17 @@ class AIAgent:
                 "messages": cleaned,
             }
 
-            with open(self.session_log_file, "w", encoding="utf-8") as f:
-                json.dump(entry, f, indent=2, ensure_ascii=False, default=str)
+            tmp_dir = str(self.session_log_file.parent) if hasattr(self.session_log_file, 'parent') else os.path.dirname(str(self.session_log_file))
+            fd, tmp_path = tempfile.mkstemp(dir=tmp_dir, suffix='.tmp', prefix='.session_')
+            try:
+                with os.fdopen(fd, 'w', encoding='utf-8') as f:
+                    json.dump(entry, f, indent=2, ensure_ascii=False, default=str)
+                    f.flush()
+                    os.fsync(f.fileno())
+                os.replace(tmp_path, str(self.session_log_file))
+            except:
+                os.unlink(tmp_path)
+                raise
 
         except Exception as e:
             if self.verbose_logging: