diff --git a/src/backend/langflow/template/frontend_node/custom_components.py b/src/backend/langflow/template/frontend_node/custom_components.py index 959571cb4..dc5853ed5 100644 --- a/src/backend/langflow/template/frontend_node/custom_components.py +++ b/src/backend/langflow/template/frontend_node/custom_components.py @@ -46,7 +46,7 @@ class Component(CustomComponent): class CustomComponentFrontendNode(FrontendNode): name: str = "CustomComponent" - display_name: str = None + display_name: Optional[str] = None beta: bool = True template: Template = Template( type_name="CustomComponent", @@ -67,10 +67,8 @@ class CustomComponentFrontendNode(FrontendNode): description: Optional[str] = None base_classes: list[str] = [] - def to_dict(self): - return super().to_dict() - def to_dict_nameless(self): + def to_dict(self) -> dict: """Returns a dict representation of the frontend node.""" self.process_base_classes() return { @@ -85,4 +83,4 @@ class CustomComponentFrontendNode(FrontendNode): "beta": self.beta, "error": self.error, }, - } + } \ No newline at end of file