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

Refactor tweak application logic in process_tweaks function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-24 16:34:02 -03:00
commit fa9af83b7c
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/processing/process.py
View file

@ -192,7 +192,7 @@ def apply_tweaks(node: Dict[str, Any], node_tweaks: Dict[str, Any]) -> None:
if tweak_name not in template_data: if tweak_name not in template_data:
logger.warning(f"Node {node.get('id')} does not have a tweak named {tweak_name}") logger.warning(f"Node {node.get('id')} does not have a tweak named {tweak_name}")
continue continue
if tweak_name and tweak_value and tweak_name in template_data: if tweak_name in template_data:
key = tweak_name if tweak_name == "file_path" else "value" key = tweak_name if tweak_name == "file_path" else "value"
template_data[tweak_name][key] = tweak_value template_data[tweak_name][key] = tweak_value
@ -220,15 +220,18 @@ def process_tweaks(graph_data: Dict[str, Any], tweaks: Union["Tweaks", Dict[str,
nodes = validate_input(graph_data, tweaks) nodes = validate_input(graph_data, tweaks)
nodes_map = {node.get("id"): node for node in nodes} nodes_map = {node.get("id"): node for node in nodes}
nodes_display_name_map = {node.get("data", {}).get("node", {}).get("display_name"): node for node in nodes}
all_nodes_tweaks = {} all_nodes_tweaks = {}
for key, value in tweaks.items(): for key, value in tweaks.items():
if isinstance(value, dict): if isinstance(value, dict):
if node := nodes_map.get(key): if node := nodes_map.get(key):
apply_tweaks(node, value) apply_tweaks(node, value)
elif node := nodes_display_name_map.get(key):
apply_tweaks(node, value)
else: else:
all_nodes_tweaks[key] = value all_nodes_tweaks[key] = value
if all_nodes_tweaks:
for node in nodes: for node in nodes:
apply_tweaks(node, all_nodes_tweaks) apply_tweaks(node, all_nodes_tweaks)