diff --git a/lang_agent/graphs/react.py b/lang_agent/graphs/react.py index ff46214..51de1d9 100644 --- a/lang_agent/graphs/react.py +++ b/lang_agent/graphs/react.py @@ -47,7 +47,7 @@ class ReactGraph(GraphBase): self.tool_manager:ToolManager = self.config.tool_manager_config.setup() memory = MemorySaver() - tools = self.tool_manager.get_langchain_tools() + tools = self.tool_manager.get_list_langchain_tools() self.agent = create_agent(self.llm, tools, checkpointer=memory) def invoke(self, *nargs, as_stream:bool=False, as_raw:bool=False, **kwargs): diff --git a/lang_agent/graphs/routing.py b/lang_agent/graphs/routing.py index 592a5ba..ac8cbbf 100644 --- a/lang_agent/graphs/routing.py +++ b/lang_agent/graphs/routing.py @@ -110,7 +110,7 @@ class RoutingGraph(GraphBase): tool_manager:ToolManager = self.config.tool_manager_config.setup() self.chat_model = create_agent(self.llm, [], checkpointer=self.memory) - self.tool_model = create_agent(self.llm, tool_manager.get_langchain_tools(), checkpointer=self.memory) + self.tool_model = create_agent(self.llm, tool_manager.get_list_langchain_tools(), checkpointer=self.memory) with open(self.config.sys_promp_json , "r") as f: self.prompt_dict = commentjson.load(f)