sqlite dir check exists
This commit is contained in:
parent
881845f284
commit
95786d559e
@ -17,6 +17,7 @@ from agent.agent_memory_cache import get_memory_cache_manager
|
||||
from .checkpoint_utils import prepare_checkpoint_message
|
||||
import aiosqlite
|
||||
from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
|
||||
import os
|
||||
|
||||
# 全局 MemorySaver 实例
|
||||
# from langgraph.checkpoint.memory import MemorySaver
|
||||
@ -75,6 +76,7 @@ async def init_agent(config: AgentConfig):
|
||||
# 初始化 checkpointer 和中间件
|
||||
checkpointer = None
|
||||
if config.session_id:
|
||||
os.makedirs("projects/memory", exist_ok=True)
|
||||
conn = aiosqlite.connect("projects/memory/checkpoints.db")
|
||||
checkpointer = AsyncSqliteSaver(conn=conn)
|
||||
await prepare_checkpoint_message(config, checkpointer)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user