diff --git a/src/agents/chatbot/graph.py b/src/agents/chatbot/graph.py index 8a21c92d..cc4ebc92 100644 --- a/src/agents/chatbot/graph.py +++ b/src/agents/chatbot/graph.py @@ -19,8 +19,6 @@ class ChatbotAgent(BaseAgent): def __init__(self, **kwargs): super().__init__(**kwargs) - self.graph = None - self.checkpointer = None self.context_schema = Context async def get_tools(self, tools: list[str] = None, mcps=None, knowledges=None): diff --git a/src/agents/reporter/graph.py b/src/agents/reporter/graph.py index ec6655b6..ff9c12b4 100644 --- a/src/agents/reporter/graph.py +++ b/src/agents/reporter/graph.py @@ -28,7 +28,7 @@ class SqlReporterAgent(BaseAgent): # 创建 SqlReporterAgent graph = create_agent( - model=load_chat_model(config.default_model), # 默认模型,会被 middleware 覆盖 + model=load_chat_model(context.model), # 使用 context 中的模型配置 system_prompt=context.system_prompt, tools=await self.get_tools(), checkpointer=await self._get_checkpointer(),