改进注释与字符串并增加对SSE支持的优化处理

izerui · izerui · commit ace7a1e77c86 · 2025-06-12T23:07:13.000+08:00
diff --git a/examples/mcp/sse_example/main.py b/examples/mcp/sse_example/main.py
@@ -8,72 +8,77 @@
 from agents import Agent, Runner, gen_trace_id, trace
 from agents.mcp import MCPServer, MCPServerSse
 from agents.model_settings import ModelSettings
+from examples.models import get_agent_chat_model
 
 
 async def run(mcp_server: MCPServer):
+    """运行代理示例"""
+    deepseek = get_agent_chat_model("deepseek-v3")
     agent = Agent(
-        name="Assistant",
-        instructions="Use the tools to answer the questions.",
+        name="助手",
+        instructions="使用工具回答问题",
         mcp_servers=[mcp_server],
         model_settings=ModelSettings(tool_choice="required"),
+        model=deepseek,
     )
 
-    # Use the `add` tool to add two numbers
-    message = "Add these numbers: 7 and 22."
-    print(f"Running: {message}")
+    # 使用`add`工具相加两个数字
+    message = "计算这两个数的和：7和22"
+    print(f"执行: {message}")
     result = await Runner.run(starting_agent=agent, input=message)
     print(result.final_output)
 
-    # Run the `get_weather` tool
-    message = "What's the weather in Tokyo?"
-    print(f"\n\nRunning: {message}")
+    # 运行`get_weather`工具
+    message = "东京的天气如何？"
+    print(f"\n\n执行: {message}")
     result = await Runner.run(starting_agent=agent, input=message)
     print(result.final_output)
 
-    # Run the `get_secret_word` tool
-    message = "What's the secret word?"
-    print(f"\n\nRunning: {message}")
+    # 运行`get_secret_word`工具
+    message = "秘密单词是什么？"
+    print(f"\n\n执行: {message}")
     result = await Runner.run(starting_agent=agent, input=message)
     print(result.final_output)
 
 
 async def main():
+    """主函数"""
     async with MCPServerSse(
-        name="SSE Python Server",
+        name="SSE Python服务器",
         params={
             "url": "http://localhost:8000/sse",
         },
     ) as server:
         trace_id = gen_trace_id()
-        with trace(workflow_name="SSE Example", trace_id=trace_id):
-            print(f"View trace: https://platform.openai.com/traces/trace?trace_id={trace_id}\n")
+        with trace(workflow_name="SSE示例", trace_id=trace_id):
+            print(f"查看追踪: https://platform.openai.com/traces/trace?trace_id={trace_id}\n")
             await run(server)
 
 
 if __name__ == "__main__":
-    # Let's make sure the user has uv installed
+    # 检查用户是否安装了uv
     if not shutil.which("uv"):
         raise RuntimeError(
-            "uv is not installed. Please install it: https://docs.astral.sh/uv/getting-started/installation/"
+            "未安装uv。请安装: https://docs.astral.sh/uv/getting-started/installation/"
         )
 
-    # We'll run the SSE server in a subprocess. Usually this would be a remote server, but for this
-    # demo, we'll run it locally at http://localhost:8000/sse
+    # 在子进程中运行SSE服务器。通常这会是一个远程服务器，
+    # 但在这个演示中，我们将在本地运行它：http://localhost:8000/sse
     process: subprocess.Popen[Any] | None = None
     try:
         this_dir = os.path.dirname(os.path.abspath(__file__))
         server_file = os.path.join(this_dir, "server.py")
 
-        print("Starting SSE server at http://localhost:8000/sse ...")
+        print("正在启动SSE服务器: http://localhost:8000/sse ...")
 
-        # Run `uv run server.py` to start the SSE server
+        # 运行`uv run server.py`启动SSE服务器
         process = subprocess.Popen(["uv", "run", server_file])
-        # Give it 3 seconds to start
+        # 等待3秒让它启动
         time.sleep(3)
 
-        print("SSE server started. Running example...\n\n")
+        print("SSE服务器已启动。正在运行示例...\n\n")
     except Exception as e:
-        print(f"Error starting SSE server: {e}")
+        print(f"启动SSE服务器时出错: {e}")
         exit(1)
 
     try:
diff --git a/examples/mcp/sse_example/server.py b/examples/mcp/sse_example/server.py
@@ -3,31 +3,34 @@
 import requests
 from mcp.server.fastmcp import FastMCP
 
-# Create server
-mcp = FastMCP("Echo Server")
+# 创建SSE服务器实例
+mcp = FastMCP("回声服务器")
 
 
 @mcp.tool()
 def add(a: int, b: int) -> int:
-    """Add two numbers"""
-    print(f"[debug-server] add({a}, {b})")
+    """两个数字相加"""
+    print(f"[调试-服务器] add({a}, {b})")
     return a + b
 
 
 @mcp.tool()
 def get_secret_word() -> str:
-    print("[debug-server] get_secret_word()")
-    return random.choice(["apple", "banana", "cherry"])
+    """获取随机秘密单词"""
+    print("[调试-服务器] get_secret_word()")
+    return random.choice(["苹果", "香蕉", "樱桃"])
 
 
 @mcp.tool()
 def get_current_weather(city: str) -> str:
-    print(f"[debug-server] get_current_weather({city})")
+    """获取指定城市的当前天气"""
+    print(f"[调试-服务器] get_current_weather({city})")
 
     endpoint = "https://wttr.in"
-    response = requests.get(f"{endpoint}/{city}")
+    response = requests.get(f"{endpoint}/{city}?format=3")  # 使用简洁格式
     return response.text
 
 
 if __name__ == "__main__":
+    # 启动SSE服务器
     mcp.run(transport="sse")
