diff --git a/src/backend/langflow/interface/chains/base.py b/src/backend/langflow/interface/chains/base.py
index 9dc8ded3f..7b8da370b 100644
--- a/src/backend/langflow/interface/chains/base.py
+++ b/src/backend/langflow/interface/chains/base.py
@@ -37,7 +37,9 @@ class ChainCreator(LangChainTypeCreator):
         try:
             if name in get_custom_nodes(self.type_name).keys():
                 return get_custom_nodes(self.type_name)[name]
-            return build_template_from_class(name, self.type_to_loader_dict)
+            return build_template_from_class(
+                name, self.type_to_loader_dict, add_function=True
+            )
         except ValueError as exc:
             raise ValueError("Chain not found") from exc
         except AttributeError as exc:
diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py
index f0f2112a6..32dbec48a 100644
--- a/src/backend/langflow/template/nodes.py
+++ b/src/backend/langflow/template/nodes.py
@@ -216,7 +216,7 @@ class InitializeAgentNode(FrontendNode):
         ],
     )
     description: str = """Construct a json agent from an LLM and tools."""
-    base_classes: list[str] = ["AgentExecutor"]
+    base_classes: list[str] = ["AgentExecutor", "function"]
 
     def to_dict(self):
         return super().to_dict()