From adfb0bd5ff4236b92b3900fc771c2ed69ad50bf8 Mon Sep 17 00:00:00 2001
From: Ruben Fiszel <ruben@windmill.dev>
Date: Mon, 30 Mar 2026 12:36:37 +0000
Subject: [PATCH] add suggestIntegration and suggestScript to flow builder
 customUI (#8615)

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
---
 frontend/src/lib/components/custom_ui.ts               |  2 ++
 .../components/flows/content/FlowInputsQuick.svelte    |  2 +-
 .../components/flows/pickers/PickHubScriptQuick.svelte | 10 +++++++---
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/frontend/src/lib/components/custom_ui.ts b/frontend/src/lib/components/custom_ui.ts
index 536471f74d..5f017ee950 100644
--- a/frontend/src/lib/components/custom_ui.ts
+++ b/frontend/src/lib/components/custom_ui.ts
@@ -42,6 +42,8 @@ export type FlowBuilderWhitelabelCustomUi = {
 	tagLabel?: string
 	aiAgent?: boolean
 	aiSandbox?: boolean
+	suggestIntegration?: boolean
+	suggestScript?: boolean
 }
 
 export type DisplayResultUi = {
diff --git a/frontend/src/lib/components/flows/content/FlowInputsQuick.svelte b/frontend/src/lib/components/flows/content/FlowInputsQuick.svelte
index 0dd5635816..db199c745a 100644
--- a/frontend/src/lib/components/flows/content/FlowInputsQuick.svelte
+++ b/frontend/src/lib/components/flows/content/FlowInputsQuick.svelte
@@ -308,7 +308,7 @@
 						bind:selectedFilter={selected}
 						resourceType
 					/>
-					{#if !selected}
+					{#if !selected && customUi?.suggestIntegration != false}
 						<div class="pl-2 py-1">
 							<a
 								href={`${$hubBaseUrlStore}?suggest_integration=true`}
diff --git a/frontend/src/lib/components/flows/pickers/PickHubScriptQuick.svelte b/frontend/src/lib/components/flows/pickers/PickHubScriptQuick.svelte
index 3d929dead7..0c548d34c9 100644
--- a/frontend/src/lib/components/flows/pickers/PickHubScriptQuick.svelte
+++ b/frontend/src/lib/components/flows/pickers/PickHubScriptQuick.svelte
@@ -36,7 +36,7 @@
 </script>
 
 <script lang="ts">
-	import { createEventDispatcher, untrack } from 'svelte'
+	import { createEventDispatcher, getContext, untrack } from 'svelte'
 	import { Skeleton } from '$lib/components/common'
 	import { classNames, createCache, sendUserToast } from '$lib/utils'
 	import { APP_TO_ICON_COMPONENT } from '$lib/components/icons'
@@ -48,6 +48,9 @@
 	import { get } from 'svelte/store'
 	import Button from '$lib/components/common/button/Button.svelte'
 	import { Alert } from '$lib/components/common'
+	import type { FlowBuilderWhitelabelCustomUi } from '$lib/components/custom_ui'
+
+	let customUi: undefined | FlowBuilderWhitelabelCustomUi = getContext('customUi')
 
 	let hubNotAvailable = $state(false)
 
@@ -184,7 +187,8 @@
 {:else if hubNotAvailable}
 	<div class="px-3 py-2 mt-2">
 		<Alert type="warning" title="Hub not available" size="xs">
-			Could not connect to the Windmill Hub. If you are in a closed environment, you can disable the Hub in the <a href="/#superadmin-settings?tab=private_hub">instance settings</a>.
+			Could not connect to the Windmill Hub. If you are in a closed environment, you can disable the
+			Hub in the <a href="/#superadmin-settings?tab=private_hub">instance settings</a>.
 		</Alert>
 	</div>
 {:else if loading}
@@ -246,7 +250,7 @@
 		<div class="text-2xs text-tercary font-extralight text-center py-2 px-3 items-center">
 			There are more items than being displayed. Refine your search.
 		</div>
-	{:else}
+	{:else if customUi?.suggestScript != false}
 		<div class="px-2 py-1">
 			<a
 				href={`${$hubBaseUrlStore}?suggest_script=true`}