rename tool

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)

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):

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()