From 5ede934a5691e4c26c22f5e79b9104322da0e6bf Mon Sep 17 00:00:00 2001 From: goulustis Date: Sat, 11 Oct 2025 14:45:57 +0800 Subject: [PATCH] add symple tests --- lang_agent/tool_manager.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/lang_agent/tool_manager.py b/lang_agent/tool_manager.py index 6e5b25d..033dde4 100644 --- a/lang_agent/tool_manager.py +++ b/lang_agent/tool_manager.py @@ -20,8 +20,12 @@ class ToolManagerConfig(InstantiateConfig): def __post_init__(self): if self.mcp_config_f is None: - self.mcp_config_f = osp.join(osp.dirname(osp.dirname(__file__)), "config", "mcp_config.json") + self.mcp_config_f = osp.join(osp.dirname(osp.dirname(__file__)), "configs", "mcp_config.json") logger.warning(f"config_f was not provided. Using default: {self.mcp_config_f}") + assert osp.exists(self.mcp_config_f), f"Default config_f {self.mcp_config_f} does not exist." + + assert osp.exists(self.mcp_config_f), f"config_f {self.mcp_config_f} does not exist." + class ToolManager: def __init__(self, config:ToolManagerConfig): @@ -42,3 +46,10 @@ class ToolManager: def get_tools(self): tools = asyncio.run(self.aget_tools()) return tools + +if __name__ == "__main__": + # NOTE: Simple test + config = ToolManagerConfig() + tool_manager = ToolManager(config) + tools = tool_manager.get_tools() + print(tools) \ No newline at end of file