windmill-components 1.532.0 → 1.537.1

package/package/components/flows/content/FlowInput.svelte

@@ -19,14 +19,14 @@ import CaptureButton from '../../triggers/CaptureButton.svelte';
 import { getFullPath, setNestedProperty, getNestedProperty, schemaFromDiff, computeDiff, applyDiff } from '../../schema/schemaUtils.svelte';
 import SideBarTab from '../../meltComponents/SideBarTab.svelte';
 import CaptureTable from '../../triggers/CaptureTable.svelte';
-import { isObjectTooBig } from '../../../utils';
+import { isObjectTooBig, readFieldsRecursively } from '../../../utils';
 import { refreshStateStore } from '../../../svelte5Utils.svelte';
-let { noEditor, disabled, onTestFlow } = $props();
+import { deepEqual } from 'fast-equals';
+let { noEditor, disabled, onTestFlow, previewOpen } = $props();
 const { flowStore, previewArgs, pathStore, initialPathStore, fakeInitialPath, flowInputEditorState } = getContext('FlowEditorContext');
 let addPropertyV2 = $state(undefined);
 let previewSchema = $state(undefined);
 let payloadData = undefined;
-let previewArguments = $state(previewArgs.val);
 let dropdownItems = $state([]);
 let diff = $state({});
 let editPanelSize = $state($flowInputEditorState?.editPanelSize ?? 0);
@@ -148,7 +148,9 @@ function schemaFromPayload(payloadData) {
 }
 function handleKeydown(event) {
     if ((event.metaKey || event.ctrlKey) && event.key === 'Enter') {
-        runPreview();
+        if (!previewOpen) {
+            runPreview();
+        }
     }
     else if (event.key === 'Enter' && previewSchema && !preventEnter) {
         applySchemaAndArgs();
@@ -158,9 +160,6 @@ function handleKeydown(event) {
     }
 }
 function runPreview() {
-    if (previewArguments) {
-        previewArgs.val = structuredClone($state.snapshot(previewArguments));
-    }
     onTestFlow?.();
 }
 function updatePreviewSchemaAndArgs(payload) {
@@ -198,8 +197,8 @@ async function updatePreviewSchema(newSchema) {
 }
 async function applySchemaAndArgs() {
     flowStore.val.schema = applyDiff(flowStore.val.schema, diff);
-    if (previewArguments) {
-        savedPreviewArgs = structuredClone($state.snapshot(previewArguments));
+    if (previewArgs.val) {
+        savedPreviewArgs = structuredClone($state.snapshot(previewArgs.val));
     }
     updatePreviewSchemaAndArgs(undefined);
     if ($flowInputEditorState) {
@@ -208,11 +207,13 @@ async function applySchemaAndArgs() {
 }
 function updatePreviewArguments(payloadData) {
     if (!payloadData) {
-        previewArguments = savedPreviewArgs;
+        if (savedPreviewArgs) {
+            previewArgs.val = savedPreviewArgs;
+        }
         return;
     }
-    savedPreviewArgs = structuredClone($state.snapshot(previewArguments));
-    previewArguments = structuredClone($state.snapshot(payloadData));
+    savedPreviewArgs = structuredClone($state.snapshot(previewArgs.val));
+    previewArgs.val = structuredClone($state.snapshot(payloadData));
 }
 let tabButtonWidth = 0;
 let connectFirstNode = $state(() => { });
@@ -267,10 +268,17 @@ function handleChange(arg, currentSchema, diffSchema, updateCurrentSchema) {
 }
 function resetArgs() {
     if (!previewSchema) {
-        // previewArguments = undefined
         savedPreviewArgs = undefined;
     }
 }
+$effect(() => {
+    if (!previewArgs && savedPreviewArgs != undefined) {
+        readFieldsRecursively(flowStore.val.schema);
+        untrack(() => {
+            resetArgs();
+        });
+    }
+});
 let historicInputs = $state(undefined);
 let captureTable = $state(undefined);
 let savedInputsPicker = $state(undefined);
@@ -295,9 +303,6 @@ function resetSelected() {
 				bind:this={editableSchemaForm}
 				bind:schema={flowStore.val.schema}
 				isFlowInput
-				on:edit={(e) => {
-					addPropertyV2?.openDrawer(e.detail)
-				}}
 				on:delete={(e) => {
 					addPropertyV2?.handleDeleteArgument([e.detail])
 				}}
@@ -305,7 +310,7 @@ function resetSelected() {
 				displayWebhookWarning
 				editTab={$flowInputEditorState?.selectedTab}
 				{previewSchema}
-				bind:args={previewArguments}
+				bind:args={previewArgs.val}
 				bind:editPanelSize={
 					() => {
 						return editPanelSize
@@ -331,15 +336,13 @@ function resetSelected() {
 					})
 				}}
 				shouldDispatchChanges={true}
-				on:change={() => {
-					previewArguments = previewArguments
+				onChange={() => {
 					if (!previewSchema) {
-						savedPreviewArgs = structuredClone($state.snapshot(previewArguments))
+						let args = $state.snapshot(previewArgs.val)
+						if (!deepEqual(args, savedPreviewArgs)) {
+							savedPreviewArgs = args
+						}
 					}
-					refreshStateStore(flowStore)
-				}}
-				on:schemaChange={() => {
-					resetArgs()
 				}}
 				bind:isValid
 				bind:dynSelectCode
@@ -408,15 +411,9 @@ function resetSelected() {
 						<AddPropertyV2
 							bind:schema={flowStore.val.schema}
 							bind:this={addPropertyV2}
-							on:change={() => {
-								refreshStateStore(flowStore)
-								if (editableSchemaForm) {
-									editableSchemaForm.updateJson()
-								}
-							}}
-							on:addNew={(e) => {
+							onAddNew={(argName) => {
 								handleEditSchema('inputEditor')
-								editableSchemaForm?.openField(e.detail)
+								editableSchemaForm?.openField(argName)
 								refreshStateStore(flowStore)
 							}}
 						>
@@ -493,7 +490,7 @@ function resetSelected() {
 								on:isEditing={(e) => {
 									preventEnter = e.detail
 								}}
-								previewArgs={previewArguments}
+								previewArgs={previewArgs.val}
 								{isValid}
 								limitPayloadSize
 								bind:this={savedInputsPicker}
@@ -516,7 +513,7 @@ function resetSelected() {
 								on:select={(e) => {
 									updatePreviewSchemaAndArgs(e.detail ?? undefined)
 								}}
-								selected={!!previewArguments}
+								selected={!!previewArgs.val}
 								bind:this={jsonInputs}
 							/>
 						</FlowInputEditor>

package/package/components/flows/content/FlowInput.svelte.d.ts

@@ -2,6 +2,7 @@ interface Props {
     noEditor: boolean;
     disabled: boolean;
     onTestFlow?: () => void;
+    previewOpen: 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/flows/content/FlowLoop.svelte

@@ -20,10 +20,17 @@ import FlowModuleSkip from './FlowModuleSkip.svelte';
 import FlowPlugConnect from '../../FlowPlugConnect.svelte';
 import PropPickerWrapper, { CONNECT } from '../propPicker/PropPickerWrapper.svelte';
 import TabsV2 from '../../common/tabs/TabsV2.svelte';
+import { useUiIntent } from '../../copilot/chat/flow/useUiIntent';
 const { previewArgs, flowStateStore, flowStore, currentEditor } = getContext('FlowEditorContext');
 let { mod = $bindable(), parentModule, previousModule, noEditor, enableAi = false } = $props();
 let editor = $state(undefined);
 let selected = $state('early-stop');
+// UI Intent handling for AI tool control
+useUiIntent(`forloopflow-${mod.id}`, {
+    openTab: (tab) => {
+        selected = tab;
+    }
+});
 const { flowPropPickerConfig } = getContext('PropPickerContext');
 flowPropPickerConfig.set(undefined);
 let stepPropPicker = $derived(getStepPropPicker(flowStateStore.val, parentModule, previousModule, mod.id, flowStore.val, previewArgs.val, false));

package/package/components/flows/content/FlowModuleComponent.svelte

@@ -30,7 +30,7 @@ import InputTransformSchemaForm from '../../InputTransformSchemaForm.svelte';
 import FlowModuleMockTransitionMessage from './FlowModuleMockTransitionMessage.svelte';
 import Tooltip from '../../Tooltip.svelte';
 import { SecondsInput } from '../../common';
-import DiffEditor from '../../DiffEditor.svelte';
+import DiffEditor, {} from '../../DiffEditor.svelte';
 import FlowModuleTimeout from './FlowModuleTimeout.svelte';
 import HighlightCode from '../../HighlightCode.svelte';
 import ToggleButtonGroup from '../../common/toggleButton-v2/ToggleButtonGroup.svelte';
@@ -41,18 +41,29 @@ import { enterpriseLicense } from '../../../stores';
 import { isCloudHosted } from '../../../cloud';
 import { loadSchemaFromModule } from '../flowInfers';
 import FlowModuleSkip from './FlowModuleSkip.svelte';
-import { JobService } from '../../../gen';
+import {} from '../../../gen';
 import { workspaceStore } from '../../../stores';
 import { checkIfParentLoop } from '../utils';
 import ModulePreviewResultViewer from '../../ModulePreviewResultViewer.svelte';
 import { refreshStateStore } from '../../../svelte5Utils.svelte';
 import { getStepHistoryLoaderContext } from '../../stepHistoryLoader.svelte';
 import AssetsDropdownButton from '../../assets/AssetsDropdownButton.svelte';
+import { useUiIntent } from '../../copilot/chat/flow/useUiIntent';
+import { editor as meditor } from 'monaco-editor';
 const { selectedId, currentEditor, previewArgs, flowStateStore, flowStore, pathStore, saveDraft, customUi, executionCount } = getContext('FlowEditorContext');
 let { flowModule = $bindable(), failureModule = false, preprocessorModule = false, parentModule = $bindable(), previousModule, scriptKind = 'script', scriptTemplate = 'script', noEditor, enableAi, savedModule = undefined, forceTestTab = false, highlightArg = undefined, isAgentTool = false } = $props();
 let workspaceScriptTag = $state(undefined);
 let workspaceScriptLang = $state(undefined);
 let diffMode = $state(false);
+let diffButtons = $state([
+    {
+        text: 'Quit diff mode',
+        color: 'red',
+        onClick: () => {
+            hideDiffMode();
+        }
+    }
+]);
 let editor = $state();
 let diffEditor = $state();
 let modulePreview = $state();
@@ -69,11 +80,16 @@ let advancedRuntimeSelected = $state('concurrency');
 let s3Kind = $state('s3_client');
 let validCode = $state(true);
 let width = $state(1200);
-let lastJob = $state(undefined);
 let testJob = $state(undefined);
 let testIsLoading = $state(false);
 let scriptProgress = $state(undefined);
 let assets = $derived((flowModule.value.type === 'rawscript' && flowModule.value.assets) || []);
+// UI Intent handling for AI tool control
+useUiIntent(`flow-${flowModule.id}`, {
+    openTab: (tab) => {
+        selectAdvanced(tab);
+    }
+});
 function onModulesChange(savedModule, flowModule) {
     // console.log('onModulesChange', savedModule, flowModule)
     return savedModule?.value?.type === 'rawscript' &&
@@ -138,41 +154,16 @@ let forceReload = $state(0);
 let editorPanelSize = $state(noEditor ? 0 : flowModule.value.type == 'script' ? 30 : 50);
 let editorSettingsPanelSize = $state(100 - untrack(() => editorPanelSize));
 let stepHistoryLoader = getStepHistoryLoaderContext();
-let lastJobId = undefined;
 function onSelectedIdChange() {
     if (!flowStateStore?.val?.[$selectedId]?.schema && flowModule) {
         reload(flowModule);
     }
-    lastJobId = undefined;
-}
-async function getLastJob() {
-    if (!flowStateStore ||
-        !flowModule.id ||
-        flowStateStore.val[flowModule.id]?.previewResult === 'never tested this far' ||
-        !flowStateStore.val[flowModule.id]?.previewJobId) {
-        return;
-    }
-    if (lastJobId == flowStateStore.val[flowModule.id]?.previewJobId ||
-        lastJob?.id == flowStateStore.val[flowModule.id]?.previewJobId ||
-        flowStateStore.val[flowModule.id]?.previewSuccess == undefined) {
-        return;
-    }
-    lastJobId = flowStateStore.val[flowModule.id]?.previewJobId;
-    const job = await JobService.getJob({
-        workspace: $workspaceStore ?? '',
-        id: flowStateStore.val[flowModule.id]?.previewJobId ?? '',
-        noCode: true
-    });
-    if (job && job.type === 'CompletedJob') {
-        lastJobId = flowStateStore.val[flowModule.id]?.previewJobId;
-        lastJob = job;
-    }
 }
 let leftPanelSize = $state(0);
 function showDiffMode() {
     diffMode = true;
     diffEditor?.setOriginal((savedModule?.value).content ?? '');
-    diffEditor?.setModified(editor?.getCode() ?? '');
+    diffEditor?.setModifiedModel(editor?.getModel());
     diffEditor?.show();
     editor?.hide();
 }
@@ -188,14 +179,6 @@ let stepPropPicker = $derived($executionCount != undefined && failureModule
 $effect.pre(() => {
     $selectedId && untrack(() => onSelectedIdChange());
 });
-$effect(() => {
-    if (testJob && testJob.type === 'CompletedJob') {
-        lastJob = $state.snapshot(testJob);
-    }
-    else if ($workspaceStore && $pathStore && flowModule?.id && flowStateStore) {
-        untrack(() => getLastJob());
-    }
-});
 let parentLoop = $derived(flowStore.val && flowModule ? checkIfParentLoop(flowStore.val, flowModule.id) : undefined);
 $effect(() => {
     if (selected === 'test') {
@@ -214,7 +197,13 @@ $effect(() => {
             showDiffMode,
             hideDiffMode,
             diffMode,
-            lastDeployedCode
+            lastDeployedCode,
+            setDiffOriginal: (code) => {
+                diffEditor?.setOriginal(code ?? '');
+            },
+            setDiffButtons: (buttons) => {
+                diffButtons = buttons;
+            }
         });
 });
 onDestroy(() => {
@@ -234,6 +223,10 @@ $effect(() => {
     }
 });
 let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.value.language : undefined);
+let modulePreviewResultViewer = $state(undefined);
+function onJobDone() {
+    modulePreviewResultViewer?.getOutputPickerInner()?.setJobPreview();
+}
 </script>
 
 <svelte:window onkeydown={onKeyDown} />
@@ -395,13 +388,12 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 											<DiffEditor
 												open={false}
 												bind:this={diffEditor}
+												modifiedModel={editor?.getModel() as meditor.ITextModel}
 												automaticLayout
 												fixedOverflowWidgets
 												defaultLang={scriptLangToEditorLang(flowModule.value.language)}
-												class="h-full"
-												showButtons={diffMode}
-												showHistoryButton={false}
-												on:hideDiffMode={hideDiffMode}
+												className="h-full"
+												buttons={diffMode ? diffButtons : []}
 											/>
 										{/key}
 									{/if}
@@ -508,6 +500,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 												bind:testIsLoading
 												bind:scriptProgress
 												focusArg={highlightArg}
+												{onJobDone}
 											/>
 										{:else if selected === 'advanced'}
 											<Tabs bind:selected={advancedSelected}>
@@ -812,15 +805,15 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 												flowModule = flowModule
 												refreshStateStore(flowStore)
 											}}
-											{lastJob}
-											{scriptProgress}
 											{testJob}
+											{scriptProgress}
 											mod={flowModule}
 											{testIsLoading}
 											disableMock={preprocessorModule || failureModule}
 											disableHistory={failureModule}
 											loadingJob={stepHistoryLoader?.stepStates[flowModule.id]?.loadingJobs}
 											tagLabel={customUi?.tagLabel}
+											bind:this={modulePreviewResultViewer}
 										/>
 									</Pane>
 								{/if}

package/package/components/flows/content/FlowModuleScript.svelte

@@ -101,7 +101,7 @@ $effect.pre(() => {
 				{:then Module}
 					<Module.default
 						open={true}
-						class="h-screen"
+						className="h-screen"
 						readOnly
 						automaticLayout
 						defaultLang={scriptLangToEditorLang(language)}

package/package/components/flows/content/FlowModuleSuspend.svelte

@@ -1,5 +1,4 @@
-<script lang="ts">import { run } from 'svelte/legacy';
-import Toggle from '../../Toggle.svelte';
+<script lang="ts">import Toggle from '../../Toggle.svelte';
 import Tooltip from '../../Tooltip.svelte';
 import InputTransformForm from '../../InputTransformForm.svelte';
 import { getContext, tick, untrack } from 'svelte';
@@ -32,16 +31,29 @@ async function loadGroups() {
         }
     };
 }
-run(() => {
+$effect(() => {
     if ($workspaceStore && allUserGroups.length === 0) {
         untrack(() => {
             loadGroups();
         });
     }
 });
+$effect(() => {
+    // If the schema is empty, remove the form
+    if (Object.keys(flowModule?.suspend?.resume_form?.schema?.properties ?? {}).length === 0) {
+        untrack(() => {
+            tick().then(() => {
+                if (!flowModule.suspend)
+                    return;
+                flowModule.suspend.resume_form = undefined;
+            });
+        });
+    }
+});
 let jsonView = $state(false);
 </script>
 
+A{JSON.stringify(flowModule.suspend?.resume_form)}B
 <Section label="Suspend/Approval/Prompt" class="w-full">
 	{#snippet action()}
 		<SuspendDrawer text="Approval/Prompt helpers" />
@@ -216,21 +228,7 @@ let jsonView = $state(false);
 		<div class="grid grid-cols-4 mt-4 gap-8">
 			<div class="col-span-2">
 				{#if flowModule?.suspend?.resume_form}
-					<EditableSchemaDrawer
-						bind:schema={flowModule.suspend.resume_form.schema}
-						on:change={(e) => {
-							const schema = e.detail
-
-							// If the schema is empty, remove the form
-							if (Object.keys(schema?.properties ?? {}).length === 0) {
-								tick().then(() => {
-									if (!flowModule.suspend) return
-									flowModule.suspend.resume_form = undefined
-								})
-							}
-						}}
-						{jsonView}
-					/>
+					<EditableSchemaDrawer bind:schema={flowModule.suspend.resume_form.schema} {jsonView} />
 				{:else if emptyString($enterpriseLicense)}
 					<Alert type="warning" title="Adding a form to the approval page is an EE feature" />
 				{:else}

package/package/components/flows/content/FlowWhileLoop.svelte

@@ -15,6 +15,7 @@ import FlowLoopIterationPreview from '../../FlowLoopIterationPreview.svelte';
 import FlowModuleDeleteAfterUse from './FlowModuleDeleteAfterUse.svelte';
 import FlowModuleSkip from './FlowModuleSkip.svelte';
 import TabsV2 from '../../common/tabs/TabsV2.svelte';
+import { useUiIntent } from '../../copilot/chat/flow/useUiIntent';
 const { flowStateStore } = getContext('FlowEditorContext');
 let { mod = $bindable(), previousModule, parentModule, noEditor } = $props();
 let selected = $state('early-stop');
@@ -22,6 +23,11 @@ let previewOpen = $state(false);
 let jobId = $state(undefined);
 let job = $state(undefined);
 let previewIterationArgs = $derived(flowStateStore.val[mod.id]?.previewArgs ?? {});
+useUiIntent(`whileloopflow-${mod.id}`, {
+    openTab: (tab) => {
+        selected = tab;
+    }
+});
 </script>
 
 <Drawer bind:open={previewOpen} alwaysOpen size="75%">

package/package/components/flows/content/ScriptEditorDrawer.svelte

@@ -14,6 +14,7 @@ let scriptEditorDrawer = $state();
 const dispatch = createEventDispatcher();
 export async function openDrawer(hash, cb) {
     script = undefined;
+    closeAnyway = false;
     scriptEditorDrawer?.openDrawer?.();
     script = await ScriptService.getScriptByHash({
         workspace: $workspaceStore,
@@ -64,7 +65,6 @@ let unsavedModalOpen = $state(false);
 async function checkForUnsavedChanges() {
     if (closeAnyway) {
         scriptEditorDrawer?.closeDrawer();
-        closeAnyway = false;
         return;
     }
     if (savedScript && script) {
@@ -72,6 +72,7 @@ async function checkForUnsavedChanges() {
         const current = cleanValueProperties(script);
         if (orderedJsonStringify(saved) !== orderedJsonStringify(current)) {
             unsavedModalOpen = true;
+            scriptEditorDrawer?.openDrawer();
         }
         else {
             scriptEditorDrawer?.closeDrawer();
@@ -79,6 +80,7 @@ async function checkForUnsavedChanges() {
     }
 }
 let args = $state({});
+let displayEditor = $state(true);
 </script>
 
 <ConfirmationModal
@@ -89,8 +91,9 @@ let args = $state({});
 		unsavedModalOpen = false
 	}}
 	on:confirmed={() => {
-		unsavedModalOpen = false
+		console.log('confirmed')
 		closeAnyway = true
+		unsavedModalOpen = false
 		scriptEditorDrawer?.closeDrawer()
 	}}
 >
@@ -107,7 +110,7 @@ let args = $state({});
 				}
 				unsavedModalOpen = false
 				closeAnyway = true
-				scriptEditorDrawer?.closeDrawer()
+				displayEditor = false
 				diffDrawer?.openDrawer()
 				diffDrawer?.setDiff({
 					title: 'Saved <> Current',
@@ -133,7 +136,6 @@ let args = $state({});
 	bind:this={scriptEditorDrawer}
 	size="1200px"
 	on:close={() => {
-		scriptEditorDrawer?.openDrawer()
 		checkForUnsavedChanges()
 	}}
 >
@@ -146,7 +148,7 @@ let args = $state({});
 			scriptEditorDrawer?.closeDrawer()
 		}}
 	>
-		{#if script}
+		{#if script && displayEditor}
 			{#key script.hash}
 				<ScriptEditor
 					showCaptures={false}
@@ -187,7 +189,7 @@ let args = $state({});
 						return
 					}
 					closeAnyway = true
-					scriptEditorDrawer?.closeDrawer()
+					displayEditor = false
 					diffDrawer?.openDrawer()
 					diffDrawer?.setDiff({
 						mode: 'simple',
@@ -227,10 +229,6 @@ let args = $state({});
 <DiffDrawer
 	bind:this={diffDrawer}
 	on:close={() => {
-		if (!closeAnyway) {
-			scriptEditorDrawer?.openDrawer()
-		} else {
-			closeAnyway = false
-		}
+		displayEditor = true
 	}}
 />

package/package/components/flows/dfs.d.ts

@@ -1,4 +1,4 @@
 import type { FlowModule } from '../../gen';
-export declare function dfs<T>(modules: FlowModule[], f: (x: FlowModule, modules: FlowModule[], branches: FlowModule[][]) => T, { skipToolNodes }?: {
+export declare function dfs<T>(modules: FlowModule[], f: (x: FlowModule, modules: FlowModule[], branches: FlowModule[][]) => T, opts?: {
     skipToolNodes?: boolean;
 }): T[];

package/package/components/flows/dfs.js

@@ -1,27 +1,27 @@
-export function dfs(modules, f, { skipToolNodes = false } = {}) {
+export function dfs(modules, f, opts = {}) {
     let result = [];
     for (const module of modules) {
         if (module.value.type == 'forloopflow' || module.value.type == 'whileloopflow') {
             result = result.concat(f(module, modules, [module.value.modules]));
-            result = result.concat(dfs(module.value.modules, f));
+            result = result.concat(dfs(module.value.modules, f, opts));
         }
         else if (module.value.type == 'branchone') {
             const allBranches = [module.value.default, ...module.value.branches.map((b) => b.modules)];
             result = result.concat(f(module, modules, allBranches));
             for (const branch of allBranches) {
-                result = result.concat(dfs(branch, f));
+                result = result.concat(dfs(branch, f, opts));
             }
         }
         else if (module.value.type == 'branchall') {
             const allBranches = module.value.branches.map((b) => b.modules);
             result = result.concat(f(module, modules, allBranches));
             for (const branch of allBranches) {
-                result = result.concat(dfs(branch, f));
+                result = result.concat(dfs(branch, f, opts));
             }
         }
-        else if (module.value.type == 'aiagent' && !skipToolNodes) {
+        else if (module.value.type == 'aiagent' && !opts.skipToolNodes) {
             result = result.concat(f(module, modules, [module.value.tools]));
-            result = result.concat(dfs(module.value.tools, f));
+            result = result.concat(dfs(module.value.tools, f, opts));
         }
         else {
             result.push(f(module, modules, []));

package/package/components/flows/flowInfers.js

@@ -82,27 +82,27 @@ export async function loadSchemaFromModule(module) {
                         }
                     ]
                 },
-                system_prompt: {
-                    type: 'string',
-                    default: 'You are a helpful assistant'
-                },
                 user_message: {
                     type: 'string'
                 },
+                system_prompt: {
+                    type: 'string'
+                },
                 max_completion_tokens: {
                     type: 'number'
                 },
                 temperature: {
-                    type: 'number'
+                    type: 'number',
+                    description: 'Controls randomness in text generation. Range: 0.0 (deterministic) to 2.0 (random).'
                 }
             },
-            required: ['provider', 'model', 'system_prompt', 'user_message'],
+            required: ['provider', 'model', 'user_message'],
             type: 'object',
             order: [
                 'provider',
                 'model',
-                'system_prompt',
                 'user_message',
+                'system_prompt',
                 'max_completion_tokens',
                 'temperature'
             ]

package/package/components/flows/flowStateUtils.svelte.js

@@ -113,8 +113,7 @@ export async function createBranchAll(id) {
 export async function createAiAgent(id) {
     const aiAgentFlowModules = {
         id,
-        value: { type: 'aiagent', tools: [], input_transforms: {} },
-        summary: 'AI Agent'
+        value: { type: 'aiagent', tools: [], input_transforms: {} }
     };
     const flowModuleState = await loadFlowModuleState(aiAgentFlowModules);
     return [aiAgentFlowModules, flowModuleState];