From f3ac062fd8f1d42137ccd2eb986a821c7f94dc9c Mon Sep 17 00:00:00 2001 From: jijiahao <15202802357@163.com> Date: Thu, 6 Nov 2025 16:04:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96mcp=E5=B7=A5=E5=85=B7?= =?UTF-8?q?=E7=AE=A1=E7=90=86=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lang_agent/tool_manager.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/lang_agent/tool_manager.py b/lang_agent/tool_manager.py index d1873a7..335b196 100644 --- a/lang_agent/tool_manager.py +++ b/lang_agent/tool_manager.py @@ -10,6 +10,7 @@ from fastmcp.tools.tool import Tool from lang_agent.config import InstantiateConfig, ToolConfig from lang_agent.base import LangToolBase +from lang_agent.client_tool_manager import ClientToolManagerConfig from lang_agent.rag.simple import SimpleRagConfig from lang_agent.dummy.calculator import CalculatorConfig @@ -24,6 +25,8 @@ import jax class ToolManagerConfig(InstantiateConfig): _target: Type = field(default_factory=lambda: ToolManager) + client_tool_manager: ClientToolManagerConfig = field(default_factory=ClientToolManagerConfig) + # tool configs here; MUST HAVE 'config' in name and must be dataclass rag_config: SimpleRagConfig = field(default_factory=SimpleRagConfig) @@ -102,9 +105,10 @@ class ToolManager: logger.info(f"skipping tool:{tool_name}") try: - from lang_agent.client_tool_manager import ClientToolManagerConfig - client_config = ClientToolManagerConfig() - self.client_tool_manager = ClientToolManager(client_config) + # client_config = self.config.client_tool_manager + # self.client_tool_manager = ClientToolManager(client_config) + # self.client_tool_manager = ClientToolManager(self.config.client_tool_manager) + self.client_tool_manager:ClientToolManager = self.config.client_tool_manager.setup() logger.info("Successfully initialized client_tool_manager for MCP tools") except Exception as e: logger.warning(f"Failed to initialize client_tool_manager: {e}")