From 75533e9319f43ff5a217d98f12848e5afd26e1ff Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-14-53.eu-north-1.compute.internal>
Date: Sun, 26 Apr 2026 14:56:52 +0000
Subject: [PATCH] fix(simulation): guarantee SQLite connection close with
 finally block

---
 backend/scripts/run_reddit_simulation.py  | 24 ++++++++++++-----------
 backend/scripts/run_twitter_simulation.py | 22 +++++++++++----------
 2 files changed, 25 insertions(+), 21 deletions(-)

diff --git a/backend/scripts/run_reddit_simulation.py b/backend/scripts/run_reddit_simulation.py
index 14907cbd..c3f2260f 100644
--- a/backend/scripts/run_reddit_simulation.py
+++ b/backend/scripts/run_reddit_simulation.py
@@ -258,9 +258,9 @@ class IPCHandler:
             agent_prompts = {}  # 记录每个agent的prompt
             
             for interview in interviews:
-                agent_id = interview.get("agent_id")
+                agent_id = interview.get("agent_id") or 0
                 prompt = interview.get("prompt", "")
-                
+
                 try:
                     agent = self.agent_graph.get_agent(agent_id)
                     actions[agent] = ManualAction(
@@ -300,20 +300,21 @@ class IPCHandler:
     def _get_interview_result(self, agent_id: int) -> Dict[str, Any]:
         """从数据库获取最新的Interview结果"""
         db_path = os.path.join(self.simulation_dir, "reddit_simulation.db")
-        
+
         result = {
             "agent_id": agent_id,
             "response": None,
             "timestamp": None
         }
-        
+
         if not os.path.exists(db_path):
             return result
-        
+
+        conn = None
         try:
             conn = sqlite3.connect(db_path)
             cursor = conn.cursor()
-            
+
             # 查询最新的Interview记录
             cursor.execute("""
                 SELECT user_id, info, created_at
@@ -322,7 +323,7 @@ class IPCHandler:
                 ORDER BY created_at DESC
                 LIMIT 1
             """, (ActionType.INTERVIEW.value, agent_id))
-            
+
             row = cursor.fetchone()
             if row:
                 user_id, info_json, created_at = row
@@ -332,12 +333,13 @@ class IPCHandler:
                     result["timestamp"] = created_at
                 except json.JSONDecodeError:
                     result["response"] = info_json
-            
-            conn.close()
-            
+
         except Exception as e:
             print(f"  读取Interview结果失败: {e}")
-        
+        finally:
+            if conn:
+                conn.close()
+
         return result
     
     async def process_commands(self) -> bool:
diff --git a/backend/scripts/run_twitter_simulation.py b/backend/scripts/run_twitter_simulation.py
index caab9e9d..655d1677 100644
--- a/backend/scripts/run_twitter_simulation.py
+++ b/backend/scripts/run_twitter_simulation.py
@@ -258,7 +258,7 @@ class IPCHandler:
             agent_prompts = {}  # 记录每个agent的prompt
             
             for interview in interviews:
-                agent_id = interview.get("agent_id")
+                agent_id = interview.get("agent_id") or 0
                 prompt = interview.get("prompt", "")
                 
                 try:
@@ -300,20 +300,21 @@ class IPCHandler:
     def _get_interview_result(self, agent_id: int) -> Dict[str, Any]:
         """从数据库获取最新的Interview结果"""
         db_path = os.path.join(self.simulation_dir, "twitter_simulation.db")
-        
+
         result = {
             "agent_id": agent_id,
             "response": None,
             "timestamp": None
         }
-        
+
         if not os.path.exists(db_path):
             return result
-        
+
+        conn = None
         try:
             conn = sqlite3.connect(db_path)
             cursor = conn.cursor()
-            
+
             # 查询最新的Interview记录
             cursor.execute("""
                 SELECT user_id, info, created_at
@@ -322,7 +323,7 @@ class IPCHandler:
                 ORDER BY created_at DESC
                 LIMIT 1
             """, (ActionType.INTERVIEW.value, agent_id))
-            
+
             row = cursor.fetchone()
             if row:
                 user_id, info_json, created_at = row
@@ -332,12 +333,13 @@ class IPCHandler:
                     result["timestamp"] = created_at
                 except json.JSONDecodeError:
                     result["response"] = info_json
-            
-            conn.close()
-            
+
         except Exception as e:
             print(f"  读取Interview结果失败: {e}")
-        
+        finally:
+            if conn:
+                conn.close()
+
         return result
     
     async def process_commands(self) -> bool: