jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add field serializer for display name in

Browse source 
CustomComponentFrontendNode
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-10 00:08:21 -03:00
commit 61673d7bf9
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/template/frontend_node/custom_components.py
View file

@ -3,6 +3,7 @@ from typing import Optional
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
from langflow.template.template.base import Template from langflow.template.template.base import Template
from pydantic import field_serializer
DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent
from typing import Optional, List, Dict, Union from typing import Optional, List, Dict, Union
@ -66,3 +67,9 @@ class CustomComponentFrontendNode(FrontendNode):
) )
description: Optional[str] = None description: Optional[str] = None
base_classes: list[str] = [] 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