From 4838a251b7eca893ee5d81e619fb1d0cb33ebed7 Mon Sep 17 00:00:00 2001 From: goulustis Date: Wed, 15 Oct 2025 17:06:45 +0800 Subject: [PATCH] rename --- lang_agent/client_tool_manager.py | 12 ++++++------ lang_agent/pipeline.py | 6 +++--- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/lang_agent/client_tool_manager.py b/lang_agent/client_tool_manager.py index 033dde4..c94de45 100644 --- a/lang_agent/client_tool_manager.py +++ b/lang_agent/client_tool_manager.py @@ -12,8 +12,8 @@ from lang_agent.config import InstantiateConfig @tyro.conf.configure(tyro.conf.SuppressFixed) @dataclass -class ToolManagerConfig(InstantiateConfig): - _target: Type = field(default_factory=lambda: ToolManager) +class ClientToolManagerConfig(InstantiateConfig): + _target: Type = field(default_factory=lambda: ClientToolManager) mcp_config_f: str = None """path to all mcp configurations; expect json file""" @@ -27,8 +27,8 @@ class ToolManagerConfig(InstantiateConfig): assert osp.exists(self.mcp_config_f), f"config_f {self.mcp_config_f} does not exist." -class ToolManager: - def __init__(self, config:ToolManagerConfig): +class ClientToolManager: + def __init__(self, config:ClientToolManagerConfig): self.config = config self.populate_module() @@ -49,7 +49,7 @@ class ToolManager: if __name__ == "__main__": # NOTE: Simple test - config = ToolManagerConfig() - tool_manager = ToolManager(config) + config = ClientToolManagerConfig() + tool_manager = ClientToolManager(config) tools = tool_manager.get_tools() print(tools) \ No newline at end of file diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 4a38852..5d9f4f4 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -13,7 +13,7 @@ from langgraph.prebuilt import create_react_agent from langgraph.checkpoint.memory import MemorySaver from lang_agent.config import InstantiateConfig -from lang_agent.client_tool_manager import ToolManager, ToolManagerConfig +from lang_agent.client_tool_manager import ClientToolManager, ClientToolManagerConfig @tyro.conf.configure(tyro.conf.SuppressFixed) @dataclass @@ -42,7 +42,7 @@ class PipelineConfig(InstantiateConfig): """what is my port""" # NOTE: For reference - tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig) + tool_manager_config: ClientToolManagerConfig = field(default_factory=ClientToolManagerConfig) def __post_init__(self): if self.api_key == "wrong-key" or self.api_key is None: @@ -68,7 +68,7 @@ class Pipeline: base_url=self.config.base_url) # NOTE: placeholder for now, add graph later - self.tool_manager:ToolManager = self.config.tool_manager_config.setup() + self.tool_manager:ClientToolManager = self.config.tool_manager_config.setup() memory = MemorySaver() # tools = self.tool_manager.get_tools() tools = []