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

update init agent

Browse Source
This commit is contained in:
朱潮 2025-12-23 22:26:59 +08:00
parent d8dc973b95
commit 9162b4258d
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
agent/deep_assistant.py
View File

@ -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( final_system_prompt = await load_system_prompt_async(
@ -152,6 +146,7 @@ async def init_agent(config: AgentConfig):
# 创建新的 agent（不再缓存） # 创建新的 agent（不再缓存）
logger.info(f"Creating new agent for session: {getattr(config, 'session_id', 'no-session')}") logger.info(f"Creating new agent for session: {getattr(config, 'session_id', 'no-session')}")
checkpointer = None
create_start = time.time() create_start = time.time()
if config.robot_type == "deep_agent": if config.robot_type == "deep_agent":
# 使用 DeepAgentX 创建 agent # 使用 DeepAgentX 创建 agent
@ -181,7 +176,12 @@ async def init_agent(config: AgentConfig):
) )
middleware.append(tool_output_middleware) 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( summarization_middleware = SummarizationMiddleware(
model=llm_instance, model=llm_instance,
max_tokens_before_summary=SUMMARIZATION_MAX_TOKENS, max_tokens_before_summary=SUMMARIZATION_MAX_TOKENS,