From f380bc40e33b2f0c6bf683e0cdeb2a51f391bc6f Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 5 Jan 2026 23:14:03 +0800 Subject: [PATCH] rename tool --- lang_agent/components/tool_manager.py | 7 ++++--- lang_agent/graphs/react.py | 2 +- lang_agent/graphs/tool_nodes.py | 2 +- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/lang_agent/components/tool_manager.py b/lang_agent/components/tool_manager.py index 0699f2f..f2ae5bc 100644 --- a/lang_agent/components/tool_manager.py +++ b/lang_agent/components/tool_manager.py @@ -69,7 +69,7 @@ class ToolManager: self.populate_modules() logger.info("available tools:") - for tool in self.get_list_langchain_tools(): + for tool in self.get_langchain_tools(): logger.info(tool.name) def _get_tool_config(self)->List[ToolConfig]: @@ -161,11 +161,12 @@ class ToolManager: self.langchain_tools.append(self.fnc_to_structool(func)) return self.langchain_tools - def get_list_langchain_tools(self)->List[StructuredTool]: + def get_langchain_tools(self)->List[StructuredTool]: + """get tools usuable in langchain""" return self.langchain_tools if __name__ == "__main__": man: ToolManager = ToolManagerConfig().setup() - for lang_tool in man.get_list_langchain_tools(): + for lang_tool in man.get_langchain_tools(): print(lang_tool.name) \ No newline at end of file diff --git a/lang_agent/graphs/react.py b/lang_agent/graphs/react.py index b2fa66b..9960b5a 100644 --- a/lang_agent/graphs/react.py +++ b/lang_agent/graphs/react.py @@ -48,7 +48,7 @@ class ReactGraph(GraphBase): self.tool_manager:ToolManager = self.config.tool_manager_config.setup() memory = MemorySaver() - tools = self.tool_manager.get_list_langchain_tools() + tools = self.tool_manager.get_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/tool_nodes.py b/lang_agent/graphs/tool_nodes.py index 221048b..b59fb4d 100644 --- a/lang_agent/graphs/tool_nodes.py +++ b/lang_agent/graphs/tool_nodes.py @@ -41,7 +41,7 @@ class ToolNode(ToolNodeBase): def populate_modules(self): self.llm = make_llm(tags=["tool_llm"]) - self.tool_agent = create_agent(self.llm, self.tool_manager.get_list_langchain_tools(), checkpointer=self.mem) + self.tool_agent = create_agent(self.llm, self.tool_manager.get_langchain_tools(), checkpointer=self.mem) with open(self.config.tool_prompt_f, "r") as f: self.sys_prompt = f.read()