windmill-components 1.522.0 → 1.531.1

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

@@ -49,7 +49,7 @@ import { refreshStateStore } from '../../../svelte5Utils.svelte';
 import { getStepHistoryLoaderContext } from '../../stepHistoryLoader.svelte';
 import AssetsDropdownButton from '../../assets/AssetsDropdownButton.svelte';
 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 } = $props();
+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);
@@ -63,7 +63,7 @@ let websocketAlive = $state({
     ruff: false,
     shellcheck: false
 });
-let selected = $state(preprocessorModule ? 'test' : 'inputs');
+let selected = $state(preprocessorModule || isAgentTool ? 'test' : 'inputs');
 let advancedSelected = $state('retries');
 let advancedRuntimeSelected = $state('concurrency');
 let s3Kind = $state('s3_client');
@@ -95,6 +95,7 @@ async function reload(flowModule) {
     reloadError = undefined;
     try {
         const { input_transforms, schema } = await loadSchemaFromModule(flowModule);
+        console.log('reload', schema);
         validCode = true;
         if (inputTransformSchemaForm) {
             inputTransformSchemaForm.setArgs(input_transforms);
@@ -102,7 +103,8 @@ async function reload(flowModule) {
         else {
             if (flowModule.value.type == 'rawscript' ||
                 flowModule.value.type == 'script' ||
-                flowModule.value.type == 'flow') {
+                flowModule.value.type == 'flow' ||
+                flowModule.value.type == 'aiagent') {
                 if (!deepEqual(flowModule.value.input_transforms, input_transforms)) {
                     flowModule.value.input_transforms = input_transforms;
                 }
@@ -114,12 +116,12 @@ async function reload(flowModule) {
             }
         }
         await tick();
-        if (!deepEqual(schema, $flowStateStore[flowModule.id]?.schema)) {
-            if (!$flowStateStore[flowModule.id]) {
-                $flowStateStore[flowModule.id] = { schema };
+        if (!deepEqual(schema, flowStateStore.val[flowModule.id]?.schema)) {
+            if (!flowStateStore.val[flowModule.id]) {
+                flowStateStore.val[flowModule.id] = { schema };
             }
             else {
-                $flowStateStore[flowModule.id].schema = schema;
+                flowStateStore.val[flowModule.id].schema = schema;
             }
         }
     }
@@ -138,32 +140,31 @@ let editorSettingsPanelSize = $state(100 - untrack(() => editorPanelSize));
 let stepHistoryLoader = getStepHistoryLoaderContext();
 let lastJobId = undefined;
 function onSelectedIdChange() {
-    if (!$flowStateStore?.[$selectedId]?.schema && flowModule) {
+    if (!flowStateStore?.val?.[$selectedId]?.schema && flowModule) {
         reload(flowModule);
     }
     lastJobId = undefined;
 }
 async function getLastJob() {
-    if (!$flowStateStore ||
+    if (!flowStateStore ||
         !flowModule.id ||
-        $flowStateStore[flowModule.id]?.previewResult === 'never tested this far' ||
-        !$flowStateStore[flowModule.id]?.previewJobId ||
-        !$flowStateStore[flowModule.id]?.previewWorkspaceId) {
+        flowStateStore.val[flowModule.id]?.previewResult === 'never tested this far' ||
+        !flowStateStore.val[flowModule.id]?.previewJobId) {
         return;
     }
-    if (lastJobId == $flowStateStore[flowModule.id]?.previewJobId ||
-        lastJob?.id == $flowStateStore[flowModule.id]?.previewJobId ||
-        $flowStateStore[flowModule.id]?.previewSuccess == undefined) {
+    if (lastJobId == flowStateStore.val[flowModule.id]?.previewJobId ||
+        lastJob?.id == flowStateStore.val[flowModule.id]?.previewJobId ||
+        flowStateStore.val[flowModule.id]?.previewSuccess == undefined) {
         return;
     }
-    lastJobId = $flowStateStore[flowModule.id]?.previewJobId;
+    lastJobId = flowStateStore.val[flowModule.id]?.previewJobId;
     const job = await JobService.getJob({
-        workspace: $flowStateStore[flowModule.id]?.previewWorkspaceId ?? '',
-        id: $flowStateStore[flowModule.id]?.previewJobId ?? '',
+        workspace: $workspaceStore ?? '',
+        id: flowStateStore.val[flowModule.id]?.previewJobId ?? '',
         noCode: true
     });
     if (job && job.type === 'CompletedJob') {
-        lastJobId = $flowStateStore[flowModule.id]?.previewJobId;
+        lastJobId = flowStateStore.val[flowModule.id]?.previewJobId;
         lastJob = job;
     }
 }
@@ -182,8 +183,8 @@ function hideDiffMode() {
 }
 let lastDeployedCode = $derived(onModulesChange(savedModule, flowModule));
 let stepPropPicker = $derived($executionCount != undefined && failureModule
-    ? getFailureStepPropPicker($flowStateStore, flowStore.val, previewArgs.val)
-    : getStepPropPicker($flowStateStore, parentModule, previousModule, flowModule.id, flowStore.val, previewArgs.val, false));
+    ? getFailureStepPropPicker(flowStateStore.val, flowStore.val, previewArgs.val)
+    : getStepPropPicker(flowStateStore.val, parentModule, previousModule, flowModule.id, flowStore.val, previewArgs.val, false));
 $effect.pre(() => {
     $selectedId && untrack(() => onSelectedIdChange());
 });
@@ -191,7 +192,7 @@ $effect(() => {
     if (testJob && testJob.type === 'CompletedJob') {
         lastJob = $state.snapshot(testJob);
     }
-    else if ($workspaceStore && $pathStore && flowModule?.id && $flowStateStore) {
+    else if ($workspaceStore && $pathStore && flowModule?.id && flowStateStore) {
         untrack(() => getLastJob());
     }
 });
@@ -252,6 +253,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 				}
 			}}
 			bind:summary={flowModule.summary}
+			{isAgentTool}
 		>
 			{#snippet header()}
 				<FlowModuleHeader
@@ -276,7 +278,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 					on:fork={async () => {
 						const [module, state] = await fork(flowModule)
 						flowModule = module
-						$flowStateStore[module.id] = state
+						flowStateStore.val[module.id] = state
 					}}
 					on:reload={async () => {
 						if (flowModule.value.type == 'script') {
@@ -295,14 +297,14 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 						const [module, state] = await createScriptFromInlineScript(
 							flowModule,
 							$selectedId,
-							$flowStateStore[flowModule.id].schema,
+							flowStateStore.val[flowModule.id].schema,
 							$pathStore
 						)
 						if (flowModule.value.type == 'rawscript') {
 							module.value.input_transforms = flowModule.value.input_transforms
 						}
 						flowModule = module
-						$flowStateStore[module.id] = state
+						flowStateStore.val[module.id] = state
 					}}
 				/>
 			{/snippet}
@@ -335,97 +337,117 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 
 				<div class="min-h-0 flex-grow" id="flow-editor-editor">
 					<Splitpanes horizontal>
-						<Pane bind:size={editorPanelSize} minSize={10} class="relative">
-							{#if flowModule.value.type === 'rawscript'}
-								{#if !noEditor}
-									{#key flowModule.id}
-										<div class="absolute top-2 right-4 z-10 flex flex-row gap-2">
-											{#if assets?.length}
-												<AssetsDropdownButton {assets} />
-											{/if}
-										</div>
-										<Editor
-											loadAsync
-											folding
-											path={$pathStore + '/' + flowModule.id}
-											bind:websocketAlive
-											bind:this={editor}
-											class="h-full relative"
-											code={flowModule.value.content}
-											scriptLang={flowModule?.value?.language}
-											automaticLayout={true}
-											cmdEnterAction={async () => {
-												selected = 'test'
-												if ($selectedId == flowModule.id) {
-													if (flowModule.value.type === 'rawscript' && editor) {
-														flowModule.value.content = editor.getCode()
+						{#if flowModule.value.type !== 'aiagent'}
+							<Pane bind:size={editorPanelSize} minSize={10} class="relative">
+								{#if flowModule.value.type === 'rawscript'}
+									{#if !noEditor}
+										{#key flowModule.id}
+											<div class="absolute top-2 right-4 z-10 flex flex-row gap-2">
+												{#if assets?.length}
+													<AssetsDropdownButton {assets} />
+												{/if}
+											</div>
+											<Editor
+												loadAsync
+												folding
+												path={$pathStore + '/' + flowModule.id}
+												bind:websocketAlive
+												bind:this={editor}
+												class="h-full relative"
+												code={flowModule.value.content}
+												scriptLang={flowModule?.value?.language}
+												automaticLayout={true}
+												cmdEnterAction={async () => {
+													selected = 'test'
+													if ($selectedId == flowModule.id) {
+														if (flowModule.value.type === 'rawscript' && editor) {
+															flowModule.value.content = editor.getCode()
+														}
+														await reload(flowModule)
+														modulePreview?.runTestWithStepArgs()
 													}
-													await reload(flowModule)
-													modulePreview?.runTestWithStepArgs()
-												}
-											}}
-											on:change={async (event) => {
-												const content = event.detail
-												if (flowModule.value.type === 'rawscript') {
-													if (flowModule.value.content !== content) {
-														flowModule.value.content = content
+												}}
+												on:change={async (event) => {
+													const content = event.detail
+													if (flowModule.value.type === 'rawscript') {
+														if (flowModule.value.content !== content) {
+															flowModule.value.content = content
+														}
+														await reload(flowModule)
 													}
-													await reload(flowModule)
-												}
-											}}
-											formatAction={() => {
-												reload(flowModule)
-												saveDraft()
-											}}
-											fixedOverflowWidgets={true}
-											args={Object.entries(flowModule.value.input_transforms).reduce(
-												(acc, [key, obj]) => {
-													acc[key] = obj.type === 'static' ? obj.value : undefined
-													return acc
-												},
-												{}
-											)}
-											key={`flow-inline-${$workspaceStore}-${$pathStore}-${flowModule.id}`}
-										/>
-										<DiffEditor
-											open={false}
-											bind:this={diffEditor}
-											automaticLayout
-											fixedOverflowWidgets
-											defaultLang={scriptLangToEditorLang(flowModule.value.language)}
-											class="h-full"
-										/>
-									{/key}
-								{/if}
-							{:else if flowModule.value.type === 'script'}
-								{#if !noEditor && (customUi?.hubCode != false || !flowModule?.value?.path?.startsWith('hub/'))}
-									<div class="border-t">
-										{#key forceReload}
-											<FlowModuleScript
-												bind:tag={workspaceScriptTag}
-												bind:language={workspaceScriptLang}
-												showAllCode={false}
-												path={flowModule.value.path}
-												hash={flowModule.value.hash}
+												}}
+												formatAction={() => {
+													reload(flowModule)
+													saveDraft()
+												}}
+												fixedOverflowWidgets={true}
+												args={Object.entries(flowModule.value.input_transforms).reduce(
+													(acc, [key, obj]) => {
+														acc[key] = obj.type === 'static' ? obj.value : undefined
+														return acc
+													},
+													{}
+												)}
+												key={`flow-inline-${$workspaceStore}-${$pathStore}-${flowModule.id}`}
+											/>
+											<DiffEditor
+												open={false}
+												bind:this={diffEditor}
+												automaticLayout
+												fixedOverflowWidgets
+												defaultLang={scriptLangToEditorLang(flowModule.value.language)}
+												class="h-full"
+												showButtons={diffMode}
+												showHistoryButton={false}
+												on:hideDiffMode={hideDiffMode}
 											/>
 										{/key}
-									</div>
+									{/if}
+								{:else if flowModule.value.type === 'script'}
+									{#if !noEditor && (customUi?.hubCode != false || !flowModule?.value?.path?.startsWith('hub/'))}
+										<div class="border-t">
+											{#key forceReload}
+												<FlowModuleScript
+													bind:tag={workspaceScriptTag}
+													bind:language={workspaceScriptLang}
+													showAllCode={false}
+													path={flowModule.value.path}
+													hash={flowModule.value.hash}
+												/>
+											{/key}
+										</div>
+									{/if}
+								{:else if flowModule.value.type === 'flow'}
+									{#key forceReload}
+										<FlowPathViewer path={flowModule.value.path} />
+									{/key}
 								{/if}
-							{:else if flowModule.value.type === 'flow'}
-								{#key forceReload}
-									<FlowPathViewer path={flowModule.value.path} />
-								{/key}
-							{/if}
-						</Pane>
-						<Pane bind:size={editorSettingsPanelSize} minSize={20}>
+							</Pane>
+						{/if}
+						<Pane
+							bind:size={
+								() => {
+									if (flowModule.value.type === 'aiagent') {
+										return 100
+									}
+									return editorSettingsPanelSize
+								},
+								(v) => {
+									if (flowModule.value.type !== 'aiagent') {
+										editorSettingsPanelSize = v
+									}
+								}
+							}
+							minSize={20}
+						>
 							<Splitpanes>
 								<Pane minSize={36} bind:size={leftPanelSize}>
 									<Tabs bind:selected>
-										{#if !preprocessorModule}
+										{#if !preprocessorModule && !isAgentTool}
 											<Tab value="inputs">Step Input</Tab>
 										{/if}
 										<Tab value="test">Test this step</Tab>
-										{#if !preprocessorModule}
+										{#if !preprocessorModule && !isAgentTool}
 											<Tab value="advanced">Advanced</Tab>
 										{/if}
 									</Tabs>
@@ -434,7 +456,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 											? 'h-[calc(100%-68px)]'
 											: 'h-[calc(100%-34px)]'}
 									>
-										{#if selected === 'inputs' && (flowModule.value.type == 'rawscript' || flowModule.value.type == 'script' || flowModule.value.type == 'flow')}
+										{#if selected === 'inputs' && (flowModule.value.type == 'rawscript' || flowModule.value.type == 'script' || flowModule.value.type == 'flow' || flowModule.value.type == 'aiagent')}
 											<div class="h-full overflow-auto bg-surface" id="flow-editor-step-input">
 												<PropPickerWrapper
 													pickableProperties={stepPropPicker.pickableProperties}
@@ -451,7 +473,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 														class="px-1 xl:px-2"
 														bind:this={inputTransformSchemaForm}
 														pickableProperties={stepPropPicker.pickableProperties}
-														schema={$flowStateStore[$selectedId]?.schema ?? {}}
+														schema={flowStateStore.val[$selectedId]?.schema ?? {}}
 														previousModuleId={previousModule?.id}
 														bind:args={
 															() => {
@@ -479,7 +501,7 @@ let rawScriptLang = $derived(flowModule.value.type == 'rawscript' ? flowModule.v
 												bind:this={modulePreview}
 												mod={flowModule}
 												{noEditor}
-												schema={$flowStateStore[$selectedId]?.schema ?? {}}
+												schema={flowStateStore.val[$selectedId]?.schema ?? {}}
 												bind:testJob
 												bind:testIsLoading
 												bind:scriptProgress

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

@@ -12,6 +12,7 @@ interface Props {
     savedModule?: FlowModule | undefined;
     forceTestTab?: boolean;
     highlightArg?: string;
+    isAgentTool?: boolean;
 }
 declare const FlowModuleComponent: import("svelte").Component<Props, {}, "flowModule" | "parentModule">;
 type FlowModuleComponent = ReturnType<typeof FlowModuleComponent>;

package/package/components/flows/content/FlowModuleEarlyStop.svelte

@@ -10,7 +10,7 @@ import { dfs } from '../previousResults';
 const { flowStateStore, flowStore, previewArgs } = getContext('FlowEditorContext');
 let { flowModule = $bindable() } = $props();
 let editor = $state(undefined);
-let stepPropPicker = $derived(getStepPropPicker($flowStateStore, undefined, undefined, flowModule.id, flowStore.val, previewArgs.val, false));
+let stepPropPicker = $derived(getStepPropPicker(flowStateStore.val, undefined, undefined, flowModule.id, flowStore.val, previewArgs.val, false));
 function checkIfParentLoop(flowStoreValue) {
     const flow = JSON.parse(JSON.stringify(flowStoreValue));
     const parents = dfs(flowModule.id, flow, true);
@@ -27,7 +27,7 @@ let isLoop = $derived(flowModule.value.type === 'forloopflow' || flowModule.valu
 let isBranchAll = $derived(flowModule.value.type === 'branchall');
 let isStopAfterIfEnabled = $derived(Boolean(flowModule.stop_after_if));
 let isStopAfterAllIterationsEnabled = $derived(Boolean(flowModule.stop_after_all_iters_if));
-let result = $derived($flowStateStore[flowModule.id]?.previewResult ?? NEVER_TESTED_THIS_FAR);
+let result = $derived(flowStateStore.val[flowModule.id]?.previewResult ?? NEVER_TESTED_THIS_FAR);
 let parentLoopId = $derived(checkIfParentLoop(flowStore.val));
 </script>
 

package/package/components/flows/content/FlowModuleSkip.svelte

@@ -8,7 +8,7 @@ import { getStepPropPicker } from '../previousResults';
 const { flowStateStore, flowStore, previewArgs } = getContext('FlowEditorContext');
 let { flowModule = $bindable(), parentModule, previousModule } = $props();
 let editor = $state(undefined);
-let stepPropPicker = $derived(getStepPropPicker($flowStateStore, parentModule, previousModule, flowModule.id, flowStore.val, previewArgs.val, false));
+let stepPropPicker = $derived(getStepPropPicker(flowStateStore.val, parentModule, previousModule, flowModule.id, flowStore.val, previewArgs.val, false));
 let isSkipEnabled = $derived(Boolean(flowModule.skip_if));
 </script>
 

package/package/components/flows/content/FlowModuleSleep.svelte

@@ -14,7 +14,7 @@ schema.properties['sleep'] = {
     type: 'number'
 };
 let editor = $state(undefined);
-const result = $flowStateStore[$selectedId]?.previewResult ?? {};
+const result = flowStateStore.val[$selectedId]?.previewResult ?? {};
 let isSleepEnabled = $derived(Boolean(flowModule.sleep));
 </script>
 

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

@@ -15,7 +15,7 @@ import SuspendDrawer from './SuspendDrawer.svelte';
 import EditableSchemaDrawer from '../../schema/EditableSchemaDrawer.svelte';
 import AddProperty from '../../schema/AddProperty.svelte';
 const { selectedId, flowStateStore } = getContext('FlowEditorContext');
-const result = $flowStateStore[$selectedId]?.previewResult ?? {};
+const result = flowStateStore.val[$selectedId]?.previewResult ?? {};
 let editor = $state(undefined);
 let { flowModule = $bindable(), previousModuleId } = $props();
 let schema = $state(emptySchema());

package/package/components/flows/content/FlowModuleWrapper.svelte

@@ -17,7 +17,7 @@ const { selectedId, flowStateStore } = getContext('FlowEditorContext');
 const { triggersState, triggersCount } = getContext('TriggerContext');
 let scriptKind = $state('script');
 let scriptTemplate = $state('script');
-let { flowModule = $bindable(), noEditor = false, enableAi = false, savedModule = undefined, parentModule = $bindable(), previousModule = undefined, forceTestTab, highlightArg } = $props();
+let { flowModule = $bindable(), noEditor = false, enableAi = false, savedModule = undefined, parentModule = $bindable(), previousModule = undefined, forceTestTab, highlightArg, isAgentTool = false } = $props();
 function initializePrimaryScheduleForTriggerScript(module) {
     const primaryIndex = triggersState.triggers.findIndex((t) => t.isPrimary);
     if (primaryIndex === -1) {
@@ -61,7 +61,7 @@ async function createModuleFromScript(path, summary, kind, hash) {
         initializePrimaryScheduleForTriggerScript(module);
     }
     flowModule = module;
-    $flowStateStore[module.id] = state;
+    flowStateStore.val[module.id] = state;
 }
 </script>
 
@@ -100,7 +100,7 @@ async function createModuleFromScript(path, summary, kind, hash) {
 					const [module, state] = await pickFlow(path, summary, flowModule.id)
 
 					flowModule = module
-					$flowStateStore[module.id] = state
+					flowStateStore.val[module.id] = state
 				}}
 			/>
 		{:else}
@@ -137,13 +137,13 @@ async function createModuleFromScript(path, summary, kind, hash) {
 					}
 
 					flowModule = module
-					$flowStateStore[module.id] = state
+					flowStateStore.val[module.id] = state
 				}}
 				failureModule={$selectedId === 'failure'}
 				preprocessorModule={$selectedId === 'preprocessor'}
 			/>
 		{/if}
-	{:else if flowModule.value.type === 'rawscript' || flowModule.value.type === 'script' || flowModule.value.type === 'flow'}
+	{:else if flowModule.value.type === 'rawscript' || flowModule.value.type === 'script' || flowModule.value.type === 'flow' || flowModule.value.type === 'aiagent'}
 		<FlowModuleComponent
 			{noEditor}
 			bind:flowModule
@@ -157,6 +157,7 @@ async function createModuleFromScript(path, summary, kind, hash) {
 			{savedModule}
 			forceTestTab={forceTestTab?.[flowModule.id]}
 			highlightArg={highlightArg?.[flowModule.id]}
+			{isAgentTool}
 		/>
 	{/if}
 {:else if flowModule.value.type === 'forloopflow' || flowModule.value.type == 'whileloopflow'}
@@ -244,4 +245,13 @@ async function createModuleFromScript(path, summary, kind, hash) {
 			{/each}
 		{/if}
 	{/each}
+{:else if flowModule.value.type === 'aiagent'}
+	{#each flowModule.value.tools as _, index (index)}
+		<FlowModuleWrapper
+			{noEditor}
+			bind:flowModule={flowModule.value.tools[index]}
+			bind:parentModule={flowModule}
+			isAgentTool
+		/>
+	{/each}
 {/if}

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

@@ -9,6 +9,7 @@ interface Props {
     previousModule?: FlowModule | undefined;
     forceTestTab?: Record<string, boolean>;
     highlightArg?: Record<string, string | undefined>;
+    isAgentTool?: boolean;
 }
 declare const FlowModuleWrapper: import("svelte").Component<Props, {}, "flowModule" | "parentModule">;
 type FlowModuleWrapper = ReturnType<typeof FlowModuleWrapper>;

package/package/components/flows/content/FlowResult.svelte

@@ -2,11 +2,11 @@
 import { workspaceStore } from '../../../stores';
 import FlowCard from '../common/FlowCard.svelte';
 import Button from '../../common/button/Button.svelte';
-let { job, isOwner, localDurationStatuses, suspendStatus, noEditor, onOpenDetails } = $props();
+let { job, isOwner, suspendStatus, noEditor, onOpenDetails } = $props();
 </script>
 
 <FlowCard {noEditor} title="Flow result">
-	{#if job && isOwner !== undefined && localDurationStatuses && suspendStatus}
+	{#if job && isOwner !== undefined && suspendStatus}
 		<div class="px-4 py-2">
 			<FlowPreviewResult
 				{job}
@@ -14,7 +14,6 @@ let { job, isOwner, localDurationStatuses, suspendStatus, noEditor, onOpenDetail
 				{isOwner}
 				hideFlowResult={false}
 				hideDownloadLogs={false}
-				{localDurationStatuses}
 				innerModules={[]}
 				{suspendStatus}
 				{extra}

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

@@ -1,11 +1,9 @@
-import type { DurationStatus } from '../../graph';
 import type { Job } from '../../../gen';
-import type { Writable } from 'svelte/store';
+import type { StateStore } from '../../../utils';
 interface Props {
     job?: Job;
     isOwner?: boolean;
-    localDurationStatuses?: Writable<Record<string, DurationStatus>>;
-    suspendStatus?: Writable<Record<string, {
+    suspendStatus?: StateStore<Record<string, {
         job: Job;
         nb: number;
     }>>;

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

@@ -21,7 +21,7 @@ let selected = $state('early-stop');
 let previewOpen = $state(false);
 let jobId = $state(undefined);
 let job = $state(undefined);
-let previewIterationArgs = $derived($flowStateStore[mod.id]?.previewArgs ?? {});
+let previewIterationArgs = $derived(flowStateStore.val[mod.id]?.previewArgs ?? {});
 </script>
 
 <Drawer bind:open={previewOpen} alwaysOpen size="75%">

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

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

package/package/components/flows/dfs.js

@@ -1,4 +1,4 @@
-export function dfs(modules, f) {
+export function dfs(modules, f, { skipToolNodes = false } = {}) {
     let result = [];
     for (const module of modules) {
         if (module.value.type == 'forloopflow' || module.value.type == 'whileloopflow') {
@@ -19,6 +19,10 @@ export function dfs(modules, f) {
                 result = result.concat(dfs(branch, f));
             }
         }
+        else if (module.value.type == 'aiagent' && !skipToolNodes) {
+            result = result.concat(f(module, modules, [module.value.tools]));
+            result = result.concat(dfs(module.value.tools, f));
+        }
         else {
             result.push(f(module, modules, []));
         }