From 799c08fd847853ebf6947d8c256f434bd110aa52 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 19:31:32 -0300 Subject: [PATCH] refactor: Update frontend_node method in utils.py to use set_base_classes_from_outputs --- src/backend/base/langflow/custom/utils.py | 2 +- src/backend/base/langflow/template/frontend_node/base.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/custom/utils.py b/src/backend/base/langflow/custom/utils.py index 7856f6c9b..49bb201a8 100644 --- a/src/backend/base/langflow/custom/utils.py +++ b/src/backend/base/langflow/custom/utils.py @@ -376,7 +376,7 @@ def build_custom_component_template_from_inputs( # Validate that there is not name overlap between inputs and outputs frontend_node.validate() # ! This should be removed when we have a better way to handle this - frontend_node.get_base_classes_from_outputs() + frontend_node.set_base_classes_from_outputs() reorder_fields(frontend_node, custom_component._get_field_order()) cc_instance = get_component_instance(custom_component, user_id=user_id) return frontend_node.to_dict(add_name=False), cc_instance diff --git a/src/backend/base/langflow/template/frontend_node/base.py b/src/backend/base/langflow/template/frontend_node/base.py index 221ed89dd..01a5e95c1 100644 --- a/src/backend/base/langflow/template/frontend_node/base.py +++ b/src/backend/base/langflow/template/frontend_node/base.py @@ -99,7 +99,7 @@ class FrontendNode(BaseModel): def add_extra_base_classes(self) -> None: pass - def get_base_classes_from_outputs(self) -> list[str]: + def set_base_classes_from_outputs(self): self.base_classes = [output_type for output in self.outputs for output_type in output.types] def validate(self) -> None: