Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化mcp工具管理函数

Browse Source
This commit is contained in:
jijiahao
2025-11-06 16:04:07 +08:00
parent 3a937fc8f8
commit f3ac062fd8
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lang_agent/tool_manager.py
View File

@@ -10,6 +10,7 @@ from fastmcp.tools.tool import Tool
from lang_agent.config import InstantiateConfig, ToolConfig from lang_agent.config import InstantiateConfig, ToolConfig
from lang_agent.base import LangToolBase from lang_agent.base import LangToolBase
from lang_agent.client_tool_manager import ClientToolManagerConfig
from lang_agent.rag.simple import SimpleRagConfig from lang_agent.rag.simple import SimpleRagConfig
from lang_agent.dummy.calculator import CalculatorConfig from lang_agent.dummy.calculator import CalculatorConfig
@@ -24,6 +25,8 @@ import jax
class ToolManagerConfig(InstantiateConfig): class ToolManagerConfig(InstantiateConfig):
_target: Type = field(default_factory=lambda: ToolManager) _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 # tool configs here; MUST HAVE 'config' in name and must be dataclass
rag_config: SimpleRagConfig = field(default_factory=SimpleRagConfig) rag_config: SimpleRagConfig = field(default_factory=SimpleRagConfig)
@@ -102,9 +105,10 @@ class ToolManager:
logger.info(f"skipping tool:{tool_name}") logger.info(f"skipping tool:{tool_name}")
try: try:
from lang_agent.client_tool_manager import ClientToolManagerConfig # client_config = self.config.client_tool_manager
client_config = ClientToolManagerConfig() # self.client_tool_manager = ClientToolManager(client_config)
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") logger.info("Successfully initialized client_tool_manager for MCP tools")
except Exception as e: except Exception as e:
logger.warning(f"Failed to initialize client_tool_manager: {e}") logger.warning(f"Failed to initialize client_tool_manager: {e}")