From 8703d4410ef29c5cb77c71e679e88431b88a90cd Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 29 Feb 2024 19:27:07 -0300 Subject: [PATCH] Fix component_template output_types assignment --- src/backend/langflow/interface/custom/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/custom/utils.py b/src/backend/langflow/interface/custom/utils.py index e1c8191ab..2e4d3c909 100644 --- a/src/backend/langflow/interface/custom/utils.py +++ b/src/backend/langflow/interface/custom/utils.py @@ -358,11 +358,13 @@ def create_component_template(component): """Create a template for a component.""" component_code = component["code"] component_output_types = component["output_types"] + # remove component_extractor = CustomComponent(code=component_code) 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