Merge pull request #1245 from AstrBotDevs/perf-mcpserver

perf: 适配 MCP 配置文件带 mcpServers 的情况(Cursor)

astrbot/core/provider/func_tool_manager.py

@@ -95,7 +95,10 @@ class MCPClient:
             mcp_server_config (dict): Configuration for the MCP server. See https://modelcontextprotocol.io/quickstart/server
         """
         cfg = mcp_server_config.copy()
-        cfg.pop("active", None)
+        if "mcpServers" in cfg and len(cfg["mcpServers"]) > 0:
+            key_0 = list(cfg["mcpServers"].keys())[0]
+            cfg = cfg["mcpServers"][key_0]
+        cfg.pop("active", None) # Remove active flag from config
         server_params = mcp.StdioServerParameters(
             **cfg,
         )
@@ -260,6 +263,11 @@ class FuncCall:
                     if data["name"] in self.mcp_client_event:
                         self.mcp_client_event[data["name"]].set()
                         self.mcp_client_event.pop(data["name"], None)
+                        self.func_list = [
+                            f
+                            for f in self.func_list
+                            if not (f.origin == "mcp" and f.mcp_server_name == data["name"])
+                        ]
                 else:
                     for name in self.mcp_client_dict.keys():
                         # await self._terminate_mcp_client(name)
@@ -267,6 +275,11 @@ class FuncCall:
                         if name in self.mcp_client_event:
                             self.mcp_client_event[name].set()
                             self.mcp_client_event.pop(name, None)
+                    self.func_list = [
+                        f
+                        for f in self.func_list
+                        if f.origin != "mcp"
+                    ]
 
     async def _init_mcp_client_task_wrapper(
         self, name: str, cfg: dict, event: asyncio.Event

astrbot/dashboard/routes/tools.py

@@ -106,6 +106,10 @@ class ToolsRoute(Route):
             # 复制所有配置字段
             for key, value in server_data.items():
                 if key not in ["name", "active", "tools"]:  # 排除特殊字段
+                    if key == "mcpServers":
+                        key_0 = list(server_data["mcpServers"].keys())[0] # 不考虑为空的情况
+                        server_config = server_data["mcpServers"][key_0]
+                    else:
                         server_config[key] = value
                     has_valid_config = True
 
@@ -163,6 +167,10 @@ class ToolsRoute(Route):
             # 复制所有配置字段
             for key, value in server_data.items():
                 if key not in ["name", "active", "tools"]:  # 排除特殊字段
+                    if key == "mcpServers":
+                        key_0 = list(server_data["mcpServers"].keys())[0] # 不考虑为空的情况
+                        server_config = server_data["mcpServers"][key_0]
+                    else:
                         server_config[key] = value
                     only_update_active = False
 

dashboard/src/views/ToolUsePage.vue

@@ -32,6 +32,7 @@
         <v-card-title class="d-flex align-center py-3 px-4">
           <v-icon color="primary" class="me-2">mdi-server</v-icon>
           <span class="text-h6">MCP 服务器</span>
+          <v-progress-circular indeterminate color="primary" size="24" style="margin-left: 16px;" v-show="loading"></v-progress-circular>
           <v-spacer></v-spacer>
           <v-btn color="primary" prepend-icon="mdi-plus" variant="tonal" @click="showMcpServerDialog = true">
             新增服务器
@@ -404,6 +405,11 @@ export default {
   mounted() {
     this.getServers();
     this.getTools();
+    setInterval(() => {
+      this.getServers();
+      this.getTools();
+    }, 5000); // 每 5 秒刷新一次服务器列表
+    
   },
 
   methods: {
@@ -420,12 +426,15 @@ export default {
     },
     
     getServers() {
+      this.loading = true
       axios.get('/api/tools/mcp/servers')
         .then(response => {
           this.mcpServers = response.data.data || [];
+          this.loading = false
         })
         .catch(error => {
           this.showError("获取 MCP 服务器列表失败: " + error.message);
+          this.loading = false
         });
     },