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

build langchain tools

Browse Source
This commit is contained in:
goulustis
2025-10-29 13:33:18 +08:00
parent c0cd0149a1
commit a661e3a724
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lang_agent/tool_manager.py
View File

@@ -89,7 +89,6 @@ class ToolManager:
"""instantiate all object with tools""" """instantiate all object with tools"""
self.tool_fncs = [] self.tool_fncs = []
self.tool_dict = {}
tool_configs = self._get_tool_config() tool_configs = self._get_tool_config()
for tool_conf in tool_configs: for tool_conf in tool_configs:
tool_name = tool_conf.get_name()[:-6] tool_name = tool_conf.get_name()[:-6]
@@ -97,10 +96,11 @@ class ToolManager:
logger.info(f"making tool:{tool_name}") logger.info(f"making tool:{tool_name}")
fnc_list = self._get_tool_fnc(tool_conf.setup()) fnc_list = self._get_tool_fnc(tool_conf.setup())
self.tool_fncs.extend(fnc_list) self.tool_fncs.extend(fnc_list)
self.tool_dict[tool_name] = fnc_list
else: else:
logger.info(f"skipping tool:{tool_name}") logger.info(f"skipping tool:{tool_name}")
self._build_langchain_tools()
def get_tool_fncs(self): def get_tool_fncs(self):
return self.tool_fncs return self.tool_fncs
@@ -118,13 +118,12 @@ class ToolManager:
else: else:
return StructuredTool.from_function(func=func) return StructuredTool.from_function(func=func)
def _build_langchain_tools(self):
self.langchain_tools = []
for func in self.get_tool_fncs():
self.langchain_tools.append(self.fnc_to_structool(func))
return self.langchain_tools
def get_list_langchain_tools(self): def get_list_langchain_tools(self):
out = [] return self.langchain_tools
for func in self.get_tool_fncs():
out.append(self.fnc_to_structool(func))
return out
def get_dict_langchain_tools(self):
return jax.tree_util.tree_map(self.fnc_to_structool, self.tool_dict)