初始化

2026-03-05 18:26:08 +08:00
commit 5073036034
22 changed files with 504 additions and 0 deletions
--- a/server/mcp_tools.py
+++ b/server/mcp_tools.py
@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any
+
+from autogen_ext.tools.mcp import StdioServerParams, mcp_server_tools
+
+from . import config
+
+
+@dataclass(frozen=True)
+class MCPServerConfig:
+    name: str
+    command: str
+    args: list[str]
+
+
+def _configured_servers() -> list[MCPServerConfig]:
+    if not config.ENABLE_MCP_TOOLS:
+        return []
+
+    servers: list[MCPServerConfig] = []
+    if config.MCP_WEATHER_SERVER_COMMAND:
+        servers.append(
+            MCPServerConfig(
+                name="weather",
+                command=config.MCP_WEATHER_SERVER_COMMAND,
+                args=config.MCP_WEATHER_SERVER_ARGS,
+            )
+        )
+    if config.MCP_WEBSEARCH_SERVER_COMMAND:
+        servers.append(
+            MCPServerConfig(
+                name="websearch",
+                command=config.MCP_WEBSEARCH_SERVER_COMMAND,
+                args=config.MCP_WEBSEARCH_SERVER_ARGS,
+            )
+        )
+    return servers
+
+
+async def load_mcp_tools() -> list[Any]:
+    configured_servers = _configured_servers()
+    if not configured_servers:
+        print("ℹ️ MCP 工具未配置，跳过加载。")
+        return []
+
+    loaded_tools: list[Any] = []
+    tool_names: set[str] = set()
+
+    for server in configured_servers:
+        params = StdioServerParams(
+            command=server.command,
+            args=server.args,
+            read_timeout_seconds=config.MCP_SERVER_READ_TIMEOUT_SECONDS,
+        )
+        try:
+            server_tools = await mcp_server_tools(params)
+            for tool in server_tools:
+                if tool.name in tool_names:
+                    print(f"⚠️ MCP 工具重名，已跳过: {tool.name}")
+                    continue
+                loaded_tools.append(tool)
+                tool_names.add(tool.name)
+            print(f"✅ MCP 服务已加载: {server.name} ({len(server_tools)} tools)")
+        except Exception as exc:
+            print(f"⚠️ MCP 服务加载失败: {server.name}, error={exc}")
+
+    if loaded_tools:
+        print(f"✅ MCP 工具总数: {len(loaded_tools)}")
+    else:
+        print("ℹ️ 未加载到任何 MCP 工具。")
+    return loaded_tools