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()
diff --git a/src/frontend/package-lock.json b/src/frontend/package-lock.json
index e24e28bd9..9b90b435e 100644
--- a/src/frontend/package-lock.json
+++ b/src/frontend/package-lock.json
@@ -40,7 +40,6 @@
                 "reactflow": "^11.5.5",
                 "tailwindcss": "^3.2.6",
                 "typescript": "^4.9.5",
-                "use-debounce": "^9.0.4",
                 "web-vitals": "^2.1.4"
             }
         },
@@ -17045,17 +17044,6 @@
                 "requires-port": "^1.0.0"
             }
         },
-        "node_modules/use-debounce": {
-            "version": "9.0.4",
-            "resolved": "https://registry.npmjs.org/use-debounce/-/use-debounce-9.0.4.tgz",
-            "integrity": "sha512-6X8H/mikbrt0XE8e+JXRtZ8yYVvKkdYRfmIhWZYsP8rcNs9hk3APV8Ua2mFkKRLcJKVdnX2/Vwrmg2GWKUQEaQ==",
-            "engines": {
-                "node": ">= 10.0.0"
-            },
-            "peerDependencies": {
-                "react": ">=16.8.0"
-            }
-        },
         "node_modules/use-sync-external-store": {
             "version": "1.2.0",
             "resolved": "https://registry.npmjs.org/use-sync-external-store/-/use-sync-external-store-1.2.0.tgz",
diff --git a/src/frontend/package.json b/src/frontend/package.json
index 0a33f0fd8..55d777687 100644
--- a/src/frontend/package.json
+++ b/src/frontend/package.json
@@ -35,7 +35,6 @@
         "reactflow": "^11.5.5",
         "tailwindcss": "^3.2.6",
         "typescript": "^4.9.5",
-        "use-debounce": "^9.0.4",
         "web-vitals": "^2.1.4"
     },
     "scripts": {
@@ -62,5 +61,5 @@
             "last 1 safari version"
         ]
     },
-    "proxy": "http://127.0.0.1:7860"
+    "proxy": "http://backend:7860"
 }
\ No newline at end of file