From 7e3b9f05bbcdec85c7db2cd519f57aecc0affcc4 Mon Sep 17 00:00:00 2001
From: Diego Imbert <70353967+diegoimbert@users.noreply.github.com>
Date: Wed, 18 Feb 2026 15:28:35 +0100
Subject: [PATCH] Fix preview scripts with wrong tag (#7999)

---
 frontend/src/lib/components/Editor.svelte                 | 8 ++++++--
 frontend/src/lib/components/ScriptEditor.svelte           | 1 +
 .../apps/components/display/dbtable/metadata.ts           | 2 ++
 .../components/flows/content/FlowModuleComponent.svelte   | 2 ++
 4 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/frontend/src/lib/components/Editor.svelte b/frontend/src/lib/components/Editor.svelte
index e7b4a2f624..46a30ffc16 100644
--- a/frontend/src/lib/components/Editor.svelte
+++ b/frontend/src/lib/components/Editor.svelte
@@ -142,6 +142,8 @@
 		rawAppRunnableKey?: string | undefined
 		// Used to provide typed queries in TypeScript when detecting assets
 		preparedAssetsSqlQueries?: InferAssetsSqlQueryDetails[] | undefined
+		// To execute preview scripts with the right worker group
+		customTag?: string
 	}
 
 	let {
@@ -171,7 +173,8 @@
 		moduleId = undefined,
 		enablePreprocessorSnippet = false,
 		rawAppRunnableKey = undefined,
-		preparedAssetsSqlQueries
+		preparedAssetsSqlQueries,
+		customTag
 	}: Props = $props()
 
 	$effect.pre(() => {
@@ -595,7 +598,8 @@
 					lang === 'graphql' ? 'graphql' : (scriptLang ?? ''),
 					resourcePath,
 					$workspaceStore,
-					(e) => console.error(`error getting ${lang} (${scriptLang}) db schema`, e)
+					(e) => console.error(`error getting ${lang} (${scriptLang}) db schema`, e),
+					{ customTag }
 				)
 			}
 			dbSchema = $dbSchemas[resourcePath]
diff --git a/frontend/src/lib/components/ScriptEditor.svelte b/frontend/src/lib/components/ScriptEditor.svelte
index 9257484ed6..4659abc0ac 100644
--- a/frontend/src/lib/components/ScriptEditor.svelte
+++ b/frontend/src/lib/components/ScriptEditor.svelte
@@ -1415,6 +1415,7 @@
 			{args}
 			{enablePreprocessorSnippet}
 			preparedAssetsSqlQueries={preparedSqlQueries.current}
+			customTag={tag}
 		/>
 		<DiffEditor
 			className="h-full"
diff --git a/frontend/src/lib/components/apps/components/display/dbtable/metadata.ts b/frontend/src/lib/components/apps/components/display/dbtable/metadata.ts
index e84fef4e85..840a7a71d9 100644
--- a/frontend/src/lib/components/apps/components/display/dbtable/metadata.ts
+++ b/frontend/src/lib/components/apps/components/display/dbtable/metadata.ts
@@ -402,6 +402,7 @@ export async function getDbSchemas(
 	errorCallback: (message: string) => void,
 	options: {
 		useLegacyScripts?: boolean // To avoid breaking app policies
+		customTag?: string
 	} = {}
 ): Promise<DBSchema | undefined> {
 	let scripts = options.useLegacyScripts ? legacyScripts : scriptsV2
@@ -416,6 +417,7 @@ export async function getDbSchemas(
 			requestBody: {
 				language: sqlScript.lang as Preview['language'],
 				content: sqlScript.code,
+				tag: options.customTag,
 				args: {
 					[sqlScript.argName]: resourcePath.startsWith('datatable://')
 						? resourcePath
diff --git a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
index 2d5b9d5182..a0f3ede833 100644
--- a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
@@ -895,6 +895,7 @@
 																key={`flow-inline-${$workspaceStore}-${$pathStore}-${flowModule.id}`}
 																moduleId={flowModule.id}
 																preparedAssetsSqlQueries={preparedSqlQueries.current}
+																customTag={flowModule.value.tag}
 															/>
 														</div>
 													</Pane>
@@ -957,6 +958,7 @@
 														key={`flow-inline-${$workspaceStore}-${$pathStore}-${flowModule.id}`}
 														moduleId={flowModule.id}
 														preparedAssetsSqlQueries={preparedSqlQueries.current}
+														customTag={flowModule.value.tag}
 													/>
 												</div>
 											{/if}