From 61673d7bf9656d6f11833200fbdcb3efc09d9135 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 10 Dec 2023 00:08:21 -0300 Subject: [PATCH] Add field serializer for display name in CustomComponentFrontendNode --- .../langflow/template/frontend_node/custom_components.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/backend/langflow/template/frontend_node/custom_components.py b/src/backend/langflow/template/frontend_node/custom_components.py index 9c9f18d5d..22fd70814 100644 --- a/src/backend/langflow/template/frontend_node/custom_components.py +++ b/src/backend/langflow/template/frontend_node/custom_components.py @@ -3,6 +3,7 @@ from typing import Optional from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode from langflow.template.template.base import Template +from pydantic import field_serializer DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent from typing import Optional, List, Dict, Union @@ -66,3 +67,9 @@ class CustomComponentFrontendNode(FrontendNode): ) description: Optional[str] = None base_classes: list[str] = [] + + @field_serializer("display_name") + def process_display_name(self, display_name: str) -> str: + """Sets the display name of the frontend node.""" + + return display_name