windmill-components 1.522.0 → 1.531.1

package/package/components/AIAgentLogViewer.svelte

@@ -0,0 +1,123 @@
+<script lang="ts">import { JobService } from '../gen';
+import { workspaceStore } from '../stores';
+import FlowLogViewerWrapper from './FlowLogViewerWrapper.svelte';
+import { z } from 'zod';
+import { onMount } from 'svelte';
+const resultSchema = z.object({
+    messages: z.array(z.object({
+        role: z.string(),
+        content: z.string().optional(),
+        agent_action: z
+            .union([
+            z.object({
+                type: z.literal('tool_call'),
+                job_id: z.string(),
+                module_id: z.string(),
+                function_name: z.string()
+            }),
+            z.object({
+                type: z.literal('message')
+            })
+        ])
+            .optional()
+    }))
+});
+let { tools, agentJob, workspaceId, onToolJobLoaded, storedToolCallJobs } = $props();
+const fakeModuleStates = $state({});
+async function loadMissingJobs(agentActions) {
+    const promises = agentActions.map(async (toolCall, idx) => {
+        if (toolCall.type === 'tool_call') {
+            let job = storedToolCallJobs?.[idx];
+            if (!job || job.type !== 'CompletedJob') {
+                job = await JobService.getJob({
+                    id: toolCall.job_id,
+                    workspace: workspaceId ?? $workspaceStore
+                });
+            }
+            fakeModuleStates[idx.toString()] = {
+                args: job.args,
+                type: job['success'] ? 'Success' : 'Failure',
+                logs: job.logs,
+                result: job['result'],
+                job_id: toolCall.job_id
+            };
+            onToolJobLoaded?.(job, idx);
+        }
+        else {
+            fakeModuleStates[idx.toString()] = {
+                type: 'Success',
+                args: {},
+                logs: '',
+                result: toolCall.content
+            };
+        }
+    });
+    await Promise.all(promises);
+}
+let job = $state(undefined);
+async function loadToolCalls() {
+    let parsedResult = resultSchema.safeParse(agentJob.result);
+    if (!parsedResult.success) {
+        console.error('Invalid result', parsedResult.error);
+        return;
+    }
+    let agentActions = parsedResult.data.messages
+        .map((m) => (m.agent_action?.type === 'message'
+        ? {
+            type: 'message',
+            content: m.content
+        }
+        : m.agent_action?.type === 'tool_call'
+            ? {
+                type: 'tool_call',
+                job_id: m.agent_action.job_id,
+                module_id: m.agent_action.module_id,
+                function_name: m.agent_action.function_name
+            }
+            : undefined))
+        .filter((m) => m !== undefined);
+    await loadMissingJobs(agentActions);
+    job = {
+        ...agentJob,
+        raw_flow: {
+            modules: agentActions
+                .map((toolCall, idx) => {
+                if (toolCall.type === 'message') {
+                    return {
+                        id: idx.toString(),
+                        value: {
+                            type: 'identity'
+                        }
+                    };
+                }
+                else {
+                    const module = tools.find((m) => m.summary === toolCall.function_name);
+                    return module
+                        ? {
+                            ...module,
+                            id: idx.toString()
+                        }
+                        : undefined;
+                }
+            })
+                .filter((m) => m !== undefined)
+        }
+    };
+}
+onMount(() => {
+    loadToolCalls();
+});
+</script>
+
+{#if job}
+	<div class="p-2">
+		<FlowLogViewerWrapper
+			{job}
+			localModuleStates={fakeModuleStates}
+			{workspaceId}
+			render={true}
+			onSelectedIteration={async () => {}}
+			mode="aiagent"
+		/>
+	</div>
+{/if}

package/package/components/AIAgentLogViewer.svelte.d.ts

@@ -0,0 +1,13 @@
+import { type CompletedJob, type FlowModule, type Job } from '../gen';
+interface Props {
+    tools: FlowModule[];
+    agentJob: Partial<CompletedJob> & Pick<CompletedJob, 'id'> & {
+        type: 'CompletedJob';
+    };
+    workspaceId?: string | undefined;
+    storedToolCallJobs?: Record<number, Job>;
+    onToolJobLoaded?: (job: Job, idx: number) => void;
+}
+declare const AiAgentLogViewer: import("svelte").Component<Props, {}, "">;
+type AiAgentLogViewer = ReturnType<typeof AiAgentLogViewer>;
+export default AiAgentLogViewer;

package/package/components/AppConnectInner.svelte

@@ -155,20 +155,20 @@ export async function loadResourceTypes() {
     });
     connectsManual = availableRts
         .filter((x) => connectAndManual.includes(x) || !Object.keys(connects ?? {}).includes(x))
-        .map((x) => [
-        x,
-        apiTokenApps[x] ?? {
+        .map((x) => ({
+        key: x,
+        ...(apiTokenApps[x] ?? {
             instructions: '',
             img: undefined,
             linkedSecret: undefined
-        }
-    ])
-        .sort((a, b) => a[0].localeCompare(b[0]));
+        })
+    }))
+        .sort((a, b) => a.key.localeCompare(b.key));
     const filteredNativeLanguages = filteredConnectsManual?.filter((o) => nativeLanguagesCategory?.includes(o[0]) ?? false);
     try {
         filteredConnectsManual = [
             ...(filteredNativeLanguages ?? []),
-            ...(filteredConnectsManual ?? []).filter(([key, _]) => !nativeLanguagesCategory.includes(key))
+            ...(filteredConnectsManual ?? []).filter(({ key }) => !nativeLanguagesCategory.includes(key))
         ];
     }
     catch (e) { }
@@ -443,7 +443,7 @@ let editScopes = $state(false);
 		{filter}
 		items={connectsManual}
 		bind:filteredItems={filteredConnectsManual}
-		f={(x) => x[0]}
+		f={(x) => x.key}
 	/>
 	{#if step == 1}
 		<div class="w-12/12 pb-2 flex flex-row my-1 gap-1">
@@ -507,7 +507,7 @@ let editScopes = $state(false);
 
 		<div class="grid sm:grid-cols-2 md:grid-cols-3 gap-x-2 gap-y-1 items-center mb-2">
 			{#if filteredConnectsManual}
-				{#each filteredConnectsManual as [key, _]}
+				{#each filteredConnectsManual as { key }}
 					{#if nativeLanguagesCategory.includes(key)}
 						<Button
 							size="sm"
@@ -530,7 +530,7 @@ let editScopes = $state(false);
 		<div class="mt-8 mb-4"></div>
 		<div class="grid sm:grid-cols-2 md:grid-cols-3 gap-x-2 gap-y-1 items-center mb-2">
 			{#if filteredConnectsManual}
-				{#each filteredConnectsManual as [key, _]}
+				{#each filteredConnectsManual as { key }}
 					{#if !nativeLanguagesCategory.includes(key)}
 						<!-- Exclude specific items -->
 						<Button

package/package/components/AutoscalingConfigEditor.svelte

@@ -9,9 +9,31 @@ import { createEventDispatcher } from 'svelte';
 import Label from './Label.svelte';
 import MultiSelect from './select/MultiSelect.svelte';
 import { safeSelectItems } from './select/utils.svelte';
+import { ConfigService } from '../gen';
 let { config = $bindable(), worker_tags } = $props();
 const dispatch = createEventDispatcher();
 let test_input = $state(3);
+let healthCheckLoading = $state(false);
+let healthCheckResult = $state(null);
+async function checkKubernetesHealth() {
+    if (!config?.integration || config.integration.type !== 'kubernetes')
+        return;
+    healthCheckLoading = true;
+    healthCheckResult = null;
+    try {
+        await ConfigService.nativeKubernetesAutoscalingHealthcheck();
+        healthCheckResult = { success: true };
+    }
+    catch (error) {
+        healthCheckResult = {
+            success: false,
+            error: error.body || error.message || 'Unknown error'
+        };
+    }
+    finally {
+        healthCheckLoading = false;
+    }
+}
 </script>
 
 <div class="flex flex-row gap-16 pt-2">
@@ -253,7 +275,7 @@ let test_input = $state(3);
 					/>
 					<ToggleButton disabled value="ecs" label="ECS (soon)" {item} />
 					<ToggleButton disabled value="nomad" label="Nomad (soon)" {item} />
-					<ToggleButton disabled value="kubernetes" label="Kubernetes (soon)" {item} />
+					<ToggleButton value="kubernetes" label="Kubernetes" {item} />
 				{/snippet}
 			</ToggleButtonGroup>
 
@@ -315,6 +337,58 @@ let test_input = $state(3);
 					</div>
 				</div>
 			{/if}
+
+			{#if config.integration.type === 'kubernetes'}
+				<div class="text-sm text-secondary mb-3">
+					Kubernetes configuration is automatically inferred from the cluster environment. 
+					The worker group name and namespace will be detected automatically.
+				</div>
+
+				<div class="flex flex-col gap-3 mt-4">
+					<div class="flex items-center gap-2">
+						<Button 
+							color="blue" 
+							size="xs" 
+							variant="contained"
+							startIcon={{ icon: ExternalLink }}
+							href="https://windmill.dev/docs/core_concepts/autoscaling#kubernetes"
+							target="_blank"
+						>
+							Setup Guide (Roles & Bindings)
+						</Button>
+						<Button 
+							color="light" 
+							size="xs" 
+							variant="contained"
+							onclick={checkKubernetesHealth}
+							disabled={healthCheckLoading}
+						>
+							{healthCheckLoading ? 'Checking...' : 'Check Health'}
+						</Button>
+					</div>
+					
+					{#if healthCheckResult !== null}
+						<div class="p-2 rounded-md text-sm {healthCheckResult.success ? 'bg-green-100 dark:bg-green-900/30 text-green-700 dark:text-green-300' : 'bg-red-50 dark:bg-red-900/20 text-red-600 dark:text-red-400'}">
+							{#if healthCheckResult.success}
+								Kubernetes autoscaling is healthy
+							{:else}
+								{healthCheckResult.error}
+								{#if healthCheckResult.error?.includes('permissions') || healthCheckResult.error?.includes('role')}
+									<br><small>Please follow the setup guide above to configure proper RBAC permissions.</small>
+								{/if}
+							{/if}
+						</div>
+					{/if}
+
+					<div class="flex flex-row gap-2">
+						<Button color="light" size="xs" variant="contained">Test scaling</Button>
+						<div class="flex text-xs flex-row gap-2 items-center">
+							<input class="!w-16" type="number" bind:value={test_input} />
+							workers
+						</div>
+					</div>
+				</div>
+			{/if}
 		{:else}
 			<ToggleButtonGroup selected={'script'} disabled class="mb-4 mt-2">
 				{#snippet children({ item })}
@@ -322,7 +396,7 @@ let test_input = $state(3);
 					<ToggleButton value="script" label="Custom script" {item} />
 					<ToggleButton value="ecs" label="ECS (soon)" {item} />
 					<ToggleButton value="nomad" label="Nomad (soon)" {item} />
-					<ToggleButton value="kubernetes" label="Kubernetes (soon)" {item} />
+					<ToggleButton value="kubernetes" label="Kubernetes" {item} />
 				{/snippet}
 			</ToggleButtonGroup>
 

package/package/components/Dev.svelte

@@ -29,7 +29,6 @@ import { workspaceAIClients } from './copilot/lib';
 import { Triggers } from './triggers/triggers.svelte';
 import { TestSteps } from './flows/testSteps.svelte';
 import { ModulesTestStates } from './modulesTest.svelte';
-import { updateDerivedModuleStatesFromTestJobs } from './flows/utils';
 let flowCopilotContext = {
     shouldUpdatePropertyType: writable({}),
     exprsToSet: writable({}),
@@ -81,7 +80,6 @@ let flowPreviewButtons = $state();
 const flowPreviewContent = $derived(flowPreviewButtons?.getFlowPreviewContent());
 const job = $derived(flowPreviewContent?.getJob());
 let showJobStatus = $state(false);
-let testModuleId = $state(undefined);
 let currentScript = $state(undefined);
 let schema = $state(emptySchema());
 const href = window.location.href;
@@ -371,7 +369,7 @@ async function replaceFlow(lastEdit) {
         console.error('issue setting new flowstore', e);
     }
 }
-const flowStateStore = writable({});
+const flowStateStore = $state({ val: {} });
 const previewArgsStore = $state({ val: {} });
 const scriptEditorDrawer = writable(undefined);
 const moving = writable(undefined);
@@ -381,8 +379,6 @@ const selectedIdStore = writable('settings-metadata');
 const triggersCount = writable(undefined);
 const modulesTestStates = new ModulesTestStates((moduleId) => {
     // Update the derived store with test job states
-    delete $derivedModuleStates[moduleId];
-    testModuleId = moduleId;
     showJobStatus = false;
 });
 const outputPickerOpenFns = $state({});
@@ -448,12 +444,12 @@ async function inferModuleArgs(selectedIdStore) {
                     mod.value.lock = undefined;
                 }
                 mod.value.input_transforms = input_transforms;
-                if (!deepEqual(schema, $flowStateStore[mod.id]?.schema)) {
-                    if (!$flowStateStore[mod.id]) {
-                        $flowStateStore[mod.id] = { schema };
+                if (!deepEqual(schema, flowStateStore.val[mod.id]?.schema)) {
+                    if (!flowStateStore.val[mod.id]) {
+                        flowStateStore.val[mod.id] = { schema };
                     }
                     else {
-                        $flowStateStore[mod.id].schema = schema;
+                        flowStateStore.val[mod.id].schema = schema;
                     }
                     reload++;
                 }
@@ -496,20 +492,10 @@ $effect(() => {
 $effect(() => {
     $selectedIdStore && untrack(() => inferModuleArgs($selectedIdStore));
 });
-const localModuleStates = $derived(flowPreviewContent?.getLocalModuleStates() ?? writable({}));
-const localDurationStatuses = $derived(flowPreviewContent?.getLocalDurationStatuses() ?? writable({}));
-const suspendStatus = $derived(flowPreviewContent?.getSuspendStatus() ?? writable({}));
+let localModuleStates = $state({});
+let suspendStatus = $state({ val: {} });
 // Create a derived store that only shows the module states when showModuleStatus is true
 // this store can also be updated
-let derivedModuleStates = writable({});
-$effect(() => {
-    derivedModuleStates.update((currentStates) => {
-        return showJobStatus ? $localModuleStates : currentStates;
-    });
-});
-$effect(() => {
-    updateDerivedModuleStatesFromTestJobs(testModuleId, modulesTestStates, derivedModuleStates);
-});
 let flowModuleSchemaMap = $state();
 function onJobDone() {
     if (!job) {
@@ -539,10 +525,8 @@ function onJobDone() {
     }
 }
 function resetModulesStates() {
-    derivedModuleStates.set({});
     showJobStatus = false;
 }
-const individualStepTests = $derived(!(showJobStatus && job) && Object.keys($derivedModuleStates).length > 0);
 const flowHasChanged = $derived(flowPreviewContent?.flowHasChanged());
 </script>
 
@@ -681,7 +665,7 @@ const flowHasChanged = $derived(flowPreviewContent?.flowHasChanged());
 						bind:this={flowPreviewButtons}
 						{onJobDone}
 						onRunPreview={() => {
-							localModuleStates.set({})
+							localModuleStates = {}
 							showJobStatus = true
 						}}
 					/>
@@ -696,19 +680,20 @@ const flowHasChanged = $derived(flowPreviewContent?.flowHasChanged());
 								disableTutorials
 								smallErrorHandler={true}
 								disableStaticInputs
-								localModuleStates={derivedModuleStates}
+								{localModuleStates}
 								onTestUpTo={flowPreviewButtons?.testUpTo}
+								testModuleStates={modulesTestStates}
 								isOwner={flowPreviewContent?.getIsOwner?.()}
 								onTestFlow={flowPreviewButtons?.runPreview}
 								isRunning={flowPreviewContent?.getIsRunning?.()}
 								onCancelTestFlow={flowPreviewContent?.cancelTest}
 								onOpenPreview={flowPreviewButtons?.openPreview}
 								onHideJobStatus={resetModulesStates}
-								{individualStepTests}
 								flowJob={job}
 								{showJobStatus}
 								onDelete={(id) => {
-									delete $derivedModuleStates[id]
+									delete localModuleStates[id]
+									delete modulesTestStates.states[id]
 								}}
 								{flowHasChanged}
 							/>
@@ -733,7 +718,6 @@ const flowHasChanged = $derived(flowPreviewContent?.flowHasChanged());
 								onTestFlow={flowPreviewButtons?.runPreview}
 								{job}
 								isOwner={flowPreviewContent?.getIsOwner()}
-								{localDurationStatuses}
 								{suspendStatus}
 								onOpenDetails={flowPreviewButtons?.openPreview}
 							/>

package/package/components/DiffEditor.svelte

@@ -16,6 +16,7 @@ export let defaultOriginal = undefined;
 export let defaultModified = undefined;
 export let readOnly = false;
 export let showButtons = false;
+export let showHistoryButton = true;
 let diffEditor;
 let diffDivEl = null;
 let editorWidth = SIDE_BY_SIDE_MIN_WIDTH;
@@ -103,9 +104,11 @@ const dispatch = createEventDispatcher();
 		<div
 			class="absolute flex flex-row gap-2 bottom-10 left-1/2 z-10 -translate-x-1/2 rounded-md p-1 w-full justify-center"
 		>
-			<Button on:click={() => dispatch('seeHistory')} variant="contained" size="sm"
-				>See changes history</Button
-			>
+			{#if showHistoryButton}
+				<Button on:click={() => dispatch('seeHistory')} variant="contained" size="sm"
+					>See changes history</Button
+				>
+			{/if}
 			<Button on:click={() => dispatch('hideDiffMode')} variant="contained" size="sm" color="red"
 				>Quit diff mode</Button
 			>

package/package/components/DiffEditor.svelte.d.ts

@@ -24,6 +24,7 @@ declare const DiffEditor: $$__sveltets_2_IsomorphicComponent<{
     defaultModified?: string | undefined | undefined;
     readOnly?: boolean | undefined;
     showButtons?: boolean | undefined;
+    showHistoryButton?: boolean | undefined;
     open?: boolean | undefined;
     setupModel?: ((lang: string, original?: string, modified?: string, modifiedLang?: string) => void) | undefined;
     setOriginal?: ((code: string) => void) | undefined;

package/package/components/DisplayResult.svelte

@@ -26,13 +26,14 @@ import { getContext, hasContext, createEventDispatcher, onDestroy } from 'svelte
 import { toJsonStr } from '../utils';
 import { userStore } from '../stores';
 import ResultStreamDisplay from './ResultStreamDisplay.svelte';
+import { twMerge } from 'tailwind-merge';
 const IMG_MAX_SIZE = 10000000;
 const TABLE_MAX_SIZE = 5000000;
 const DISPLAY_MAX_SIZE = 100000;
 const dispatch = createEventDispatcher();
 let resultKind = $state();
 let hasBigInt = $state(false);
-let { result, requireHtmlApproval = false, filename = undefined, disableExpand = false, jobId = undefined, workspaceId = undefined, hideAsJson = false, noControls = false, drawerOpen = $bindable(false), nodeId = undefined, language = undefined, appPath = undefined, customUi = undefined, isTest = true, externalToolbarAvailable = false, forceJson = $bindable(false), result_stream = undefined, fixTableSizingToParent = false, copilot_fix, children, loading = false } = $props();
+let { result, requireHtmlApproval = false, filename = undefined, disableExpand = false, jobId = undefined, workspaceId = undefined, hideAsJson = false, noControls = false, drawerOpen = $bindable(false), nodeId = undefined, language = undefined, appPath = undefined, customUi = undefined, isTest = true, externalToolbarAvailable = false, forceJson = $bindable(false), result_stream = undefined, fixTableSizingToParent = false, copilot_fix, children, loading = false, growVertical = false } = $props();
 let enableHtml = $state(false);
 let s3FileDisplayRawMode = $state(false);
 function isTableRow(result) {
@@ -401,11 +402,11 @@ $effect(() => {
 	<div class="text-red-400">Non displayable object</div>
 {:else}
 	<div
-		class="inline-highlight relative grow flex flex-col h-full {['plain', 'markdown'].includes(
-			resultKind ?? ''
-		)
-			? ''
-			: 'min-h-[160px]'}"
+		class={twMerge(
+			'inline-highlight relative grow flex flex-col',
+			['plain', 'markdown'].includes(resultKind ?? '') ? 'min-h-0' : 'min-h-[160px]',
+			growVertical ? '' : 'h-full'
+		)}
 	>
 		{#if result != undefined && length != undefined && largeObject != undefined}
 			<div class="flex justify-between items-center w-full">
@@ -460,7 +461,8 @@ $effect(() => {
 			</div>
 			<div class="grow relative">
 				{#if !forceJson && resultKind === 'table-col'}
-					{@const data = 'table-col' in result ? result['table-col'] : result}
+					{@const data =
+						typeof result === 'object' && 'table-col' in result ? result['table-col'] : result}
 					<AutoDataTable
 						class={fixTableSizingToParent
 							? 'absolute inset-0 [&>div]:h-full [&>div]:min-h-[10rem]'
@@ -468,7 +470,8 @@ $effect(() => {
 						objects={objectOfArraysToObjects(data)}
 					/>
 				{:else if !forceJson && resultKind === 'table-row'}
-					{@const data = 'table-row' in result ? result['table-row'] : result}
+					{@const data =
+						typeof result === 'object' && 'table-row' in result ? result['table-row'] : result}
 					<AutoDataTable
 						class={fixTableSizingToParent
 							? 'absolute inset-0 [&>div]:h-full [&>div]:min-h-[10rem]'
@@ -476,7 +479,10 @@ $effect(() => {
 						objects={arrayOfRowsToObjects(data)}
 					/>
 				{:else if !forceJson && resultKind === 'table-row-object'}
-					{@const data = 'table-row-object' in result ? result['table-row-object'] : result}
+					{@const data =
+						typeof result === 'object' && 'table-row-object' in result
+							? result['table-row-object']
+							: result}
 					<AutoDataTable
 						class={fixTableSizingToParent
 							? 'absolute inset-0 [&>div]:h-full [&>div]:min-h-[10rem]'
@@ -781,7 +787,7 @@ $effect(() => {
 										>
 									</button>
 								{:else if !s3object?.disable_download}
-									<FileDownload {s3object} />
+									<FileDownload {workspaceId} {s3object} {appPath} />
 								{:else}
 									<div class="flex text-secondary pt-2">{s3object?.s3} (download disabled)</div>
 								{/if}

package/package/components/DisplayResult.svelte.d.ts

@@ -22,6 +22,7 @@ interface Props {
     fixTableSizingToParent?: boolean;
     copilot_fix?: import('svelte').Snippet;
     children?: import('svelte').Snippet;
+    growVertical?: boolean;
 }
 interface $$__sveltets_2_IsomorphicComponent<Props extends Record<string, any> = any, Events extends Record<string, any> = any, Slots extends Record<string, any> = any, Exports = {}, Bindings = string> {
     new (options: import('svelte').ComponentConstructorOptions<Props>): import('svelte').SvelteComponent<Props, Events, Slots> & {

package/package/components/EditableSchemaForm.svelte

@@ -93,6 +93,7 @@ function alignOrderWithProperties(schema) {
 function onSchemaChange() {
     let editSchema = false;
     if (alignOrderWithProperties(schema)) {
+        console.log('alignOrderWithProperties', JSON.stringify(schema, null, 2));
         editSchema = true;
     }
     let lkeys = schema?.order ?? Object.keys(schema?.properties ?? {});
@@ -585,7 +586,8 @@ function updateDynSelectCode(functionName, lang = 'bun') {
 																										property_1: {
 																											type: 'string'
 																										}
-																									}
+																									},
+																									order: ['property_1']
 																								},
 																								{
 																									title: 'Option 2',
@@ -598,7 +600,8 @@ function updateDynSelectCode(functionName, lang = 'bun') {
 																										property_2: {
 																											type: 'string'
 																										}
-																									}
+																									},
+																									order: ['property_2']
 																								}
 																							]
 																						}

package/package/components/Editor.svelte

@@ -83,7 +83,7 @@ import { MonacoLanguageClient } from 'monaco-languageclient';
 import { toSocket, WebSocketMessageReader, WebSocketMessageWriter } from 'vscode-ws-jsonrpc';
 import { CloseAction, ErrorAction, RequestType } from 'vscode-languageclient';
 import { MonacoBinding } from 'y-monaco';
-import { dbSchemas, copilotInfo, codeCompletionSessionEnabled, lspTokenStore, formatOnSave, vimMode } from '../stores';
+import { dbSchemas, copilotInfo, codeCompletionSessionEnabled, lspTokenStore, formatOnSave, vimMode, relativeLineNumbers } from '../stores';
 import { editorConfig, updateOptions } from '../editorUtils';
 import { createHash as randomHash } from '../editorLangUtils';
 import { workspaceStore } from '../stores';
@@ -107,9 +107,10 @@ import AIChatInlineWidget from './copilot/chat/AIChatInlineWidget.svelte';
 import { writable } from 'svelte/store';
 import { formatResourceTypes } from './copilot/chat/script/core';
 import FakeMonacoPlaceHolder from './FakeMonacoPlaceHolder.svelte';
-import { editorPositionMap } from '../utils';
+import { editorPositionMap, readFieldsRecursively } from '../utils';
 import { extToLang, langToExt } from '../editorLangUtils';
 import { aiChatManager } from './copilot/chat/AIChatManager.svelte';
+import { getDbSchemas } from './apps/components/display/dbtable/utils';
 // import EditorTheme from './EditorTheme.svelte'
 let divEl = $state(null);
 let editor = $state(null);
@@ -405,10 +406,17 @@ function addSqlTypeCompletions() {
     });
 }
 let sqlSchemaCompletor = undefined;
-function updateSchema() {
+async function updateSchema() {
     const newSchemaRes = lang === 'graphql' ? args?.api : args?.database;
     if (typeof newSchemaRes === 'string') {
-        dbSchema = $dbSchemas[newSchemaRes.replace('$res:', '')];
+        const resourcePath = newSchemaRes.replace('$res:', '');
+        dbSchema = $dbSchemas[resourcePath];
+        if (lang === 'graphql' && dbSchema === undefined) {
+            await getDbSchemas(lang, resourcePath, $workspaceStore, $dbSchemas, (e) => {
+                console.error('error getting graphql db schema', e);
+            });
+            dbSchema = $dbSchemas[resourcePath];
+        }
     }
     else {
         dbSchema = undefined;
@@ -425,8 +433,10 @@ function addDBSchemaCompletions() {
     if (!schemaLang || !schema) {
         return;
     }
+    console.log('adding db schema completions', schemaLang);
     if (schemaLang === 'graphql') {
         graphqlService ||= initializeMode();
+        console.log('setting schema config', schema);
         graphqlService?.setSchemaConfig([
             {
                 uri: 'my-schema.graphql',
@@ -519,8 +529,8 @@ let aiChatEditorHandler = $state(undefined);
 let showInlineAIChat = $state(false);
 let inlineAIChatSelection = $state(null);
 let selectedCode = $state('');
-export function reviewAndApplyCode(code) {
-    aiChatEditorHandler?.reviewAndApply(code);
+export function reviewAndApplyCode(code, applyAll = false) {
+    aiChatEditorHandler?.reviewAndApply(code, applyAll);
 }
 function addChatHandler(editor) {
     try {
@@ -990,7 +1000,7 @@ async function loadMonaco() {
     onFileChanges();
     try {
         editor = meditor.create(divEl, {
-            ...editorConfig(code ?? '', lang, automaticLayout, fixedOverflowWidgets),
+            ...editorConfig(code ?? '', lang, automaticLayout, fixedOverflowWidgets, $relativeLineNumbers),
             model,
             fontSize: !small ? 14 : 12,
             lineNumbersMinChars,
@@ -1313,9 +1323,12 @@ $effect(() => {
         : sqlTypeCompletor?.dispose();
 });
 $effect(() => {
-    lang && args && $dbSchemas && untrack(() => updateSchema());
+    console.log('updating schema', lang, $dbSchemas);
+    readFieldsRecursively(args);
+    lang && $dbSchemas && untrack(() => updateSchema());
 });
 $effect(() => {
+    console.log('updating db schema completions', dbSchema, lang);
     initialized &&
         dbSchema &&
         ['sql', 'graphql'].includes(lang) &&
@@ -1361,6 +1374,11 @@ $effect(() => {
 $effect(() => {
     files && model && untrack(() => onFileChanges());
 });
+$effect(() => {
+    editor?.updateOptions({
+        lineNumbers: $relativeLineNumbers ? 'relative' : 'on'
+    });
+});
 </script>
 
 <svelte:window onkeydown={onKeyDown} />