make langchain tools
This commit is contained in:
@@ -15,6 +15,8 @@ from lang_agent.rag.simple import SimpleRagConfig
|
|||||||
from lang_agent.dummy.calculator import CalculatorConfig
|
from lang_agent.dummy.calculator import CalculatorConfig
|
||||||
from catering_end.lang_tool import CartToolConfig, CartTool
|
from catering_end.lang_tool import CartToolConfig, CartTool
|
||||||
|
|
||||||
|
from langchain.tools import StructuredTool
|
||||||
|
|
||||||
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
||||||
@dataclass
|
@dataclass
|
||||||
class ToolManagerConfig(InstantiateConfig):
|
class ToolManagerConfig(InstantiateConfig):
|
||||||
@@ -69,3 +71,7 @@ class ToolManager:
|
|||||||
|
|
||||||
def get_tool_fncs(self):
|
def get_tool_fncs(self):
|
||||||
return self.tool_fncs
|
return self.tool_fncs
|
||||||
|
|
||||||
|
|
||||||
|
def get_langchain_tools(self):
|
||||||
|
return [StructuredTool.from_function(func=func) for func in self.get_tool_fncs()]
|
||||||
Reference in New Issue
Block a user