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

refactor: Update frontend_node method in utils.py to use set_base_classes_from_outputs

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 19:31:32 -03:00
commit 799c08fd84
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/custom/utils.py
View file

@ -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

2
src/backend/base/langflow/template/frontend_node/base.py
View file

@ -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: