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

rename

Browse Source
This commit is contained in:
goulustis
2025-10-15 17:06:45 +08:00
parent e58dd4477f
commit 4838a251b7
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lang_agent/client_tool_manager.py
View File

@@ -12,8 +12,8 @@ from lang_agent.config import InstantiateConfig
@tyro.conf.configure(tyro.conf.SuppressFixed) @tyro.conf.configure(tyro.conf.SuppressFixed)
@dataclass @dataclass
class ToolManagerConfig(InstantiateConfig): class ClientToolManagerConfig(InstantiateConfig):
_target: Type = field(default_factory=lambda: ToolManager) _target: Type = field(default_factory=lambda: ClientToolManager)
mcp_config_f: str = None mcp_config_f: str = None
"""path to all mcp configurations; expect json file""" """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." assert osp.exists(self.mcp_config_f), f"config_f {self.mcp_config_f} does not exist."
class ToolManager: class ClientToolManager:
def __init__(self, config:ToolManagerConfig): def __init__(self, config:ClientToolManagerConfig):
self.config = config self.config = config
self.populate_module() self.populate_module()
@@ -49,7 +49,7 @@ class ToolManager:
if __name__ == "__main__": if __name__ == "__main__":
# NOTE: Simple test # NOTE: Simple test
config = ToolManagerConfig() config = ClientToolManagerConfig()
tool_manager = ToolManager(config) tool_manager = ClientToolManager(config)
tools = tool_manager.get_tools() tools = tool_manager.get_tools()
print(tools) print(tools)

6
lang_agent/pipeline.py
View File

@@ -13,7 +13,7 @@ from langgraph.prebuilt import create_react_agent
from langgraph.checkpoint.memory import MemorySaver from langgraph.checkpoint.memory import MemorySaver
from lang_agent.config import InstantiateConfig 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) @tyro.conf.configure(tyro.conf.SuppressFixed)
@dataclass @dataclass
@@ -42,7 +42,7 @@ class PipelineConfig(InstantiateConfig):
"""what is my port""" """what is my port"""
# NOTE: For reference # NOTE: For reference
tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig) tool_manager_config: ClientToolManagerConfig = field(default_factory=ClientToolManagerConfig)
def __post_init__(self): def __post_init__(self):
if self.api_key == "wrong-key" or self.api_key is None: if self.api_key == "wrong-key" or self.api_key is None:
@@ -68,7 +68,7 @@ class Pipeline:
base_url=self.config.base_url) base_url=self.config.base_url)
# NOTE: placeholder for now, add graph later # 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() memory = MemorySaver()
# tools = self.tool_manager.get_tools() # tools = self.tool_manager.get_tools()
tools = [] tools = []