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

Fix component_template output_types assignment

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-29 19:27:07 -03:00
commit 8703d4410e
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/custom/utils.py
View file

@ -358,11 +358,13 @@ def create_component_template(component):
"""Create a template for a component.""" """Create a template for a component."""
component_code = component["code"] component_code = component["code"]
component_output_types = component["output_types"] component_output_types = component["output_types"]
# remove
component_extractor = CustomComponent(code=component_code) component_extractor = CustomComponent(code=component_code)
component_template = build_custom_component_template(component_extractor) component_template = build_custom_component_template(component_extractor)
component_template["output_types"] = component_output_types if not component_template["output_types"] and component_output_types:
component_template["output_types"] = component_output_types
return component_template return component_template