From f86c8e8cab8b51125afa5939bf0075e39504e490 Mon Sep 17 00:00:00 2001 From: Raven95676 Date: Thu, 17 Jul 2025 23:17:23 +0800 Subject: [PATCH] perf: ensure MCP client termination in cleanup process --- astrbot/core/provider/func_tool_manager.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/astrbot/core/provider/func_tool_manager.py b/astrbot/core/provider/func_tool_manager.py index f8cacff75..cea2e4f38 100644 --- a/astrbot/core/provider/func_tool_manager.py +++ b/astrbot/core/provider/func_tool_manager.py @@ -356,12 +356,14 @@ class FuncCall: await self._init_mcp_client(name, cfg) await event.wait() logger.info(f"收到 MCP 客户端 {name} 终止信号") - await self._terminate_mcp_client(name) except Exception as e: import traceback traceback.print_exc() logger.error(f"初始化 MCP 客户端 {name} 失败: {e}") + finally: + # 无论如何都能清理 + await self._terminate_mcp_client(name) async def _init_mcp_client(self, name: str, config: dict) -> None: """初始化单个MCP客户端""" @@ -629,8 +631,3 @@ class FuncCall: def __repr__(self): return str(self.func_list) - - async def terminate(self): - for name in self.mcp_client_dict.keys(): - await self._terminate_mcp_client(name) - logger.debug(f"清理 MCP 客户端 {name} 资源")