From e807c46b68fdf26ef097f27c02d20dc49b017dd4 Mon Sep 17 00:00:00 2001
From: Dev Aggarwal <devxpy@gmail.com>
Date: Wed, 25 Dec 2024 22:50:10 +0530
Subject: [PATCH] Handle `Workspace.DoesNotExist` to ensure stability

---
 recipes/Functions.py | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/recipes/Functions.py b/recipes/Functions.py
index 794c77728..ca5194d30 100644
--- a/recipes/Functions.py
+++ b/recipes/Functions.py
@@ -138,6 +138,8 @@ def render_variables(self):
             description="Pass custom parameters to your function and access the parent workflow data. "
             "Variables will be passed down as the first argument to your anonymous JS function.",
         )
+
+        options = set(gui.session_state.get("secrets") or [])
         with gui.div(className="d-flex align-items-center gap-3 mb-2"):
             gui.markdown(
                 "###### "
@@ -147,26 +149,25 @@ def render_variables(self):
                 unsafe_allow_html=True,
             )
             try:
+                workspace = self.current_workspace
+            except Workspace.DoesNotExist:
+                pass
+            else:
                 edit_secret_button_with_dialog(
-                    self.current_workspace,
+                    workspace,
                     self.request.user,
                     trigger_label=f"{icons.add} Add",
                     trigger_type="tertiary",
                     trigger_className="p-1 mb-2",
                 )
-            except Workspace.DoesNotExist:
-                pass
-        options = list(
-            set(
-                self.current_workspace.managed_secrets.order_by(
-                    "-created_at"
-                ).values_list("name", flat=True)
-            )
-            | set(gui.session_state.get("secrets", []))
-        )
+                options |= set(
+                    self.current_workspace.managed_secrets.order_by(
+                        "-created_at"
+                    ).values_list("name", flat=True)
+                )
         gui.multiselect(
             label="",
-            options=options,
+            options=list(options),
             key="secrets",
             allow_none=True,
         )