From e327c381bebba3b7ee8873b83bb99678c3e2923f Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Fri, 19 Jan 2024 17:58:35 -0300 Subject: [PATCH] Add function to apply tweaks on vertex --- src/backend/langflow/processing/process.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index ee7ce607d..64d982d28 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -273,6 +273,12 @@ def apply_tweaks(node: Dict[str, Any], node_tweaks: Dict[str, Any]) -> None: key = tweak_name if tweak_name == "file_path" else "value" template_data[tweak_name][key] = tweak_value +def apply_tweaks_on_vertex(vertex: Vertex, node_tweaks: Dict[str, Any]) -> None: + for tweak_name, tweak_value in node_tweaks.items(): + if tweak_name and tweak_value and tweak_name in vertex.params: + vertex.params[tweak_name] = tweak_value + + def process_tweaks(graph_data: Dict[str, Any], tweaks: Dict[str, Dict[str, Any]]) -> Dict[str, Any]: """