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

add symple tests

Browse Source
This commit is contained in:
goulustis
2025-10-11 14:45:57 +08:00
parent 0ed5c6ec8a
commit 5ede934a56
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lang_agent/tool_manager.py
View File

@@ -20,8 +20,12 @@ class ToolManagerConfig(InstantiateConfig):
def __post_init__(self): def __post_init__(self):
if self.mcp_config_f is None: 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}") 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: class ToolManager:
def __init__(self, config:ToolManagerConfig): def __init__(self, config:ToolManagerConfig):
@@ -42,3 +46,10 @@ class ToolManager:
def get_tools(self): def get_tools(self):
tools = asyncio.run(self.aget_tools()) tools = asyncio.run(self.aget_tools())
return tools return tools
if __name__ == "__main__":
# NOTE: Simple test
config = ToolManagerConfig()
tool_manager = ToolManager(config)
tools = tool_manager.get_tools()
print(tools)