From cf99a5ffa7425c5b13e8560ba5de8f6822526c73 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 10 Jun 2024 18:59:07 -0300 Subject: [PATCH] refactor: Update setup.py to handle new output types in source node --- src/backend/base/langflow/initial_setup/setup.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/initial_setup/setup.py b/src/backend/base/langflow/initial_setup/setup.py index 203d75b89..1ae955c59 100644 --- a/src/backend/base/langflow/initial_setup/setup.py +++ b/src/backend/base/langflow/initial_setup/setup.py @@ -190,7 +190,20 @@ def update_edges_with_latest_component_versions(project_data): if source_node and target_node: source_node_data = source_node.get("data").get("node") target_node_data = target_node.get("data").get("node") - new_output_types = source_node_data.get("output_types") + output_data = next( + (output for output in source_node_data.get("outputs", []) if output["name"] == source_handle["name"]), + None, + ) + if output_data: + if len(output_data.get("types")) == 1: + new_output_types = output_data.get("types") + elif output_data.get("selected"): + new_output_types = [output_data.get("selected")] + else: + new_output_types = [] + else: + new_output_types = [] + if source_handle["output_types"] != new_output_types: edge_changes_log[source_node_data["display_name"]].append( {