RomanNum3ral
/
MicroFish
mirror of https://github.com/666ghj/MiroFish.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(simulation): guarantee SQLite connection close with finally block

This commit is contained in:
Ubuntu 2026-04-26 14:56:52 +00:00
parent 117eabf607
commit 75533e9319
2 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
backend/scripts/run_reddit_simulation.py
View File

@ -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:

22
backend/scripts/run_twitter_simulation.py
View File

@ -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: