windmill-components 1.408.2 → 1.430.3

package/package/components/common/table/AppRow.svelte

@@ -8,7 +8,7 @@ import Button from '../button/Button.svelte';
 import Row from './Row.svelte';
 import DraftBadge from '../../DraftBadge.svelte';
 import Badge from '../badge/Badge.svelte';
-import { ExternalLink, Eye, File, FileJson, FileUp, GitFork, Globe, History, Pen, Share, Trash, Clipboard, Loader2 } from 'lucide-svelte';
+import { ExternalLink, Eye, File, FileJson, FolderOpen, GitFork, ChevronUpSquare, History, Pen, Share, Trash, Clipboard, Loader2 } from 'lucide-svelte';
 import { goto as gotoUrl } from '$app/navigation';
 import { page } from '$app/stores';
 import { DELETE, copyToClipboard } from '../../../utils';
@@ -140,7 +140,7 @@ async function loadAppJson() {
 					},
 					{
 						displayName: 'Move/Rename',
-						icon: FileUp,
+						icon: FolderOpen,
 						action: () => {
 							moveDrawer.openDrawer(path, summary, 'app')
 						},
@@ -151,7 +151,7 @@ async function loadAppJson() {
 						? [
 								{
 									displayName: 'Deploy to staging/prod',
-									icon: Globe,
+									icon: ChevronUpSquare,
 									action: () => {
 										deploymentDrawer.openDrawer(path, 'app')
 									},

package/package/components/common/table/FlowRow.svelte

@@ -13,7 +13,7 @@ import DraftBadge from '../../DraftBadge.svelte';
 import { sendUserToast } from '../../../toast';
 import { DELETE, copyToClipboard, isOwner } from '../../../utils';
 import { isDeployable } from '../../../utils_deployable';
-import { Pen, GitFork, Trash, List, FileUp, Globe, Calendar, Share, Archive, Clipboard, Eye, HistoryIcon } from 'lucide-svelte';
+import { Pen, GitFork, Trash, List, FolderOpen, ChevronUpSquare, Calendar, Share, Archive, Clipboard, Eye, HistoryIcon } from 'lucide-svelte';
 import FlowHistory from '../../flows/FlowHistory.svelte';
 import { getDeployUiSettings } from '../../home/deploy_ui';
 export let flow;
@@ -159,7 +159,7 @@ let flowHistory;
 					},
 					{
 						displayName: 'Move/Rename',
-						icon: FileUp,
+						icon: FolderOpen,
 						action: () => {
 							moveDrawer.openDrawer(path, flow.summary, 'flow')
 						},
@@ -177,7 +177,7 @@ let flowHistory;
 						? [
 								{
 									displayName: 'Deploy to staging/prod',
-									icon: Globe,
+									icon: ChevronUpSquare,
 									action: () => {
 										deploymentDrawer.openDrawer(path, 'flow')
 									},

package/package/components/common/table/RawAppRow.svelte

@@ -10,7 +10,7 @@ import Drawer from '../drawer/Drawer.svelte';
 import DrawerContent from '../drawer/DrawerContent.svelte';
 import FileInput from '../fileInput/FileInput.svelte';
 import { goto } from '../../../navigation';
-import { FileUp, Globe, Pen, Share, Trash } from 'lucide-svelte';
+import { FolderOpen, Globe, Pen, Share, Trash } from 'lucide-svelte';
 import { isDeployable } from '../../../utils_deployable';
 import { getDeployUiSettings } from '../../home/deploy_ui';
 export let app;
@@ -87,7 +87,7 @@ const dispatch = createEventDispatcher();
 				return [
 					{
 						displayName: 'Move/Rename',
-						icon: FileUp,
+						icon: FolderOpen,
 						action: () => {
 							moveDrawer.openDrawer(path, summary, 'raw_app')
 						},

package/package/components/common/table/ScriptRow.svelte

@@ -14,7 +14,7 @@ import { sendUserToast } from '../../../toast';
 import { copyToClipboard, DELETE, isOwner } from '../../../utils';
 import { isDeployable } from '../../../utils_deployable';
 import { LanguageIcon } from '../languageIcons';
-import { Archive, Calendar, Code, Copy, Eye, FileUp, GitFork, List, Pen, Share, Trash, History } from 'lucide-svelte';
+import { Archive, Calendar, Code, Copy, Eye, FolderOpen, ChevronUpSquare, GitFork, List, Pen, Share, Trash, History } from 'lucide-svelte';
 import ScriptVersionHistory from '../../ScriptVersionHistory.svelte';
 import { Drawer, DrawerContent } from '..';
 import NoMainFuncBadge from '../../NoMainFuncBadge.svelte';
@@ -181,7 +181,7 @@ let versionsDrawerOpen = false;
 					},
 					{
 						displayName: 'Move/Rename',
-						icon: FileUp,
+						icon: FolderOpen,
 						action: () => {
 							moveDrawer.openDrawer(script.path, script.summary, 'script')
 						},
@@ -192,7 +192,7 @@ let versionsDrawerOpen = false;
 						? [
 								{
 									displayName: 'Deploy to staging/prod',
-									icon: FileUp,
+									icon: ChevronUpSquare,
 									action: () => {
 										deploymentDrawer.openDrawer(script.path, 'script')
 									},

package/package/components/common/tabs/Tab.svelte

@@ -24,7 +24,7 @@ const fontSizeClasses = {
 
 <button
 	class={twMerge(
-		'border-b-2 py-1 px-2 cursor-pointer transition-all z-10 ease-linear font-normal text-primary',
+		'border-b-2 py-1 px-2 cursor-pointer transition-all z-10 ease-linear font-normal text-tertiary',
 		isSelected
 			? 'wm-tab-active font-main'
 			: 'border-gray-300 dark:border-gray-600 border-opacity-0 hover:border-opacity-100 ',

package/package/components/common/toggleButton-v2/ToggleButtonGroup.svelte

@@ -5,6 +5,7 @@
 import { writable } from 'svelte/store';
 import { TabGroup, TabList } from '@rgossiaux/svelte-headlessui';
 import { twMerge } from 'tailwind-merge';
+export let id = undefined;
 export let selected;
 export let noWFull = false;
 export let disabled = false;
@@ -28,6 +29,7 @@ setContext('ToggleButtonGroup', {
 </script>
 
 <TabGroup
+	{id}
 	class={twMerge(
 		`h-8 flex ${noWFull ? '' : 'w-full'} ${disabled ? 'disabled' : ''}`,
 		$$props.class

package/package/components/common/toggleButton-v2/ToggleButtonGroup.svelte.d.ts

@@ -7,6 +7,7 @@ import { type Writable } from 'svelte/store';
 declare const __propDef: {
     props: {
         [x: string]: any;
+        id?: string | null | undefined;
         selected: any;
         noWFull?: boolean | undefined;
         disabled?: boolean | undefined;

package/package/components/copilot/CodeCompletionStatus.svelte

@@ -6,7 +6,7 @@ function storeSetting() {
 }
 </script>
 
-{#if $copilotInfo.exists_openai_resource_path && $copilotInfo.code_completion_enabled}
+{#if $copilotInfo.exists_ai_resource && $copilotInfo.code_completion_enabled}
 	<Toggle
 		size="xs"
 		bind:checked={$codeCompletionSessionEnabled}

package/package/components/copilot/CronGen.svelte

@@ -6,18 +6,31 @@ import { sendUserToast } from '../../toast';
 import { copilotInfo } from '../../stores';
 import { base } from '../../base';
 export let schedule;
+export let cronVersion;
 let instructions = '';
 let instructionsField = undefined;
 let genLoading = false;
 let abortController = new AbortController();
 $: instructionsField && setTimeout(() => instructionsField?.focus(), 100);
-const SYSTEM = "You are a helpful assitant for creating CRON schedules. The structure is 'second minute hour dayOfMonth month dayOfWeek'. Weekdays are Sunday (1), Monday (2), Tuesday (3), Wednesday (4), Thursday (5), Friday (6), Saturday (7). You only return the CRON string without any wrapping characters. If it is invalid, you will return an error message preceeded by 'ERROR:'.";
+const SYSTEM_V2 = "You are a helpful assistant for creating CRON schedules using both standard and extended Croner patterns. The structure is 'second minute hour dayOfMonth month dayOfWeek'. Supported modifiers: ? (wildcard), L (last day/weekday), # (nth occurrence of a weekday), and W (closest weekday). Weekdays are Sunday (0 or 7), Monday (1), Tuesday (2), Wednesday (3), Thursday (4), Friday (5), Saturday (6). Ensure syntax is valid, including optional seconds and special modifiers. You only return either the CRON string without any leading/closing quotes or an error message prefixed with 'ERROR:'.";
+const SYSTEM_V1 = "You are a helpful assitant for creating CRON schedules. The structure is 'second minute hour dayOfMonth month dayOfWeek'. Weekdays are Sunday (1), Monday (2), Tuesday (3), Wednesday (4), Thursday (5), Friday (6), Saturday (7). You only return the CRON string without any wrapping characters. If it is invalid, you will return an error message preceeded by 'ERROR:'.";
+$: updateSystemPrompt(cronVersion);
+function updateSystemPrompt(version) {
+    if (version === 'v2') {
+        SYSTEM = SYSTEM_V2;
+    }
+    else {
+        SYSTEM = SYSTEM_V1;
+    }
+}
+let SYSTEM = SYSTEM_V2;
 const USER = 'CRON schedule instructions: {instructions}';
 async function generateCron() {
     genLoading = true;
     abortController = new AbortController();
+    const aiProvider = $copilotInfo.ai_provider;
     try {
-        const response = await getNonStreamingCompletion([
+        const messages = [
             {
                 role: 'system',
                 content: SYSTEM
@@ -26,7 +39,8 @@ async function generateCron() {
                 role: 'user',
                 content: USER.replace('{instructions}', instructions)
             }
-        ], abortController);
+        ];
+        const response = await getNonStreamingCompletion(messages, abortController, aiProvider);
         if (response.startsWith('ERROR:')) {
             throw response.replace('ERROR:', '').trim();
         }
@@ -62,7 +76,7 @@ async function generateCron() {
 			on:click={genLoading ? () => abortController?.abort() : undefined}
 		/>
 	</svelte:fragment>
-	{#if $copilotInfo.exists_openai_resource_path}
+	{#if $copilotInfo.exists_ai_resource}
 		<div class="flex w-96">
 			<input
 				bind:this={instructionsField}
@@ -97,7 +111,7 @@ async function generateCron() {
 		<div class="block text-primary">
 			<p class="text-sm"
 				>Enable Windmill AI in the <a
-					href="{base}/workspace_settings?tab=openai"
+					href="{base}/workspace_settings?tab=ai"
 					target="_blank"
 					class="inline-flex flex-row items-center gap-1"
 					>workspace settings <ExternalLink size={16} /></a

package/package/components/copilot/CronGen.svelte.d.ts

@@ -2,6 +2,7 @@ import { SvelteComponent } from "svelte";
 declare const __propDef: {
     props: {
         schedule: string;
+        cronVersion: string;
     };
     events: {
         [evt: string]: CustomEvent<any>;

package/package/components/copilot/FlowCopilotStatus.svelte

@@ -20,7 +20,7 @@ $: copilotStatus.length > 0 ? copilotPopover?.open() : copilotPopover?.close();
 $: copilotStatus && copilotPopover?.refresh();
 </script>
 
-{#if $copilotInfo.exists_openai_resource_path}
+{#if $copilotInfo.exists_ai_resource}
 	<ManualPopover bind:this={copilotPopover}>
 		<Button
 			size="xs"
@@ -126,7 +126,7 @@ $: copilotStatus && copilotPopover?.refresh();
 		<div class="block text-primary">
 			<p class="text-sm"
 				>Enable Windmill AI in the <a
-					href="{base}/workspace_settings?tab=openai"
+					href="{base}/workspace_settings?tab=ai"
 					target="_blank"
 					class="inline-flex flex-row items-center gap-1"
 					>workspace settings <ExternalLink size={16} /></a

package/package/components/copilot/IteratorGen.svelte

@@ -52,12 +52,13 @@ ${YAML.stringify(availableData)}</available>
 Reply with the most probable answer, do not explain or discuss.
 Use javascript object dot notation to access the properties.
 Only output the expression, do not explain or discuss.`;
+        const aiProvider = $copilotInfo.ai_provider;
         generatedContent = await getNonStreamingCompletion([
             {
                 role: 'user',
                 content: user
             }
-        ], abortController);
+        ], abortController, aiProvider);
     }
     catch (err) {
         if (!abortController.signal.aborted) {
@@ -69,7 +70,7 @@ Only output the expression, do not explain or discuss.`;
     }
 }
 export function onKeyUp(event) {
-    if (!$copilotInfo.exists_openai_resource_path || !$stepInputCompletionEnabled) {
+    if (!$copilotInfo.exists_ai_resource || !$stepInputCompletionEnabled) {
         return;
     }
     if (event.key === 'Tab') {
@@ -100,7 +101,7 @@ function cancelOnOutOfFocus() {
 $: if (!focused) {
     cancelOnOutOfFocus();
 }
-$: if ($copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled && focused) {
+$: if ($copilotInfo.exists_ai_resource && $stepInputCompletionEnabled && focused) {
     automaticGeneration();
 }
 function cancel() {
@@ -111,7 +112,7 @@ $: dispatch('showExpr', generatedContent);
 let out = true; // hack to prevent regenerating answer when accepting the answer due to mouseenter on new icon
 </script>
 
-{#if $copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled}
+{#if $copilotInfo.exists_ai_resource && $stepInputCompletionEnabled}
 	<ManualPopover showTooltip={!empty && generatedContent.length > 0} placement="bottom" class="p-2">
 		<Button
 			size="xs"

package/package/components/copilot/MetadataGen.svelte

@@ -1,4 +1,4 @@
-<script>import { getCompletion } from './lib';
+<script>import { getCompletion, getResponseFromEvent } from './lib';
 import { isInitialCode } from '../../script_helpers';
 import { Check, Loader2, Wand2 } from 'lucide-svelte';
 import { copilotInfo, metadataCompletionEnabled } from '../../stores';
@@ -98,11 +98,11 @@ async function generateContent(automatic = false) {
                 content: config.user.replace(`{${config.placeholderName}}`, placeholderContent)
             }
         ];
-        const response = await getCompletion(messages, abortController);
+        const aiProvider = $copilotInfo.ai_provider;
+        const response = await getCompletion(messages, abortController, aiProvider);
         generatedContent = '';
         for await (const chunk of response) {
-            const toks = chunk.choices[0]?.delta?.content || '';
-            generatedContent += toks;
+            generatedContent += getResponseFromEvent(chunk, aiProvider);
         }
     }
     catch (err) {
@@ -119,7 +119,7 @@ async function generateContent(automatic = false) {
         loading = false;
     }
 }
-if ($copilotInfo.exists_openai_resource_path &&
+if ($copilotInfo.exists_ai_resource &&
     $metadataCompletionEnabled &&
     generateOnAppear &&
     !content &&
@@ -135,7 +135,7 @@ $: if (content) {
     dispatch('change', { content });
 }
 $: active =
-    $copilotInfo.exists_openai_resource_path &&
+    $copilotInfo.exists_ai_resource &&
         $metadataCompletionEnabled &&
         !content &&
         (loading || focused || !!generatedContent) &&
@@ -166,7 +166,7 @@ onDestroy(() => {
 	class={twMerge('relative', $$props.class)}
 	bind:clientWidth={width}
 	on:keydown={(event) => {
-		if (!$copilotInfo.exists_openai_resource_path || !$metadataCompletionEnabled) {
+		if (!$copilotInfo.exists_ai_resource || !$metadataCompletionEnabled) {
 			return
 		}
 		if (event.key === 'Tab') {
@@ -239,7 +239,7 @@ onDestroy(() => {
 	</div>
 	{#if elementType === 'textarea'}
 		<div>
-			<div class="flex flex-row-reverse text-2xs text-tertiary -mt-4">GH Markdown</div>
+			<div class="flex flex-row-reverse !text-3xs text-tertiary -mt-4">GH Markdown</div>
 			<textarea
 				bind:this={el}
 				bind:value={content}

package/package/components/copilot/PredicateGen.svelte

@@ -48,12 +48,13 @@ Here's a summary of the available data:
 ${YAML.stringify(availableData)}</available>
 If the branching is made inside a for-loop, the iterator value is accessible as flow_input.iter.value
 Only return the expression without any wrapper. Do not explain or discuss.`;
+        const aiProvider = $copilotInfo.ai_provider;
         const result = await getNonStreamingCompletion([
             {
                 role: 'user',
                 content: user
             }
-        ], abortController);
+        ], abortController, aiProvider);
         dispatch('setExpr', result);
         dispatch('updateSummary', instructions);
     }
@@ -68,7 +69,7 @@ Only return the expression without any wrapper. Do not explain or discuss.`;
 }
 </script>
 
-{#if $copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled}
+{#if $copilotInfo.exists_ai_resource && $stepInputCompletionEnabled}
 	<Popup
 		floatingConfig={{ strategy: 'absolute', placement: 'bottom-end' }}
 		containerClasses="border rounded-lg shadow-lg p-4 bg-surface"

package/package/components/copilot/RegexGen.svelte

@@ -20,6 +20,7 @@ async function onGenerate() {
     savePrompt();
     genLoading = true;
     abortController = new AbortController();
+    const aiProvider = $copilotInfo.ai_provider;
     try {
         const res = await getNonStreamingCompletion([
             {
@@ -30,7 +31,7 @@ async function onGenerate() {
                 role: 'user',
                 content: funcDesc
             }
-        ], abortController);
+        ], abortController, aiProvider);
         dispatch('gen', { res: res, prompt: funcDesc });
         funcDesc = '';
     }
@@ -105,7 +106,7 @@ function clearPromptHistory() {
 		/>
 	</svelte:fragment>
 	<div class="block text-primary z-[2000]">
-		{#if $copilotInfo.exists_openai_resource_path}
+		{#if $copilotInfo.exists_ai_resource}
 			<div class="flex flex-col gap-4">
 				<div class="flex w-96">
 					<input
@@ -158,7 +159,7 @@ function clearPromptHistory() {
 		{:else}
 			<p class="text-sm">
 				Enable Windmill AI in the <a
-					href="{base}/workspace_settings?tab=openai"
+					href="{base}/workspace_settings?tab=ai"
 					target="_blank"
 					class="inline-flex flex-row items-center gap-1"
 				>

package/package/components/copilot/ScriptFix.svelte

@@ -32,6 +32,7 @@ async function onFix(closePopup) {
     try {
         genLoading = true;
         abortController = new AbortController();
+        const aiProvider = $copilotInfo.ai_provider;
         await copilot({
             language: lang,
             code: editor?.getCode() || '',
@@ -39,7 +40,7 @@ async function onFix(closePopup) {
             dbSchema: dbSchema,
             type: 'fix',
             workspace: $workspaceStore
-        }, generatedCode, abortController, generatedExplanation);
+        }, generatedCode, abortController, aiProvider, generatedExplanation);
         setupDiff();
         diffEditor?.setModified($generatedCode);
         await sleep(500);
@@ -167,12 +168,12 @@ $: updateSchema(lang, args);
 					</Button>
 				</svelte:fragment>
 				{@const fixAction = (_) => {
-					if ($copilotInfo.exists_openai_resource_path) {
+					if ($copilotInfo.exists_ai_resource) {
 						onFix(() => close(null))
 					}
 				}}
 				<div use:fixAction>
-					{#if $copilotInfo.exists_openai_resource_path}
+					{#if $copilotInfo.exists_ai_resource}
 						<div class="w-[42rem] min-h-[3rem] max-h-[34rem] overflow-y-scroll">
 							{#if $generatedCode.length > 0}
 								<div class="overflow-x-scroll">
@@ -192,7 +193,7 @@ $: updateSchema(lang, args);
 							<p class="text-sm"
 								>Enable Windmill AI in the <a
 									class="inline-flex flex-row items-center gap-1"
-									href="{base}/workspace_settings?tab=openai"
+									href="{base}/workspace_settings?tab=ai"
 									target="_blank">workspace settings</a
 								></p
 							></div

package/package/components/copilot/ScriptGen.svelte

@@ -47,6 +47,7 @@ async function onGenerate(closePopup) {
         return;
     }
     savePrompt();
+    const aiProvider = $copilotInfo.ai_provider;
     try {
         genLoading = true;
         blockPopupOpen = true;
@@ -60,7 +61,7 @@ async function onGenerate(closePopup) {
                 dbSchema: dbSchema,
                 type: 'edit',
                 workspace: $workspaceStore
-            }, generatedCode, abortController);
+            }, generatedCode, abortController, aiProvider);
         }
         else {
             await copilot({
@@ -70,7 +71,7 @@ async function onGenerate(closePopup) {
                 dbSchema: dbSchema,
                 type: 'gen',
                 workspace: $workspaceStore
-            }, generatedCode, abortController);
+            }, generatedCode, abortController, aiProvider);
         }
         setupDiff();
         diffEditor?.setModified($generatedCode);
@@ -350,7 +351,7 @@ onDestroy(() => {
 						<LoadingIcon />
 					{/if}
 				</div>
-			{:else if $copilotInfo.exists_openai_resource_path}
+			{:else if $copilotInfo.exists_ai_resource}
 				<div class="flex flex-col gap-4">
 					<div class="flex flex-row justify-between items-center">
 						<ToggleButtonGroup class="w-auto shrink-0" bind:selected={mode}>
@@ -359,7 +360,14 @@ onDestroy(() => {
 						</ToggleButtonGroup>
 
 						<div class="text-[0.6rem] text-secondary opacity-60 flex flex-row items-center gap-0.5">
-							GPT-4o<Bot size={14} />
+							{#if $copilotInfo.ai_provider === 'openai'}
+								GPT-4o
+							{:else if $copilotInfo.ai_provider === 'anthropic'}
+								Claude-3.5
+							{:else}
+								Codestral
+							{/if}
+							<Bot size={14} />
 						</div>
 					</div>
 					<div class="flex w-96 items-start">
@@ -455,7 +463,7 @@ onDestroy(() => {
 			{:else}
 				<p class="text-sm">
 					Enable Windmill AI in the <a
-						href="{base}/workspace_settings?tab=openai"
+						href="{base}/workspace_settings?tab=ai"
 						target="_blank"
 						class="inline-flex flex-row items-center gap-1"
 					>

package/package/components/copilot/StepGen.svelte

@@ -57,7 +57,7 @@ async function getHubCompletions(text) {
     }
 }
 async function onGenerate() {
-    if (!selectedCompletion && !$copilotInfo.exists_openai_resource_path) {
+    if (!selectedCompletion && !$copilotInfo.exists_ai_resource) {
         sendUserToast('Windmill AI is not enabled, you can activate it in the workspace settings', true);
         return;
     }

package/package/components/copilot/StepInputGen.svelte

@@ -87,12 +87,13 @@ If none of the available results are appropriate, are already used or are more a
 Reply with the most probable answer, do not explain or discuss.
 Use javascript object dot notation to access the properties.
 Only return the expression without any wrapper.`;
+        const aiProvider = $copilotInfo.ai_provider;
         generatedContent = await getNonStreamingCompletion([
             {
                 role: 'user',
                 content: user
             }
-        ], abortController);
+        ], abortController, aiProvider);
         if (pickableProperties &&
             generatedContent.startsWith('flow_input.') &&
             generatedContent.split('.')[1] &&
@@ -113,7 +114,7 @@ Only return the expression without any wrapper.`;
     }
 }
 export function onKeyUp(event) {
-    if (!$copilotInfo.exists_openai_resource_path || !$stepInputCompletionEnabled) {
+    if (!$copilotInfo.exists_ai_resource || !$stepInputCompletionEnabled) {
         return;
     }
     if (event.key === 'Tab') {
@@ -151,7 +152,7 @@ function cancelOnOutOfFocus() {
 $: if (!focused) {
     cancelOnOutOfFocus();
 }
-$: if ($copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled && focused) {
+$: if ($copilotInfo.exists_ai_resource && $stepInputCompletionEnabled && focused) {
     automaticGeneration();
 }
 $: dispatch('showExpr', generatedContent);
@@ -160,7 +161,7 @@ let out = true; // hack to prevent regenerating answer when accepting the answer
 let openInputsModal = false;
 </script>
 
-{#if $copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled}
+{#if $copilotInfo.exists_ai_resource && $stepInputCompletionEnabled}
 	<FlowCopilotInputsModal
 		on:confirmed={async () => {
 			createFlowInput()

package/package/components/copilot/StepInputsGen.svelte

@@ -65,12 +65,13 @@ Your answer has to be in the following format (one line per input):
 input_name1: expression1
 input_name2: expression2
 ...`;
+        const aiProvider = $copilotInfo.ai_provider;
         generatedContent = await getNonStreamingCompletion([
             {
                 role: 'user',
                 content: user
             }
-        ], abortController);
+        ], abortController, aiProvider);
         parsedInputs = generatedContent.split('\n').map((x) => x.split(': '));
         const exprs = {};
         newFlowInputs = [];
@@ -140,7 +141,7 @@ let openInputsModal = false;
 </script>
 
 <div class="flex flex-row justify-end">
-	{#if $copilotInfo.exists_openai_resource_path && $stepInputCompletionEnabled}
+	{#if $copilotInfo.exists_ai_resource && $stepInputCompletionEnabled}
 		<FlowCopilotInputsModal
 			on:confirmed={async () => {
 				createFlowInputs()
@@ -208,10 +209,10 @@ let openInputsModal = false;
 				</Button>
 			</svelte:fragment>
 			<p class="text-sm">
-				{#if !$copilotInfo.exists_openai_resource_path}
+				{#if !$copilotInfo.exists_ai_resource}
 					Enable Windmill AI in the{' '}
 					<a
-						href="{base}/workspace_settings?tab=openai"
+						href="{base}/workspace_settings?tab=ai"
 						target="_blank"
 						class="inline-flex flex-row items-center gap-1"
 					>

package/package/components/copilot/{TestOpenaiKey.svelte → TestAiKey.svelte}

@@ -3,6 +3,7 @@ import Button from '../common/button/Button.svelte';
 import { testKey } from './lib';
 export let disabled = false;
 export let apiKey = undefined;
+export let aiProvider;
 let loading = false;
 </script>
 
@@ -27,7 +28,8 @@ let loading = false;
 						content: "this is a test, simply reply with 'ok'"
 					}
 				],
-				abortController
+				abortController,
+				aiProvider
 			})
 			sendUserToast('Valid key')
 		} catch (err) {

package/package/components/copilot/TestAiKey.svelte.d.ts

@@ -0,0 +1,19 @@
+import { SvelteComponent } from "svelte";
+import { type AiProviderTypes } from './lib';
+declare const __propDef: {
+    props: {
+        disabled?: boolean | undefined;
+        apiKey?: string | undefined;
+        aiProvider: AiProviderTypes;
+    };
+    events: {
+        [evt: string]: CustomEvent<any>;
+    };
+    slots: {};
+};
+export type TestAiKeyProps = typeof __propDef.props;
+export type TestAiKeyEvents = typeof __propDef.events;
+export type TestAiKeySlots = typeof __propDef.slots;
+export default class TestAiKey extends SvelteComponent<TestAiKeyProps, TestAiKeyEvents, TestAiKeySlots> {
+}
+export {};

package/package/components/copilot/completion.d.ts

@@ -1 +1,2 @@
-export declare function editorCodeCompletion(before: string, after: string, lang: string, abortController: AbortController): Promise<string | undefined>;
+import { type AiProviderTypes } from './lib';
+export declare function editorCodeCompletion(before: string, after: string, lang: string, abortController: AbortController, aiProvider: AiProviderTypes): Promise<string | undefined>;

package/package/components/copilot/completion.js

@@ -66,7 +66,7 @@ const prompt = `\`\`\`{language}
 {before}<completion_tokens>{after}
 \`\`\`
 `;
-export async function editorCodeCompletion(before, after, lang, abortController) {
+export async function editorCodeCompletion(before, after, lang, abortController, aiProvider) {
     codeCompletionLoading.set(true);
     const messages = [
         {
@@ -82,7 +82,7 @@ export async function editorCodeCompletion(before, after, lang, abortController)
         }
     ];
     try {
-        const result = await getNonStreamingCompletion(messages, abortController);
+        const result = await getNonStreamingCompletion(messages, abortController, aiProvider);
         const match = result.match(/```[a-zA-Z]+\n([\s\S]*?)\n```/);
         let completion = match?.[1] || '';
         return completion;