diff --git a/m12_agents_sdk_swarm/agent.py b/m12_agents_sdk_swarm/agent.py index 8f7cd5b..e5a68b5 100644 --- a/m12_agents_sdk_swarm/agent.py +++ b/m12_agents_sdk_swarm/agent.py @@ -2,9 +2,8 @@ from agents.mcp import MCPServerStdio, MCPServerStdioParams from openai import AsyncOpenAI from agents.agent import Agent from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel -from config import OPENAI_API_KEY from m12_agents_sdk_swarm.tools import execute_refund, check_seat -from config import AMAP_MAPS_API_KEY +from config import AMAP_MAPS_API_KEY,OPENAI_API_KEY client = AsyncOpenAI(api_key=OPENAI_API_KEY, base_url="https://api.deepseek.com") model = OpenAIChatCompletionsModel(model="deepseek-chat", openai_client=client) diff --git a/m12_agents_sdk_swarm/main.py b/m12_agents_sdk_swarm/main.py index 0c48c83..bb7ad31 100644 --- a/m12_agents_sdk_swarm/main.py +++ b/m12_agents_sdk_swarm/main.py @@ -33,7 +33,11 @@ async def main(): messages.append({"role": "user", "content": user_input}) # 启动流式响应 - result = Runner.run_streamed(cur_agent, input=messages, context=context_variables) + result = Runner.run_streamed( + cur_agent, + input=messages, + context=context_variables + ) # 状态变量 current_agent_name = None # 当前Agent名称