rename

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)

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 = []