From b6482f1cf6bfb2d81de032160101355178984708 Mon Sep 17 00:00:00 2001
From: Annyfee <2287551746@qq.com>
Date: Sat, 27 Dec 2025 09:48:52 +0800
Subject: [PATCH] chore:minor adjustments

---
 m12_agents_sdk_swarm/agent.py | 3 +--
 m12_agents_sdk_swarm/main.py  | 6 +++++-
 2 files changed, 6 insertions(+), 3 deletions(-)

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名称