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

rename tool

Browse Source
This commit is contained in:
goulustis
2026-01-05 23:14:03 +08:00
parent b1d2c83cb5
commit f380bc40e3
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lang_agent/components/tool_manager.py
View File

@@ -69,7 +69,7 @@ class ToolManager:
self.populate_modules() self.populate_modules()
logger.info("available tools:") logger.info("available tools:")
for tool in self.get_list_langchain_tools(): for tool in self.get_langchain_tools():
logger.info(tool.name) logger.info(tool.name)
def _get_tool_config(self)->List[ToolConfig]: def _get_tool_config(self)->List[ToolConfig]:
@@ -161,11 +161,12 @@ class ToolManager:
self.langchain_tools.append(self.fnc_to_structool(func)) self.langchain_tools.append(self.fnc_to_structool(func))
return self.langchain_tools 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 return self.langchain_tools
if __name__ == "__main__": if __name__ == "__main__":
man: ToolManager = ToolManagerConfig().setup() 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) print(lang_tool.name)

2
lang_agent/graphs/react.py
View File

@@ -48,7 +48,7 @@ class ReactGraph(GraphBase):
self.tool_manager:ToolManager = 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_list_langchain_tools() tools = self.tool_manager.get_langchain_tools()
self.agent = create_agent(self.llm, tools, checkpointer=memory) self.agent = create_agent(self.llm, tools, checkpointer=memory)
def invoke(self, *nargs, as_stream:bool=False, as_raw:bool=False, **kwargs): def invoke(self, *nargs, as_stream:bool=False, as_raw:bool=False, **kwargs):

2
lang_agent/graphs/tool_nodes.py
View File

@@ -41,7 +41,7 @@ class ToolNode(ToolNodeBase):
def populate_modules(self): def populate_modules(self):
self.llm = make_llm(tags=["tool_llm"]) 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: with open(self.config.tool_prompt_f, "r") as f:
self.sys_prompt = f.read() self.sys_prompt = f.read()