diff --git a/frontend/src/lib/components/FlowPreviewContent.svelte b/frontend/src/lib/components/FlowPreviewContent.svelte
index 68eb925c48..d88023ba6a 100644
--- a/frontend/src/lib/components/FlowPreviewContent.svelte
+++ b/frontend/src/lib/components/FlowPreviewContent.svelte
@@ -45,6 +45,9 @@
 		render?: boolean
 		onJobDone?: () => void
 		upToId?: string | undefined
+		customUi?: {
+			tagLabel?: string | undefined
+		}
 	}
 
 	let {
@@ -568,6 +571,7 @@
 					bind:rightColumnSelect
 					bind:isOwner
 					{render}
+					customUi={customUi}
 				/>
 			{:else}
 				<div class="italic text-tertiary h-full grow"> Flow status will be displayed here </div>
diff --git a/frontend/src/lib/components/FlowStatusViewer.svelte b/frontend/src/lib/components/FlowStatusViewer.svelte
index 14adb0ac62..a479ac438e 100644
--- a/frontend/src/lib/components/FlowStatusViewer.svelte
+++ b/frontend/src/lib/components/FlowStatusViewer.svelte
@@ -28,6 +28,9 @@
 		job?: Job | undefined
 		render?: boolean
 		suspendStatus?: any
+		customUi?: {
+			tagLabel?: string | undefined
+		}
 	}
 
 	let {
@@ -49,7 +52,8 @@
 		localDurationStatuses = $bindable(writable({})),
 		job = $bindable(undefined),
 		render = true,
-		suspendStatus = $bindable(writable({}))
+		suspendStatus = $bindable(writable({})),
+		customUi
 	}: Props = $props()
 
 	let lastJobId: string = jobId
@@ -115,4 +119,5 @@
 	{wideResults}
 	bind:rightColumnSelect
 	{render}
+	{customUi}
 />
diff --git a/frontend/src/lib/components/FlowStatusViewerInner.svelte b/frontend/src/lib/components/FlowStatusViewerInner.svelte
index a575235df6..86fb633074 100644
--- a/frontend/src/lib/components/FlowStatusViewerInner.svelte
+++ b/frontend/src/lib/components/FlowStatusViewerInner.svelte
@@ -87,6 +87,9 @@
 		rightColumnSelect?: 'timeline' | 'node_status' | 'node_definition' | 'user_states'
 		localModuleStates?: Writable<Record<string, GraphModuleState>>
 		localDurationStatuses?: Writable<Record<string, DurationStatus>>
+		customUi?: {
+			tagLabel?: string | undefined
+		}
 	}
 
 	let {
@@ -115,7 +118,8 @@
 		job = $bindable(undefined),
 		rightColumnSelect = $bindable('timeline'),
 		localModuleStates = writable({}),
-		localDurationStatuses = writable({})
+		localDurationStatuses = writable({}),
+		customUi
 	}: Props = $props()
 	let recursiveRefresh: Record<string, (clear, root) => Promise<void>> = $state({})
 
@@ -1336,6 +1340,7 @@
 
 									{#if selectedNode == 'end'}
 										<FlowJobResult
+											tagLabel={customUi?.tagLabel}
 											workspaceId={job?.workspace_id}
 											jobId={job?.id}
 											filename={job.id}
@@ -1411,6 +1416,7 @@
 											</div>
 										{/if}
 										<FlowJobResult
+											tagLabel={customUi?.tagLabel}
 											workspaceId={job?.workspace_id}
 											jobId={node.job_id}
 											noBorder
diff --git a/frontend/src/lib/components/custom_ui.ts b/frontend/src/lib/components/custom_ui.ts
index ccab5b871d..a99db25cd2 100644
--- a/frontend/src/lib/components/custom_ui.ts
+++ b/frontend/src/lib/components/custom_ui.ts
@@ -37,6 +37,7 @@ export type FlowBuilderWhitelabelCustomUi = {
 	downloadLogs?: boolean
 	tagSelectPlaceholder?: string
 	tagSelectNoLabel?: boolean
+	tagLabel?: string
 }
 
 export type DisplayResultUi = {
diff --git a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
index 5602383ea4..5a8b681173 100644
--- a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
@@ -881,6 +881,7 @@
 											disableMock={preprocessorModule || failureModule}
 											disableHistory={failureModule}
 											loadingJob={stepHistoryLoader?.stepStates[flowModule.id]?.loadingJobs}
+											tagLabel={customUi?.tagLabel}
 										/>
 									</Pane>
 								{/if}
diff --git a/frontend/src/routes/test_dev/sdk_flow/+page.svelte b/frontend/src/routes/test_dev/sdk_flow/+page.svelte
index f1ff418cc9..a70a0f5801 100644
--- a/frontend/src/routes/test_dev/sdk_flow/+page.svelte
+++ b/frontend/src/routes/test_dev/sdk_flow/+page.svelte
@@ -34,6 +34,7 @@
 	let flowStateStore = writable({})
 
 	let customUi: FlowBuilderWhitelabelCustomUi = {
+		tagLabel: 'agent',
 		// disableAi: true
 	}
 </script>