diff --git a/lang_agent/tool_manager.py b/lang_agent/tool_manager.py index 41151f7..0b21f0a 100644 --- a/lang_agent/tool_manager.py +++ b/lang_agent/tool_manager.py @@ -16,7 +16,8 @@ from lang_agent.rag.simple import SimpleRagConfig from lang_agent.dummy.calculator import CalculatorConfig from catering_end.lang_tool import CartToolConfig, CartTool -from langchain.tools import StructuredTool +# from langchain.tools import StructuredTool +from langchain_core.tools.structured import StructuredTool @tyro.conf.configure(tyro.conf.SuppressFixed) @dataclass @@ -115,5 +116,4 @@ class ToolManager: StructuredTool.from_function(func=func) ) - return out - # return [StructuredTool.from_function(func=func) for func in self.get_tool_fncs()] \ No newline at end of file + return out \ No newline at end of file