From eb7cbd29bf48a5587791ef586f642af56a4ab11c Mon Sep 17 00:00:00 2001
From: Ruben Fiszel <ruben@windmill.dev>
Date: Sat, 13 Sep 2025 09:40:59 +0000
Subject: [PATCH] fix: allow custom models in ai agent step

---
 .../src/lib/components/AIProviderPicker.svelte | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/frontend/src/lib/components/AIProviderPicker.svelte b/frontend/src/lib/components/AIProviderPicker.svelte
index 36226c9c34..b32453aa8f 100644
--- a/frontend/src/lib/components/AIProviderPicker.svelte
+++ b/frontend/src/lib/components/AIProviderPicker.svelte
@@ -40,12 +40,19 @@
 	let modelsCache = new Map<AIProvider, string[]>()
 
 	// Reactive items for the Select component
-	let items = $derived(
-		availableModels.map((model) => ({
+	let items = $derived.by(() => {
+		const r = availableModels.map((model) => ({
 			label: model,
 			value: model
 		}))
-	)
+		if (value?.model && !availableModels.find((model) => model === value.model)) {
+			r.push({
+				label: value.model,
+				value: value.model
+			})
+		}
+		return r
+	})
 
 	// Provider options for the toggle button group
 	const providerOptions = Object.entries(AI_PROVIDERS).map(([key, details]) => ({
@@ -187,6 +194,11 @@
 				bind:value={value.model}
 				placeholder="Select model"
 				disabled={disabled || !value?.kind || !resourceValueToPath(value?.resource)}
+				onCreateItem={(r) => {
+					availableModels.push(r)
+					value.model = r
+				}}
+				createText="Press enter to use custom model"
 				{loading}
 				clearable={false}
 				noItemsMsg={'No models available'}