get langchain tools

This commit is contained in:
2025-10-15 17:35:04 +08:00
parent 32e5369b99
commit 394e44ba83

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 ClientToolManager, ClientToolManagerConfig from lang_agent.tool_manager import ToolManager, ToolManagerConfig
@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: ClientToolManagerConfig = field(default_factory=ClientToolManagerConfig) tool_manager_config: ToolManagerConfig = field(default_factory=ToolManagerConfig)
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,10 +68,10 @@ 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:ClientToolManager = self.config.tool_manager_config.setup() self.tool_manager:ToolManager = self.config.tool_manager_config.setup()
memory = MemorySaver() memory = MemorySaver()
# tools = self.tool_manager.get_tools() tools = self.tool_manager.get_langchain_tools()
tools = [] # tools = []
self.agent = create_react_agent(self.llm, tools, checkpointer=memory) self.agent = create_react_agent(self.llm, tools, checkpointer=memory)