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
def find_last_node(data):
def find_last_node(nodes, edges):
"""
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)
@ -29,7 +28,9 @@ def ungroup_node(group_node_data, base_flow):
g_edges = flow["data"]["edges"]
# 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(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
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.
@ -194,7 +195,7 @@ def update_source_handle(new_edge, flow_data):
Returns:
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_source_handle = new_edge["data"]["sourceHandle"]
new_source_handle["id"] = last_node["id"]
@ -202,7 +203,7 @@ def update_source_handle(new_edge, flow_data):
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.
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)
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:
updated_edges.append(new_edge)