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