diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py
index c51f9ce78..14a29bd44 100644
--- a/src/backend/langflow/api/v1/endpoints.py
+++ b/src/backend/langflow/api/v1/endpoints.py
@@ -19,7 +19,7 @@ from langflow.interface.types import (
     build_langchain_template_custom_component,
 )
 
-from langflow.interface.types import langchain_types_dict
+from langflow.interface.types import build_langchain_types_dict
 from langflow.database.base import get_session
 from sqlmodel import Session
 
@@ -29,7 +29,7 @@ router = APIRouter(tags=["Base"])
 
 @router.get("/all")
 def get_all():
-    return langchain_types_dict
+    return build_langchain_types_dict()
 
 
 # For backwards compatibility we will keep the old endpoint
diff --git a/src/backend/langflow/interface/custom/code_parser.py b/src/backend/langflow/interface/custom/code_parser.py
index e86c12cef..aa4191448 100644
--- a/src/backend/langflow/interface/custom/code_parser.py
+++ b/src/backend/langflow/interface/custom/code_parser.py
@@ -1,7 +1,8 @@
 import ast
+import inspect
 import traceback
 
-from typing import Dict, Any, Union
+from typing import Dict, Any, Type, Union
 from fastapi import HTTPException
 
 
@@ -14,10 +15,15 @@ class CodeParser:
     A parser for Python source code, extracting code details.
     """
 
-    def __init__(self, code: str) -> None:
+    def __init__(self, code: Union[str, Type]) -> None:
         """
         Initializes the parser with the provided code.
         """
+        if isinstance(code, type):
+            if not inspect.isclass(code):
+                raise ValueError("The provided code must be a class.")
+            # If the code is a class, get its source code
+            code = inspect.getsource(code)
         self.code = code
         self.data: Dict[str, Any] = {
             "imports": [],
diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py
index b1ba8573f..494c931e5 100644
--- a/src/backend/langflow/interface/types.py
+++ b/src/backend/langflow/interface/types.py
@@ -233,6 +233,3 @@ def build_langchain_template_custom_component(custom_component: CustomComponent)
         frontend_node.get("base_classes").append(base_class)
 
     return frontend_node
-
-
-langchain_types_dict = build_langchain_types_dict()