From f37d3606446d23f8b11a94ea1ce5f5d4836fae17 Mon Sep 17 00:00:00 2001
From: Arnaud <31803803+Araden14@users.noreply.github.com>
Date: Fri, 8 May 2026 18:57:36 +0200
Subject: [PATCH] fix(flows): populate error handler input args from failure
 picker (#9087)

* fix(flows): populate error handler input args from failure picker

* style(flows): fix indentation in failure-step branch
---
 .../lib/components/flows/stepsInputArgs.svelte.ts   | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/frontend/src/lib/components/flows/stepsInputArgs.svelte.ts b/frontend/src/lib/components/flows/stepsInputArgs.svelte.ts
index 025bca1bfe..06a4972ee5 100644
--- a/frontend/src/lib/components/flows/stepsInputArgs.svelte.ts
+++ b/frontend/src/lib/components/flows/stepsInputArgs.svelte.ts
@@ -4,10 +4,10 @@ import {
 	dfs,
 	getPreviousModule,
 	getStepPropPicker,
-	type PickableProperties
+	type PickableProperties,
+	getFailureStepPropPicker
 } from './previousResults'
 import { evalValue } from './utils.svelte'
-
 export class StepsInputArgs {
 	#stepsEvaluated = $state<Record<string, Record<string, any>>>({})
 	#steps = $state<Record<string, Record<string, any>>>({})
@@ -158,6 +158,15 @@ export class StepsInputArgs {
 		flow: OpenFlow | undefined,
 		previewArgs: Record<string, any> | undefined
 	) {
+		if (id === 'failure' && flow && flow.value.failure_module && flowState) {
+			const picker = getFailureStepPropPicker(flowState, flow, previewArgs)
+			this.initializeFromSchema(
+				flow.value.failure_module,
+				flowState['failure']?.schema ?? {},
+				picker.pickableProperties
+			)
+			return
+		}
 		if (!flowState || !flow) {
 			return
 		}