From 394e44ba8367d6fdcc875f646d590971b7340fca Mon Sep 17 00:00:00 2001 From: goulustis Date: Wed, 15 Oct 2025 17:35:04 +0800 Subject: [PATCH] get langchain tools --- lang_agent/pipeline.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 5d9f4f4..aa55484 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 ClientToolManager, ClientToolManagerConfig +from lang_agent.tool_manager import ToolManager, ToolManagerConfig @tyro.conf.configure(tyro.conf.SuppressFixed) @dataclass @@ -42,7 +42,7 @@ class PipelineConfig(InstantiateConfig): """what is my port""" # NOTE: For reference - tool_manager_config: ClientToolManagerConfig = field(default_factory=ClientToolManagerConfig) + tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig) def __post_init__(self): if self.api_key == "wrong-key" or self.api_key is None: @@ -68,10 +68,10 @@ class Pipeline: base_url=self.config.base_url) # NOTE: placeholder for now, add graph later - self.tool_manager:ClientToolManager = self.config.tool_manager_config.setup() + self.tool_manager:ToolManager = self.config.tool_manager_config.setup() memory = MemorySaver() - # tools = self.tool_manager.get_tools() - tools = [] + tools = self.tool_manager.get_langchain_tools() + # tools = [] self.agent = create_react_agent(self.llm, tools, checkpointer=memory)