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

🐛 fix(utils.py): fix find_last_node function to receive nodes and edges as arguments instead of data dictionary

Browse source 
🐛 fix(utils.py): fix update_source_handle function to receive g_nodes and g_edges as arguments instead of flow_data
🐛 fix(utils.py): fix get_updated_edges function to receive g_nodes and g_edges as arguments instead of base_flow and group_node_id
✨ feat(utils.py): add support for passing nodes and edges as arguments to find_last_node, update_source_handle, and get_updated_edges functions to improve reusability and modularity
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-05 13:08:09 -03:00
commit 06bbfa2e04
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/graph/graph/utils.py
View file

@ -2,11 +2,10 @@ from collections import deque
import copy import copy
def find_last_node(data): def find_last_node(nodes, edges):
""" """
This function receives a flow and returns the last node. This function receives a flow and returns the last node.
""" """
nodes, edges = data["nodes"], data["edges"]
return next((n for n in nodes if all(e["source"] != n["id"] for e in edges)), None) return next((n for n in nodes if all(e["source"] != n["id"] for e in edges)), None)
@ -29,7 +28,9 @@ def ungroup_node(group_node_data, base_flow):
g_edges = flow["data"]["edges"] g_edges = flow["data"]["edges"]
# Redirect edges to the correct proxy node # Redirect edges to the correct proxy node
updated_edges = get_updated_edges(base_flow, g_nodes, group_node_data["id"]) updated_edges = get_updated_edges(
base_flow, g_nodes, g_edges, group_node_data["id"]
)
# Update template values # Update template values
update_template(template, g_nodes) update_template(template, g_nodes)
@ -183,7 +184,7 @@ def set_new_target_handle(proxy_id, new_edge, target_handle, node):
new_edge["data"]["targetHandle"] = new_target_handle new_edge["data"]["targetHandle"] = new_target_handle
def update_source_handle(new_edge, flow_data): def update_source_handle(new_edge, g_nodes, g_edges):
""" """
Updates the source handle of a given edge to the last node in the flow data. Updates the source handle of a given edge to the last node in the flow data.
@ -194,7 +195,7 @@ def update_source_handle(new_edge, flow_data):
Returns: Returns:
dict: The updated edge with the new source handle. dict: The updated edge with the new source handle.
""" """
last_node = copy.deepcopy(find_last_node(flow_data)) last_node = copy.deepcopy(find_last_node(g_nodes, g_edges))
new_edge["source"] = last_node["id"] new_edge["source"] = last_node["id"]
new_source_handle = new_edge["data"]["sourceHandle"] new_source_handle = new_edge["data"]["sourceHandle"]
new_source_handle["id"] = last_node["id"] new_source_handle["id"] = last_node["id"]
@ -202,7 +203,7 @@ def update_source_handle(new_edge, flow_data):
return new_edge return new_edge
def get_updated_edges(base_flow, g_nodes, group_node_id): def get_updated_edges(base_flow, g_nodes, g_edges, group_node_id):
""" """
Given a base flow, a list of graph nodes and a group node id, returns a list of updated edges. Given a base flow, a list of graph nodes and a group node id, returns a list of updated edges.
An updated edge is an edge that has its target or source handle updated based on the group node id. An updated edge is an edge that has its target or source handle updated based on the group node id.
@ -222,7 +223,7 @@ def get_updated_edges(base_flow, g_nodes, group_node_id):
new_edge = update_target_handle(new_edge, g_nodes, group_node_id) new_edge = update_target_handle(new_edge, g_nodes, group_node_id)
if new_edge["source"] == group_node_id: if new_edge["source"] == group_node_id:
new_edge = update_source_handle(new_edge, base_flow) new_edge = update_source_handle(new_edge, g_nodes, g_edges)
if edge["target"] == group_node_id or edge["source"] == group_node_id: if edge["target"] == group_node_id or edge["source"] == group_node_id:
updated_edges.append(new_edge) updated_edges.append(new_edge)