From d19f95f2ec0126b1c944c060d270fcdee311d187 Mon Sep 17 00:00:00 2001
From: Peter Clement <peter@budibase.com>
Date: Tue, 6 May 2025 11:31:14 +0100
Subject: [PATCH] use api response as a mount check

---
 .../src/pages/builder/portal/settings/ai/AISettings.spec.js   | 1 +
 .../builder/src/pages/builder/portal/settings/ai/index.svelte | 4 +---
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/packages/builder/src/pages/builder/portal/settings/ai/AISettings.spec.js b/packages/builder/src/pages/builder/portal/settings/ai/AISettings.spec.js
index e5bf15d947..4b3a8404c8 100644
--- a/packages/builder/src/pages/builder/portal/settings/ai/AISettings.spec.js
+++ b/packages/builder/src/pages/builder/portal/settings/ai/AISettings.spec.js
@@ -85,6 +85,7 @@ describe("AISettings", () => {
     })
 
     it("should display the 'Enable BB AI' button", async () => {
+      API.getConfig.mockResolvedValueOnce({ config: {} })
       setupDOM()
       await waitFor(() => {
         const enableButton = instance.getByText("Enable BB AI")
diff --git a/packages/builder/src/pages/builder/portal/settings/ai/index.svelte b/packages/builder/src/pages/builder/portal/settings/ai/index.svelte
index cc881bb101..efa2642c43 100644
--- a/packages/builder/src/pages/builder/portal/settings/ai/index.svelte
+++ b/packages/builder/src/pages/builder/portal/settings/ai/index.svelte
@@ -35,7 +35,6 @@
     },
   })
 
-  let mounted = false
   let aiConfig: AIConfig
   let configModal: { show: () => void; hide: () => void }
   let portalModal: { show: () => void; hide: () => void }
@@ -154,14 +153,13 @@
       aiConfig = (await API.getConfig(ConfigType.AI)) as AIConfig
       const licenseKeyResponse = await API.getLicenseKey()
       hasLicenseKey = licenseKeyResponse?.licenseKey
-      mounted = true
     } catch {
       notifications.error("Error fetching AI settings")
     }
   })
 </script>
 
-{#if mounted}
+{#if aiConfig}
   <Layout noPadding>
     <Layout gap="XS" noPadding>
       <div class="header">