From 20555a4e88f011e0e6e7dc78ac7d2fd4688ab8e6 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 4 Oct 2023 20:45:29 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(base.py):=20remove=20unused?= =?UTF-8?q?=20process=5Fflow=20function=20call=20in=20Graph=20class=20to?= =?UTF-8?q?=20improve=20code=20readability=20and=20performance?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🐛 fix(utils.py): change condition from `node["data"]["node"]["flow"]` to `node["data"]["node"].get("flow")` to handle cases where "flow" key is missing in node data 🐛 fix(utils.py): fix incorrect variable name in get_updated_edges function, change `new_edge` to `edge` in the condition --- src/backend/langflow/graph/graph/base.py | 1 - src/backend/langflow/graph/graph/utils.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index 5d2688a0a..1a2ce2625 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -45,7 +45,6 @@ class Graph: """ if "data" in payload: payload = payload["data"] - payload = process_flow(payload) try: nodes = payload["nodes"] edges = payload["edges"] diff --git a/src/backend/langflow/graph/graph/utils.py b/src/backend/langflow/graph/graph/utils.py index e5d44db6f..de6385ba2 100644 --- a/src/backend/langflow/graph/graph/utils.py +++ b/src/backend/langflow/graph/graph/utils.py @@ -152,7 +152,7 @@ def set_new_target_handle(proxy_id, new_edge, target_handle, node): "type": _type, "id": proxy_id, } - if node["data"]["node"]["flow"]: + if node["data"]["node"].get("flow"): new_target_handle["proxy"] = { "field": node["data"]["node"]["template"][field]["proxy"]["field"], "id": node["data"]["node"]["template"][field]["proxy"]["id"], @@ -203,6 +203,6 @@ def get_updated_edges(base_flow, g_nodes, group_node_id): if new_edge["source"] == group_node_id: new_edge = update_source_handle(new_edge, g_nodes) - if new_edge["target"] == group_node_id or new_edge["source"] == group_node_id: + if edge["target"] == group_node_id or edge["source"] == group_node_id: updated_edges.append(new_edge) return updated_edges