From f4b6c6084e27519eb7589310a6dc603a540160c1 Mon Sep 17 00:00:00 2001
From: Adria Navarro <adria@budibase.com>
Date: Fri, 11 Apr 2025 12:19:18 +0200
Subject: [PATCH] Add flags

---
 .../data/_components/AITableGeneration.svelte    | 16 ++++++++++++----
 packages/types/src/sdk/featureFlag.ts            |  2 ++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/packages/builder/src/pages/builder/app/[application]/data/_components/AITableGeneration.svelte b/packages/builder/src/pages/builder/app/[application]/data/_components/AITableGeneration.svelte
index 7d2fd11c7e..391b82a0a1 100644
--- a/packages/builder/src/pages/builder/app/[application]/data/_components/AITableGeneration.svelte
+++ b/packages/builder/src/pages/builder/app/[application]/data/_components/AITableGeneration.svelte
@@ -1,9 +1,13 @@
 <script lang="ts">
   import { API } from "@/api"
+  import { featureFlag } from "@/helpers"
   import { ActionButton, Input } from "@budibase/bbui"
+  import { FeatureFlag } from "@budibase/types"
 
   let prompt: string
 
+  $: isEnabled = featureFlag.isEnabled(FeatureFlag.AI_TABLE_GENERATION)
+
   async function submitPrompt(message: string) {
     await API.generateTables(message)
   }
@@ -23,17 +27,21 @@
       bind:value={prompt}
       placeholder="Generate data using AI..."
       on:keydown={onInputKeydown}
+      disabled={!isEnabled}
     />
   </div>
 
-  {#each ["Create a table called tickets with title, description, status fields", "Create a table called students with name and address fields"] as prompt}
-    <ActionButton on:click={() => submitPrompt(prompt)}>{prompt}</ActionButton>
-  {/each}
+  {#if isEnabled}
+    {#each ["Create a table called tickets with title, description, status fields", "Create a table called students with name and address fields"] as prompt}
+      <ActionButton on:click={() => submitPrompt(prompt)}>{prompt}</ActionButton
+      >
+    {/each}
+  {/if}
 </div>
 
 <style>
   .ai-generation {
-    max-width: 753px;
+    width: 753px;
     display: grid;
     gap: 10px;
 
diff --git a/packages/types/src/sdk/featureFlag.ts b/packages/types/src/sdk/featureFlag.ts
index 3a8bd2ae7d..83376e1d6a 100644
--- a/packages/types/src/sdk/featureFlag.ts
+++ b/packages/types/src/sdk/featureFlag.ts
@@ -1,6 +1,7 @@
 export enum FeatureFlag {
   USE_ZOD_VALIDATOR = "USE_ZOD_VALIDATOR",
   AI_JS_GENERATION = "AI_JS_GENERATION",
+  AI_TABLE_GENERATION = "AI_TABLE_GENERATION",
 
   // Account-portal
   DIRECT_LOGIN_TO_ACCOUNT_PORTAL = "DIRECT_LOGIN_TO_ACCOUNT_PORTAL",
@@ -9,6 +10,7 @@ export enum FeatureFlag {
 export const FeatureFlagDefaults: Record<FeatureFlag, boolean> = {
   [FeatureFlag.USE_ZOD_VALIDATOR]: false,
   [FeatureFlag.AI_JS_GENERATION]: false,
+  [FeatureFlag.AI_TABLE_GENERATION]: false,
 
   // Account-portal
   [FeatureFlag.DIRECT_LOGIN_TO_ACCOUNT_PORTAL]: false,