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

🐛 fix(base.py): change assignment of source_handle and target_handle to use get() method to handle missing keys gracefully

Browse source 
🐛 fix(base.py): change assignment of parent_node_id and parent_is_top_level to use state dictionary instead of _data attribute
🐛 fix(test_graph.py): change arguments passed to update_source_handle() function to match its signature
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-06 16:19:42 -03:00
commit 4d6c731a9d
3 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_graph.py
View file

@ -497,7 +497,7 @@ def test_update_source_handle():
"edges": [{"source": "some_node"}],
}
updated_edge = update_source_handle(
new_edge, {"nodes": flow_data["nodes"], "edges": flow_data["edges"]}
new_edge, flow_data["nodes"], flow_data["edges"]
)
assert updated_edge["source"] == "last_node"
assert updated_edge["data"]["sourceHandle"]["id"] == "last_node"