Skip to content

Commit 5b966f3

Browse files
committed
Modified tweak behaviour to be overridable if env variable is set on the GUI.
1 parent aaeda7e commit 5b966f3

2 files changed

Lines changed: 34 additions & 0 deletions

File tree

src/backend/base/langflow/processing/process.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,16 +171,33 @@ def apply_tweaks(node: dict[str, Any], node_tweaks: dict[str, Any]) -> None:
171171
for k, v in tweak_value.items():
172172
k_ = "file_path" if field_type == "file" else k
173173
template_data[tweak_name][k_] = v
174+
# If the user didn't explicitly set load_from_db in the dict,
175+
# we default to False for the override.
176+
if "load_from_db" not in tweak_value and "load_from_db" in template_data[tweak_name]:
177+
template_data[tweak_name]["load_from_db"] = False
174178
else:
175179
key = "file_path" if field_type == "file" else "value"
176180
template_data[tweak_name][key] = tweak_value
181+
if "load_from_db" in template_data[tweak_name]:
182+
template_data[tweak_name]["load_from_db"] = False
177183

178184

179185
def apply_tweaks_on_vertex(vertex: Vertex, node_tweaks: dict[str, Any]) -> None:
180186
for tweak_name, tweak_value in node_tweaks.items():
181187
if tweak_name and tweak_value and tweak_name in vertex.params:
182188
vertex.params[tweak_name] = tweak_value
183189

190+
# Determine if we should load from DB
191+
tweak_load_from_db = False
192+
if isinstance(tweak_value, dict):
193+
tweak_load_from_db = tweak_value.get("load_from_db", False)
194+
195+
if tweak_load_from_db:
196+
if tweak_name not in vertex.load_from_db_fields:
197+
vertex.load_from_db_fields.append(tweak_name)
198+
elif tweak_name in vertex.load_from_db_fields:
199+
vertex.load_from_db_fields.remove(tweak_name)
200+
184201

185202
def process_tweaks(
186203
graph_data: dict[str, Any], tweaks: Tweaks | dict[str, dict[str, Any]], *, stream: bool = False

src/lfx/src/lfx/processing/process.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,16 +196,33 @@ def apply_tweaks(node: dict[str, Any], node_tweaks: dict[str, Any]) -> None:
196196
for k, v in tweak_value.items():
197197
k_ = "file_path" if field_type == "file" else k
198198
template_data[tweak_name][k_] = v
199+
# If the user didn't explicitly set load_from_db in the dict,
200+
# we default to False for the override.
201+
if "load_from_db" not in tweak_value and "load_from_db" in template_data[tweak_name]:
202+
template_data[tweak_name]["load_from_db"] = False
199203
else:
200204
key = "file_path" if field_type == "file" else "value"
201205
template_data[tweak_name][key] = tweak_value
206+
if "load_from_db" in template_data[tweak_name]:
207+
template_data[tweak_name]["load_from_db"] = False
202208

203209

204210
def apply_tweaks_on_vertex(vertex: Vertex, node_tweaks: dict[str, Any]) -> None:
205211
for tweak_name, tweak_value in node_tweaks.items():
206212
if tweak_name and tweak_value and tweak_name in vertex.params:
207213
vertex.params[tweak_name] = tweak_value
208214

215+
# Determine if we should load from DB
216+
tweak_load_from_db = False
217+
if isinstance(tweak_value, dict):
218+
tweak_load_from_db = tweak_value.get("load_from_db", False)
219+
220+
if tweak_load_from_db:
221+
if tweak_name not in vertex.load_from_db_fields:
222+
vertex.load_from_db_fields.append(tweak_name)
223+
elif tweak_name in vertex.load_from_db_fields:
224+
vertex.load_from_db_fields.remove(tweak_name)
225+
209226

210227
def process_tweaks(
211228
graph_data: dict[str, Any], tweaks: Tweaks | dict[str, dict[str, Any]], *, stream: bool = False

0 commit comments

Comments
 (0)