From af7c1c614dc26ce1ab4f1bb86e0b5571a5b61c52 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 9 Dec 2023 23:37:52 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(custom=5Fcomponents.py):=20c?= =?UTF-8?q?hange=20display=5Fname=20attribute=20type=20from=20str=20to=20O?= =?UTF-8?q?ptional[str]=20to=20allow=20for=20None=20value=20=F0=9F=94=80?= =?UTF-8?q?=20refactor(custom=5Fcomponents.py):=20remove=20unused=20to=5Fd?= =?UTF-8?q?ict=5Fnameless=20method=20and=20update=20to=5Fdict=20method=20s?= =?UTF-8?q?ignature=20to=20specify=20return=20type?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/template/frontend_node/custom_components.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) 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