From 9162b4258dbe3fe8a993412852cf02e41110495c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=B1=E6=BD=AE?= <zhuchaowe@users.noreply.github.com>
Date: Tue, 23 Dec 2025 22:26:59 +0800
Subject: [PATCH] update init agent

---
 agent/deep_assistant.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/agent/deep_assistant.py b/agent/deep_assistant.py
index 707cb2d..f77d16d 100644
--- a/agent/deep_assistant.py
+++ b/agent/deep_assistant.py
@@ -109,13 +109,7 @@ async def init_agent(config: AgentConfig):
     """
 
 
-    # 从连接池获取 checkpointer
-    checkpointer = None
-    if config.session_id:
-        from .checkpoint_manager import get_checkpointer_manager
-        manager = get_checkpointer_manager()
-        checkpointer = await manager.acquire_for_agent()
-        await prepare_checkpoint_message(config, checkpointer)
+
 
     # 加载配置
     final_system_prompt = await load_system_prompt_async(
@@ -152,6 +146,7 @@ async def init_agent(config: AgentConfig):
     # 创建新的 agent（不再缓存）
     logger.info(f"Creating new agent for session: {getattr(config, 'session_id', 'no-session')}")
 
+    checkpointer = None
     create_start = time.time()
     if config.robot_type == "deep_agent":
         # 使用 DeepAgentX 创建 agent
@@ -181,7 +176,12 @@ async def init_agent(config: AgentConfig):
         )
         middleware.append(tool_output_middleware)
 
-        if checkpointer:
+        # 从连接池获取 checkpointer
+        if config.session_id:
+            from .checkpoint_manager import get_checkpointer_manager
+            manager = get_checkpointer_manager()
+            checkpointer = await manager.acquire_for_agent()
+            await prepare_checkpoint_message(config, checkpointer)
             summarization_middleware = SummarizationMiddleware(
                 model=llm_instance,
                 max_tokens_before_summary=SUMMARIZATION_MAX_TOKENS,