diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py index 074a85499..4801da086 100644 --- a/src/backend/langflow/template/frontend_node/base.py +++ b/src/backend/langflow/template/frontend_node/base.py @@ -14,6 +14,7 @@ class FrontendNode(BaseModel): description: str base_classes: List[str] name: str = "" + display_name: str = "" def to_dict(self) -> dict: return { @@ -21,7 +22,8 @@ class FrontendNode(BaseModel): "template": self.template.to_dict(self.format_field), "description": self.description, "base_classes": self.base_classes, - } + "display_name": self.display_name or self.name, + }, } def add_extra_fields(self) -> None: