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

🐛 fix(utils.py): refactor update_template function to improve readability and maintainability

Browse source 
✨ feat(utils.py): add support for nested proxy dictionary in template to handle field and id separately
✅ test(test_graph.py): add test cases for process_flow function to ensure correct processing of grouped chat data
✅ test(test_graph.py): add test cases for update_template function to ensure correct updating of template
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-05 11:40:40 -03:00
commit de7c80de3f
2 changed files with 44 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -80,8 +80,11 @@ def update_template(template, g_nodes):
Returns:
None
"""
for key in template.keys():
field, id_ = template[key]["proxy"]
for _, value in template.items():
if not value.get("proxy"):
continue
proxy_dict = value["proxy"]
field, id_ = proxy_dict["field"], proxy_dict["id"]
node_index = next((i for i, n in enumerate(g_nodes) if n["id"] == id_), -1)
if node_index != -1:
display_name = None
@ -98,7 +101,7 @@ def update_template(template, g_nodes):
"name"
]
g_nodes[node_index]["data"]["node"]["template"][field] = template[key]
g_nodes[node_index]["data"]["node"]["template"][field] = value
g_nodes[node_index]["data"]["node"]["template"][field]["show"] = show
g_nodes[node_index]["data"]["node"]["template"][field][
"advanced"