From 16e6aefa10db9c9eac6844fe71ad95aef351598d Mon Sep 17 00:00:00 2001 From: Ibis Prevedello Date: Thu, 9 Mar 2023 14:07:32 -0300 Subject: [PATCH] refac: fix key and nodes --- langflow/backend/langflow_backend/interface/loading.py | 1 + langflow/backend/langflow_backend/utils/util.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/langflow/backend/langflow_backend/interface/loading.py b/langflow/backend/langflow_backend/interface/loading.py index 82ce0106c..5882b898d 100644 --- a/langflow/backend/langflow_backend/interface/loading.py +++ b/langflow/backend/langflow_backend/interface/loading.py @@ -21,6 +21,7 @@ def replace_zero_shot_prompt_with_prompt_template(nodes): ] node["data"] = build_prompt_template(prompt=node["data"], tools=tools) break + return nodes def process_data_graph(data_graph: Dict[str, Any]): diff --git a/langflow/backend/langflow_backend/utils/util.py b/langflow/backend/langflow_backend/utils/util.py index 192b16d5f..17d669426 100644 --- a/langflow/backend/langflow_backend/utils/util.py +++ b/langflow/backend/langflow_backend/utils/util.py @@ -283,7 +283,7 @@ def format_dict(d, name: Optional[str] = None): "temperature", "model_name", ] - or any(text in key for text in ["password", "token", "api", "key"]) + or "api_key" in key ) # Add password field