diff --git a/src/backend/langflow/template/frontend_node/tools.py b/src/backend/langflow/template/frontend_node/tools.py index fa3942bd2..6b6903fdc 100644 --- a/src/backend/langflow/template/frontend_node/tools.py +++ b/src/backend/langflow/template/frontend_node/tools.py @@ -53,7 +53,7 @@ class ToolNode(FrontendNode): ], ) description: str = "Converts a chain, agent or function into a tool." - base_classes: list[str] = ["Tool"] + base_classes: list[str] = ["Tool", "BaseTool"] def to_dict(self): return super().to_dict()