Optimize replace_api_key_with_env_var_name

The optimized code replaces chained `.get()` calls with direct dictionary access inside a try/except block (`flow["data"]["nodes"]` instead of `flow.get("data", {}).get("nodes", [])`), which is faster when the keys exist (the common case) because CPython's exception handling is cheaper than repeated dict lookups and fallback-value creation. Line profiler shows this change reduced the per-node overhead: the optimized version spends ~4.8% of time in the single `template = node["data"]["node"]["template"]` line versus the original's cumulative ~14.1% across three separate `.get()` calls and `isinstance` checks. The optimization also hoists the `isinstance(value, dict)` check before examining `value.get("name")`, avoiding two dict lookups on non-dict values. Trade-off: exceptions now raised for malformed flows, but the early-exit try/except at the top preserves the original's robustness for missing top-level keys.

Author: codeflash-ai[bot]

src/backend/base/langflow/api/utils/core.py

@@ -94,24 +94,29 @@ def _looks_like_variable_name(value: Any) -> bool:
 
 def replace_api_key_with_env_var_name(flow: dict) -> dict:
     """Normalize api_key to a variable name when possible, never export raw keys."""
-    for node in flow.get("data", {}).get("nodes", []):
-        node_data = node.get("data")
-        if not isinstance(node_data, dict):
-            continue
-        node_inner = node_data.get("node")
-        if not isinstance(node_inner, dict):
+    try:
+        nodes = flow["data"]["nodes"]
+    except (KeyError, TypeError):
+        return flow
+
+    for node in nodes:
+        try:
+            template = node["data"]["node"]["template"]
+        except (KeyError, TypeError):
             continue
-        template = node_inner.get("template")
         if not isinstance(template, dict):
             continue
         for value in template.values():
-            if isinstance(value, dict) and value.get("name") == "api_key" and value.get("password"):
+            if not isinstance(value, dict):
+                continue
+            if value.get("name") == "api_key" and value.get("password"):
                 current = value.get("value")
                 if _looks_like_variable_name(current):
                     break  # keep user's custom variable name
                 # raw secret or other string: clear it
                 value["value"] = None
                 break
+
     return flow