zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sqlite dir check exists

Browse Source
This commit is contained in:
朱潮 2025-12-18 11:32:43 +08:00
parent 881845f284
commit 95786d559e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/deep_assistant.py
View File

@ -17,6 +17,7 @@ from agent.agent_memory_cache import get_memory_cache_manager
from .checkpoint_utils import prepare_checkpoint_message from .checkpoint_utils import prepare_checkpoint_message
import aiosqlite import aiosqlite
from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
import os
# 全局 MemorySaver 实例 # 全局 MemorySaver 实例
# from langgraph.checkpoint.memory import MemorySaver # from langgraph.checkpoint.memory import MemorySaver
@ -75,6 +76,7 @@ async def init_agent(config: AgentConfig):
# 初始化 checkpointer 和中间件 # 初始化 checkpointer 和中间件
checkpointer = None checkpointer = None
if config.session_id: if config.session_id:
os.makedirs("projects/memory", exist_ok=True)
conn = aiosqlite.connect("projects/memory/checkpoints.db") conn = aiosqlite.connect("projects/memory/checkpoints.db")
checkpointer = AsyncSqliteSaver(conn=conn) checkpointer = AsyncSqliteSaver(conn=conn)
await prepare_checkpoint_message(config, checkpointer) await prepare_checkpoint_message(config, checkpointer)