From ab199f47a77d26a419be91a873670a85df931238 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Wed, 5 Aug 2026 16:47:47 +0100
Subject: [PATCH 01/10] - tweak hardcoded params - add custom param ui - add
 image-text-to-video tag

---
 gradio/workflow.py                            |  48 ++-
 .../workflow/WorkflowNodeSF.svelte            | 313 +++++++++++++++++-
 js/workflowcanvas/workflow/model-api.ts       |  16 +-
 js/workflowcanvas/workflow/node-library.ts    |   8 +
 .../workflow/workflow-executor.ts             |  16 +-
 .../workflow/workflow-store.test.ts           | 156 +++++++++
 js/workflowcanvas/workflow/workflow-store.ts  |  53 ++-
 js/workflowcanvas/workflow/workflow-types.ts  |   1 +
 8 files changed, 575 insertions(+), 36 deletions(-)

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 65b8b48bfd1..4b4ad25ad80 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -885,17 +885,15 @@ def process_item(item):
     "text-to-video": "text_to_video",
     "image-to-image": "image_to_image",
     "image-to-video": "image_to_video",
+    "image-text-to-video": "image_to_video",
     "image-classification": "image_classification",
     "object-detection": "object_detection",
     "image-segmentation": "image_segmentation",
-    "image-to-text": "image_to_text",
     "automatic-speech-recognition": "automatic_speech_recognition",
     "audio-classification": "audio_classification",
-    "visual-question-answering": "visual_question_answering",
-    "document-question-answering": "document_question_answering",
-    # Not visual_question_answering: the Hub routes every image-text-to-text
-    # model as `conversational`, and no provider serves the VQA task at all,
-    # so a task-specific call fails for every model carrying this tag.
+    "image-to-text": "chat_completion",
+    "visual-question-answering": "chat_completion",
+    "document-question-answering": "chat_completion",
     "image-text-to-text": "chat_completion",
 }
 
@@ -1006,6 +1004,20 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
                 f"{model_name} returned no text (finish_reason={finish_reason})."
             )
         return json.dumps([text])
+    try:
+        fn_params = set(inspect.signature(fn).parameters)
+    except (TypeError, ValueError):
+        fn_params = set()
+    if "extra_body" in fn_params:
+        known: dict = {}
+        extras: dict = {}
+        for k, v in clean.items():
+            (known if k in fn_params else extras)[k] = v
+        if extras:
+            clean = {
+                **known,
+                "extra_body": {**(known.get("extra_body") or {}), **extras},
+            }
     if endpoint == "text_generation":
         clean.setdefault("max_new_tokens", 512)
         try:
@@ -1080,7 +1092,11 @@ def call_model(
         client = InferenceClient(model=model_id, token=hf_token, provider=provider)
         args = json.loads(args_json)
         if isinstance(args, dict):
-            endpoint = pipeline_tag or ""
+            endpoint = (
+                _PIPELINE_TAG_TO_ENDPOINT.get(pipeline_tag or "")
+                or pipeline_tag
+                or ""
+            )
             return _dispatch_model_endpoint(client, endpoint, args)
 
         task = pipeline_tag or "text-generation"
@@ -1114,20 +1130,20 @@ def call_model(
             }
             return _dispatch_model_endpoint(client, endpoint, kwargs)
 
-        # Fallback for tasks not handled above: chat_completion (works for most
-        # text models across providers), then a raw POST as last resort.
-        try:
-            r = client.chat_completion(
-                [{"role": "user", "content": a0}], max_tokens=512
+        def _resolve(v):
+            return (
+                _img_url(v)
+                if isinstance(v, dict) and ("url" in v or "path" in v)
+                else v
             )
-            return json.dumps([r.choices[0].message.content])
-        except Exception:
-            pass
+
+        a1_missing = a1 is None or a1 == ""
+        payload = _resolve(a0) if a1_missing else [_resolve(a0), _resolve(a1)]
         headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}
         fallback_resp = httpx.post(
             f"https://api-inference.huggingface.co/models/{model_id}",
             headers=headers,
-            json={"inputs": a0 if not a1 else [a0, a1]},
+            json={"inputs": payload},
             timeout=60,
         )
         fallback_resp.raise_for_status()
diff --git a/js/workflowcanvas/workflow/WorkflowNodeSF.svelte b/js/workflowcanvas/workflow/WorkflowNodeSF.svelte
index 40313a80201..84cc0b91ce1 100644
--- a/js/workflowcanvas/workflow/WorkflowNodeSF.svelte
+++ b/js/workflowcanvas/workflow/WorkflowNodeSF.svelte
@@ -1,6 +1,12 @@
 <script lang="ts">
 	import { getContext } from "svelte";
-	import { resizeNode, setNodeSize, workflow } from "./workflow-store";
+	import {
+		add_custom_port,
+		remove_custom_port,
+		resizeNode,
+		setNodeSize,
+		workflow
+	} from "./workflow-store";
 	import NodeWidget from "./NodeWidget.svelte";
 	import PlayIcon from "./icons/PlayIcon.svelte";
 	import OpenLinkIcon from "./icons/OpenLinkIcon.svelte";
@@ -193,6 +199,34 @@
 		}
 	}
 
+	let showCustomParamForm = $state(false);
+	let customParamName = $state("");
+	let customParamType = $state<PortType>("text");
+
+	function submitCustomParam(): void {
+		const raw = customParamName.trim();
+		if (!raw) return;
+		// to match the InferenceClient param naming convention (snake_case)
+		const id = raw
+			.toLowerCase()
+			.replace(/[^a-z0-9_]+/g, "_")
+			.replace(/^_+|_+$/g, "");
+		if (!id) return;
+		if (node.inputs.some((p) => p.id === id)) return;
+		const label = id
+			.replace(/_/g, " ")
+			.replace(/\b\w/g, (c) => c.toUpperCase());
+		add_custom_port(node.id, {
+			id,
+			label,
+			type: customParamType,
+			required: false
+		});
+		customParamName = "";
+		customParamType = "text";
+		showCustomParamForm = false;
+	}
+
 	function castChoiceValue(v: string, portType: PortType): NodeDataValue {
 		if (portType === "number") {
 			const n = Number(v);
@@ -489,20 +523,28 @@
 		</div>
 	{/if}
 
-	<!-- Input ports -->
 	{#if node.inputs.length > 0}
-		{@const hiddenCount = node.inputs.filter(
+		{@const orderedInputs = [
+			...node.inputs.filter((p) => p.custom),
+			...node.inputs.filter((p) => !p.custom)
+		]}
+		{@const hiddenCount = orderedInputs.filter(
 			(p) =>
-				p.required === false && !connectedPorts.has(`${node.id}:${p.id}:input`)
+				!p.custom &&
+				p.required === false &&
+				!connectedPorts.has(`${node.id}:${p.id}:input`)
 		).length}
 		{@const collapsible = hiddenCount > 0}
 		<div class="ports" class:widget-ports={hasWidget}>
-			{#each node.inputs as port}
+			{#each orderedInputs as port}
 				{@const portConnected = connectedPorts.has(
 					`${node.id}:${port.id}:input`
 				)}
 				{@const visible =
-					showAllInputs || portConnected || port.required !== false}
+					showAllInputs ||
+					portConnected ||
+					port.required !== false ||
+					port.custom}
 				{#if visible}
 					{@const inlineWidget =
 						!portConnected &&
@@ -546,7 +588,7 @@
 							<span
 								class="port-label"
 								class:port-label-optional={port.required === false}
-								>{port.label}</span
+								class:port-label-custom={port.custom}>{port.label}</span
 							>
 							{#if !inlineWidget}
 								<span
@@ -554,6 +596,21 @@
 									style="color: {PORT_COLOR[port.type]}">{port.type}</span
 								>
 							{/if}
+							{#if port.custom && !readOnly}
+								<button
+									class="port-remove-btn"
+									title="Remove custom param"
+									aria-label="Remove custom param {port.label}"
+									onpointerdown={(e) => e.stopPropagation()}
+									onmousedown={(e) => e.stopPropagation()}
+									onclick={(e) => {
+										e.stopPropagation();
+										remove_custom_port(node.id, port.id);
+									}}
+								>
+									✕
+								</button>
+							{/if}
 						{/if}
 						{#if inlineWidget}
 							{#if port.type === "number"}
@@ -702,6 +759,77 @@
 							: "s"}{/if}
 				</button>
 			{/if}
+			{#if !readOnly && node.kind === "transform" && node.model_id}
+				<div
+					class="custom-param-wrap"
+					onpointerdown={(e) => e.stopPropagation()}
+					onmousedown={(e) => e.stopPropagation()}
+				>
+					{#if showCustomParamForm}
+						<div class="custom-param-form">
+							<input
+								class="custom-param-input"
+								type="text"
+								bind:value={customParamName}
+								placeholder="param name (e.g. strength)"
+								onkeydown={(e) => {
+									if (e.key === "Enter") {
+										e.preventDefault();
+										submitCustomParam();
+									}
+									if (e.key === "Escape") showCustomParamForm = false;
+								}}
+							/>
+							<div class="custom-param-form-row">
+								<select
+									class="custom-param-select"
+									bind:value={customParamType}
+								>
+									<option value="text">text</option>
+									<option value="number">number</option>
+									<option value="boolean">boolean</option>
+									<option value="image">image</option>
+									<option value="audio">audio</option>
+								</select>
+								<button
+									class="custom-param-add"
+									onclick={(e) => {
+										e.stopPropagation();
+										submitCustomParam();
+									}}
+									disabled={!customParamName.trim()}
+								>
+									Add
+								</button>
+								<button
+									class="custom-param-cancel"
+									title="Cancel"
+									aria-label="Cancel adding custom param"
+									onclick={(e) => {
+										e.stopPropagation();
+										showCustomParamForm = false;
+										customParamName = "";
+									}}
+								>
+									✕
+								</button>
+							</div>
+						</div>
+					{:else}
+						<button
+							class="custom-param-btn"
+							onclick={(e) => {
+								e.stopPropagation();
+								showCustomParamForm = true;
+							}}
+							title="Add a param the default schema doesn't include (e.g. provider-specific overrides)"
+						>
+							<span class="custom-param-btn-icon" aria-hidden="true">+</span> Add
+							param
+						</button>
+					{/if}
+				</div>
+			{/if}
 		</div>
 	{/if}
 
@@ -1377,6 +1505,135 @@
 		color: #8b8d98;
 	}
 
+	.custom-param-btn {
+		display: block;
+		width: 100%;
+		padding: 4px 12px;
+		border: none;
+		background: transparent;
+		font-family: "JetBrains Mono", monospace;
+		font-size: 9px;
+		font-weight: 600;
+		color: #3e3f4d;
+		cursor: pointer;
+		text-align: left;
+		letter-spacing: 0.03em;
+		transition: color 0.15s;
+	}
+
+	.custom-param-btn:hover {
+		color: #8b8d98;
+	}
+
+	/* Fixed-width slot so the "+" aligns with .ports-toggle's "▾" one row up. */
+	.custom-param-btn-icon {
+		display: inline-block;
+		width: 0.8em;
+		text-align: left;
+	}
+
+	/* Padding mirrors .port-inline-config so the input's left edge lines up
+	   with the other inline text inputs (Prompt / Scheduler etc.). */
+	.custom-param-form {
+		display: flex;
+		flex-direction: column;
+		gap: 4px;
+		padding: 1px 12px 3px 20px;
+	}
+
+	.custom-param-form-row {
+		display: flex;
+		gap: 4px;
+		align-items: center;
+	}
+
+	.custom-param-input {
+		width: 100%;
+	}
+
+	.custom-param-select {
+		flex: 1;
+	}
+
+	.custom-param-input,
+	.custom-param-select {
+		font-family: "JetBrains Mono", monospace;
+		font-size: 10px;
+		padding: 0 7px;
+		border: 1px solid #1e1f2a;
+		border-radius: 4px;
+		background: transparent;
+		color: inherit;
+		min-width: 0;
+		box-sizing: border-box;
+		height: 24px;
+		line-height: 22px;
+	}
+
+	.custom-param-input:focus,
+	.custom-param-select:focus {
+		outline: none;
+		border-color: #3e3f4d;
+	}
+
+	.custom-param-input::placeholder {
+		color: #4a4b58;
+	}
+
+	.custom-param-add,
+	.custom-param-cancel {
+		display: inline-flex;
+		align-items: center;
+		justify-content: center;
+		height: 24px;
+		font-family: "JetBrains Mono", monospace;
+		font-size: 11px;
+		font-weight: 600;
+		padding: 0 10px;
+		border: none;
+		background: transparent;
+		color: #6b6e78;
+		cursor: pointer;
+		line-height: 1;
+	}
+
+	.custom-param-cancel {
+		font-size: 14px;
+		padding: 0 8px;
+	}
+
+	.custom-param-add:not(:disabled):hover,
+	.custom-param-cancel:hover {
+		color: #b8b9c4;
+	}
+
+	.custom-param-add:disabled {
+		opacity: 0.35;
+		cursor: not-allowed;
+	}
+
+	.port-label-custom {
+		font-style: italic;
+	}
+
+	/* order + margin-left: auto pin ✕ to the far right after any inline
+	   widget, without stealing the widget's width. */
+	.port-remove-btn {
+		order: 999;
+		margin-left: auto;
+		padding: 0 4px;
+		border: none;
+		background: transparent;
+		color: #6b6d78;
+		font-size: 10px;
+		cursor: pointer;
+		line-height: 1;
+	}
+
+	.port-remove-btn:hover {
+		color: #ef4444;
+	}
+
 	.port-inline-config {
 		padding: 1px 12px 3px 20px;
 	}
@@ -1557,6 +1814,48 @@
 		color: #6b6e78;
 	}
 
+	:global(body:not(.dark)) .custom-param-btn {
+		color: #8b8d98;
+	}
+
+	:global(body:not(.dark)) .custom-param-btn:hover {
+		color: #3e3f4d;
+	}
+
+	:global(body:not(.dark)) .custom-param-input,
+	:global(body:not(.dark)) .custom-param-select {
+		background: #f8f9fb;
+		border-color: #e2e4ea;
+		color: #1a1b25;
+	}
+
+	:global(body:not(.dark)) .custom-param-input:focus,
+	:global(body:not(.dark)) .custom-param-select:focus {
+		border-color: #b8b9c4;
+	}
+
+	:global(body:not(.dark)) .custom-param-input::placeholder {
+		color: #c0c2cc;
+	}
+
+	:global(body:not(.dark)) .custom-param-add,
+	:global(body:not(.dark)) .custom-param-cancel {
+		color: #8b8d98;
+	}
+
+	:global(body:not(.dark)) .custom-param-add:not(:disabled):hover,
+	:global(body:not(.dark)) .custom-param-cancel:hover {
+		color: #1a1b25;
+	}
+
+	:global(body:not(.dark)) .port-remove-btn {
+		color: #c0c2cc;
+	}
+
+	:global(body:not(.dark)) .port-remove-btn:hover {
+		color: #ef4444;
+	}
+
 	.node-endpoint-row {
 		padding: 2px 12px 0 0;
 		display: flex;
diff --git a/js/workflowcanvas/workflow/model-api.ts b/js/workflowcanvas/workflow/model-api.ts
index 323fecef395..897d10723ec 100644
--- a/js/workflowcanvas/workflow/model-api.ts
+++ b/js/workflowcanvas/workflow/model-api.ts
@@ -33,6 +33,7 @@ export const PIPELINE_TAG_TO_ENDPOINT: Record<string, string> = {
 	"text-to-video": "text_to_video",
 	"image-to-image": "image_to_image",
 	"image-to-video": "image_to_video",
+	"image-text-to-video": "image_to_video",
 	"text-generation": "text_generation",
 	"text2text-generation": "text_generation",
 	conversational: "text_generation",
@@ -48,12 +49,15 @@ export const PIPELINE_TAG_TO_ENDPOINT: Record<string, string> = {
 	"image-classification": "image_classification",
 	"object-detection": "object_detection",
 	"image-segmentation": "image_segmentation",
-	"image-to-text": "image_to_text",
 	"automatic-speech-recognition": "automatic_speech_recognition",
 	"audio-classification": "audio_classification",
-	"visual-question-answering": "visual_question_answering",
-	"document-question-answering": "document_question_answering",
-	// Vision-language models are served as `conversational`, so they go through
-	// chat completions rather than a task-specific endpoint.
-	"image-text-to-text": "chat_completion"
+	"audio-to-audio": "audio_to_audio",
+	"tabular-classification": "tabular_classification",
+	"tabular-regression": "tabular_regression",
+	"table-question-answering": "table_question_answering",
+	"zero-shot-image-classification": "zero_shot_image_classification",
+	"image-text-to-text": "chat_completion",
+	"visual-question-answering": "chat_completion",
+	"document-question-answering": "chat_completion",
+	"image-to-text": "chat_completion"
 };
diff --git a/js/workflowcanvas/workflow/node-library.ts b/js/workflowcanvas/workflow/node-library.ts
index 5cd4b3bc328..f29c6afed9d 100644
--- a/js/workflowcanvas/workflow/node-library.ts
+++ b/js/workflowcanvas/workflow/node-library.ts
@@ -134,6 +134,13 @@ export const TASK_SCHEMAS: Record<string, TaskSchema> = {
 		inputs: [{ id: "in_0", label: "Image", type: "image" }],
 		outputs: [{ id: "out_0", label: "Video", type: "video" }]
 	},
+	"image-text-to-video": {
+		inputs: [
+			{ id: "in_0", label: "Image", type: "image" },
+			{ id: "in_1", label: "Prompt", type: "text" }
+		],
+		outputs: [{ id: "out_0", label: "Video", type: "video" }]
+	},
 	"depth-estimation": {
 		inputs: [{ id: "in_0", label: "Image", type: "image" }],
 		outputs: [{ id: "out_0", label: "Depth", type: "image" }]
@@ -274,6 +281,7 @@ export const PIPELINE_TO_CATEGORY: Record<string, string> = {
 	// Video
 	"text-to-video": "video",
 	"image-to-video": "video",
+	"image-text-to-video": "video",
 	"video-classification": "video",
 	// 3D
 	"text-to-3d": "3d",
diff --git a/js/workflowcanvas/workflow/workflow-executor.ts b/js/workflowcanvas/workflow/workflow-executor.ts
index 1344bcf156d..17f6883f647 100644
--- a/js/workflowcanvas/workflow/workflow-executor.ts
+++ b/js/workflowcanvas/workflow/workflow-executor.ts
@@ -424,14 +424,24 @@ export async function executeWorkflow(
 					if (!serverCallModel) {
 						throw new Error("Model call function not available");
 					}
+					// Custom-port values need names — pack as a keyed dict so
+					// the backend's dict-args branch can pass them as kwargs.
+					const hasCustomPorts = node.inputs.some((p) => p.custom);
+					const modelArgs = hasCustomPorts
+						? (Object.fromEntries(
+								node.inputs.map((port, i) => [port.id, args[i]])
+							) as unknown)
+						: args;
 					// Prefer browser-side streaming for chat-completion-compatible
-					// text tasks so the UI receives tokens as they arrive. The
-					// Python path stays for every other task.
+					// text tasks so the UI receives tokens as they arrive. Skip
+					// streaming when there are custom ports — the streaming path
+					// only sends the prompt and would drop the extras.
 					const tag = node.pipeline_tag ?? "text-generation";
 					const streamable =
 						(tag === "text-generation" ||
 							tag === "text2text-generation" ||
 							tag === "conversational") &&
+						!hasCustomPorts &&
 						!!stream_text_generation;
 					if (streamable) {
 						const prompt =
@@ -452,7 +462,7 @@ export async function executeWorkflow(
 							serverCallModel(
 								node.model_id,
 								tag,
-								JSON.stringify(args),
+								JSON.stringify(modelArgs),
 								node.provider
 							),
 							new Promise<never>((_, reject) =>
diff --git a/js/workflowcanvas/workflow/workflow-store.test.ts b/js/workflowcanvas/workflow/workflow-store.test.ts
index 75874a974f3..307a24e387f 100644
--- a/js/workflowcanvas/workflow/workflow-store.test.ts
+++ b/js/workflowcanvas/workflow/workflow-store.test.ts
@@ -4,6 +4,9 @@ import {
 	workflow,
 	addOperator,
 	addEdge,
+	add_custom_port,
+	remove_custom_port,
+	init_model_node_ports,
 	replaceNodeSource,
 	removeNode,
 	removeEdge,
@@ -755,3 +758,156 @@ describe("sanitize_for_save", () => {
 		expect(JSON.stringify(original)).toBe(snapshot);
 	});
 });
+
+describe("add_custom_port / remove_custom_port", () => {
+	beforeEach(resetWorkflow);
+
+	function _modelOp(): string {
+		return addOperator(
+			{
+				kind: "model",
+				label: "Model",
+				model_id: "org/model",
+				endpoint: "text_to_image",
+				endpoints: [
+					{
+						name: "text_to_image",
+						inputs: [{ id: "prompt", label: "Prompt", type: "text" }],
+						outputs: [{ id: "out_0", label: "Image", type: "image" }]
+					}
+				],
+				inputs: [{ id: "prompt", label: "Prompt", type: "text" }],
+				outputs: [{ id: "out_0", label: "Image", type: "image" }],
+				width: 220,
+				height: 90
+			} as unknown as Omit<OperatorNode, "id" | "role" | "x" | "y" | "data">,
+			0,
+			0
+		);
+	}
+
+	test("adds a custom port marked with custom: true", () => {
+		const id = _modelOp();
+		add_custom_port(id, {
+			id: "strength",
+			label: "Strength",
+			type: "number",
+			required: false
+		});
+		const node = get(workflow).operators.find((n) => n.id === id)!;
+		expect(node.inputs).toHaveLength(2);
+		expect(node.inputs[1]).toMatchObject({
+			id: "strength",
+			custom: true,
+			required: false
+		});
+	});
+
+	test("removes a custom port and any edges connected to it", () => {
+		const id = _modelOp();
+		add_custom_port(id, {
+			id: "strength",
+			label: "Strength",
+			type: "number",
+			required: false
+		});
+		const refId = addOperator(
+			{
+				kind: "space",
+				label: "src",
+				inputs: [],
+				outputs: [{ id: "out", label: "n", type: "number" }],
+				width: 100,
+				height: 80
+			} as unknown as Omit<OperatorNode, "id" | "role" | "x" | "y" | "data">,
+			0,
+			0
+		);
+		addEdge({
+			from_node_id: refId,
+			from_port_id: "out",
+			to_node_id: id,
+			to_port_id: "strength",
+			type: "number"
+		});
+		expect(get(workflow).edges).toHaveLength(1);
+		remove_custom_port(id, "strength");
+		const node = get(workflow).operators.find((n) => n.id === id)!;
+		expect(node.inputs).toHaveLength(1);
+		expect(get(workflow).edges).toHaveLength(0);
+	});
+
+	test("remove_custom_port leaves non-custom ports alone", () => {
+		const id = _modelOp();
+		remove_custom_port(id, "prompt"); // "prompt" is a schema port, not custom
+		const node = get(workflow).operators.find((n) => n.id === id)!;
+		expect(node.inputs).toHaveLength(1);
+		expect(node.inputs[0].id).toBe("prompt");
+	});
+
+	test("init_model_node_ports preserves custom ports across schema refresh", () => {
+		const id = _modelOp();
+		add_custom_port(id, {
+			id: "strength",
+			label: "Strength",
+			type: "number",
+			required: false
+		});
+		// Simulate a schema refresh that would normally reset inputs.
+		init_model_node_ports([
+			{
+				name: "text_to_image",
+				inputs: [
+					{ id: "prompt", label: "Prompt", type: "text" },
+					{ id: "seed", label: "Seed", type: "number", required: false }
+				],
+				outputs: [{ id: "out_0", label: "Image", type: "image" }]
+			}
+		]);
+		const node = get(workflow).operators.find((n) => n.id === id)!;
+		const ids = node.inputs.map((p) => p.id);
+		expect(ids).toContain("prompt");
+		expect(ids).toContain("seed");
+		expect(ids).toContain("strength"); // custom port survived
+		expect(node.inputs.find((p) => p.id === "strength")?.custom).toBe(true);
+	});
+
+	test("switch_endpoint preserves custom ports across endpoint change", () => {
+		const id = addOperator(
+			{
+				kind: "space",
+				label: "Model",
+				endpoint: "a",
+				endpoints: [
+					{
+						name: "a",
+						inputs: [{ id: "in_a", label: "A", type: "text" }],
+						outputs: [{ id: "out_0", label: "Out", type: "text" }]
+					},
+					{
+						name: "b",
+						inputs: [{ id: "in_b", label: "B", type: "text" }],
+						outputs: [{ id: "out_0", label: "Out", type: "text" }]
+					}
+				],
+				inputs: [{ id: "in_a", label: "A", type: "text" }],
+				outputs: [{ id: "out_0", label: "Out", type: "text" }],
+				width: 220,
+				height: 90
+			} as unknown as Omit<OperatorNode, "id" | "role" | "x" | "y" | "data">,
+			0,
+			0
+		);
+		add_custom_port(id, {
+			id: "strength",
+			label: "Strength",
+			type: "number",
+			required: false
+		});
+		switch_endpoint(id, "b");
+		const node = get(workflow).operators.find((n) => n.id === id)!;
+		const ids = node.inputs.map((p) => p.id);
+		expect(ids).toContain("in_b");
+		expect(ids).toContain("strength"); // custom port survived endpoint switch
+	});
+});
diff --git a/js/workflowcanvas/workflow/workflow-store.ts b/js/workflowcanvas/workflow/workflow-store.ts
index 9b504e40b8f..a482889e707 100644
--- a/js/workflowcanvas/workflow/workflow-store.ts
+++ b/js/workflowcanvas/workflow/workflow-store.ts
@@ -439,25 +439,70 @@ export function init_model_node_ports(
 			if (!endpointName) return n;
 			const sig = schemas.find((s) => s.name === endpointName);
 			if (!sig) return { ...n, endpoints: schemas };
+			const schemaIds = new Set(sig.inputs.map((p) => p.id));
+			const customPorts = (n.inputs ?? []).filter(
+				(p) => p.custom && !schemaIds.has(p.id)
+			);
 			return {
 				...n,
 				endpoint: endpointName,
 				endpoints: schemas,
-				inputs: sig.inputs,
+				inputs: [...sig.inputs, ...customPorts],
 				outputs: sig.outputs
 			};
 		})
 	}));
 }
 
+export function add_custom_port(nodeId: string, port: Port): void {
+	workflow.update((wf) => {
+		const node = wf.operators.find((n) => n.id === nodeId);
+		if (!node || node.inputs.some((p) => p.id === port.id)) return wf;
+		return {
+			...wf,
+			operators: wf.operators.map((n) =>
+				n.id === nodeId
+					? { ...n, inputs: [...n.inputs, { ...port, custom: true }] }
+					: n
+			)
+		};
+	});
+}
+
+export function remove_custom_port(nodeId: string, portId: string): void {
+	workflow.update((wf) => {
+		const node = wf.operators.find((n) => n.id === nodeId);
+		if (!node?.inputs.some((p) => p.id === portId && p.custom)) return wf;
+		return reconcileComponentRoles({
+			...wf,
+			operators: wf.operators.map((n) =>
+				n.id === nodeId
+					? {
+							...n,
+							inputs: n.inputs.filter((p) => !(p.id === portId && p.custom))
+						}
+					: n
+			),
+			edges: wf.edges.filter(
+				(e) => !(e.to_node_id === nodeId && e.to_port_id === portId)
+			)
+		});
+	});
+}
+
 export function switch_endpoint(nodeId: string, endpointName: string): void {
 	workflow.update((wf) => {
 		const node = wf.operators.find((n) => n.id === nodeId);
 		if (!node || !node.endpoints) return wf;
 		const sig = node.endpoints.find((e) => e.name === endpointName);
 		if (!sig || sig.name === node.endpoint) return wf;
-
-		const input_by_id = new Map(sig.inputs.map((p) => [p.id, p]));
+		const schemaIds = new Set(sig.inputs.map((p) => p.id));
+		const customPorts = (node.inputs ?? []).filter(
+			(p) => p.custom && !schemaIds.has(p.id)
+		);
+		const input_by_id = new Map(
+			[...sig.inputs, ...customPorts].map((p) => [p.id, p])
+		);
 		const output_by_id = new Map(sig.outputs.map((p) => [p.id, p]));
 
 		// Switching endpoints prunes edges whose ports vanished, so a component
@@ -469,7 +514,7 @@ export function switch_endpoint(nodeId: string, endpointName: string): void {
 					? {
 							...n,
 							endpoint: sig.name,
-							inputs: sig.inputs,
+							inputs: [...sig.inputs, ...customPorts],
 							outputs: sig.outputs,
 							data: {}
 						}
diff --git a/js/workflowcanvas/workflow/workflow-types.ts b/js/workflowcanvas/workflow/workflow-types.ts
index e006eb90e62..821256a3c01 100644
--- a/js/workflowcanvas/workflow/workflow-types.ts
+++ b/js/workflowcanvas/workflow/workflow-types.ts
@@ -32,6 +32,7 @@ export interface Port {
 	output_index?: number;
 	choices?: string[];
 	multiselect?: boolean;
+	custom?: boolean;
 }
 
 export interface FileValue {

From 4f4b694d54b5bba8d69f7074da76a6985692c396 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Wed, 5 Aug 2026 17:31:49 +0100
Subject: [PATCH 02/10] tweak

---
 gradio/workflow.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 4b4ad25ad80..5a312271954 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -1093,9 +1093,7 @@ def call_model(
         args = json.loads(args_json)
         if isinstance(args, dict):
             endpoint = (
-                _PIPELINE_TAG_TO_ENDPOINT.get(pipeline_tag or "")
-                or pipeline_tag
-                or ""
+                _PIPELINE_TAG_TO_ENDPOINT.get(pipeline_tag or "") or pipeline_tag or ""
             )
             return _dispatch_model_endpoint(client, endpoint, args)
 

From da4244c177f1fbbc9dc36d041afa6155be325f87 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Wed, 5 Aug 2026 23:10:20 +0100
Subject: [PATCH 03/10] hf oauth tweak

---
 .../workflow/WorkflowCanvas.svelte              |  3 ++-
 js/workflowcanvas/workflow/hf-auth.svelte.ts    | 17 +++++++++++------
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/js/workflowcanvas/workflow/WorkflowCanvas.svelte b/js/workflowcanvas/workflow/WorkflowCanvas.svelte
index d4d05220edf..a2132100a34 100644
--- a/js/workflowcanvas/workflow/WorkflowCanvas.svelte
+++ b/js/workflowcanvas/workflow/WorkflowCanvas.svelte
@@ -144,12 +144,13 @@
 			!oauthHintShown &&
 			auth.isHFSpace &&
 			auth.writeAccessKnown &&
+			auth.oauthAvailableKnown &&
 			!auth.canWrite &&
 			!auth.oauthAvailable
 		) {
 			oauthHintShown = true;
 			showToast(
-				"Sign-in has not beed enabled on this Space. The author should add `hf_oauth: true` to the README so users can run workflows on their own inference quota, and authors can edit.",
+				"Sign-in has not been enabled on this Space. The author should add `hf_oauth: true` to the README so users can run workflows on their own inference quota, and authors can edit.",
 				0,
 				"warning"
 			);
diff --git a/js/workflowcanvas/workflow/hf-auth.svelte.ts b/js/workflowcanvas/workflow/hf-auth.svelte.ts
index 76775e4e380..e9c7f67aea6 100644
--- a/js/workflowcanvas/workflow/hf-auth.svelte.ts
+++ b/js/workflowcanvas/workflow/hf-auth.svelte.ts
@@ -106,6 +106,7 @@ export function createHFAuth(getServer: () => Record<string, any>) {
 	// Defaults to false so a broken "Sign in" button never flashes before the
 	// server confirms it — only shown once known to work.
 	let oauthAvailable = $state(false);
+	let oauthAvailableKnown = $state(false);
 
 	function clearIdentity(): void {
 		user = "";
@@ -146,13 +147,14 @@ export function createHFAuth(getServer: () => Record<string, any>) {
 		const s = getServer();
 		if (!s?.get_oauth_available) {
 			oauthAvailable = false;
-			return;
-		}
-		try {
-			oauthAvailable = (await s.get_oauth_available()) === "true";
-		} catch {
-			oauthAvailable = false;
+		} else {
+			try {
+				oauthAvailable = (await s.get_oauth_available()) === "true";
+			} catch {
+				oauthAvailable = false;
+			}
 		}
+		oauthAvailableKnown = true;
 	}
 
 	async function init(): Promise<void> {
@@ -301,6 +303,9 @@ export function createHFAuth(getServer: () => Record<string, any>) {
 		get oauthAvailable() {
 			return oauthAvailable;
 		},
+		get oauthAvailableKnown() {
+			return oauthAvailableKnown;
+		},
 		init,
 		setPAT,
 		signIn,

From b4ae3f6c74490ce1e4f34284ad46909190aab369 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Thu, 6 Aug 2026 14:52:50 +0100
Subject: [PATCH 04/10] add demos

---
 demo/workflow_vlm_chat/run.py        | 13 +++++++++++++
 demo/workflow_vlm_chat/workflow.json |  1 +
 scripts/copy_demos.py                |  2 ++
 3 files changed, 16 insertions(+)
 create mode 100644 demo/workflow_vlm_chat/run.py
 create mode 100644 demo/workflow_vlm_chat/workflow.json

diff --git a/demo/workflow_vlm_chat/run.py b/demo/workflow_vlm_chat/run.py
new file mode 100644
index 00000000000..dbfd3023a7e
--- /dev/null
+++ b/demo/workflow_vlm_chat/run.py
@@ -0,0 +1,13 @@
+"""Ask a vision-language model about an image.
+
+Drop in an image, type a question, run. The VLM answers in text. Uses the
+`chat_completion` schema so the workflow works with any image-text-to-text
+model regardless of which Inference Provider serves it.
+"""
+
+import gradio as gr
+
+demo = gr.Workflow(graph="workflow.json")
+
+if __name__ == "__main__":
+    demo.launch()
diff --git a/demo/workflow_vlm_chat/workflow.json b/demo/workflow_vlm_chat/workflow.json
new file mode 100644
index 00000000000..0ece66602ad
--- /dev/null
+++ b/demo/workflow_vlm_chat/workflow.json
@@ -0,0 +1 @@
+{"schema_version":"2","name":"Ask About an Image","runtime":{"default":"client"},"references":[{"label":"Text","inputs":[{"id":"in","label":"Text","type":"text"}],"outputs":[{"id":"out","label":"Text","type":"text"}],"width":220,"height":163,"asset_type":"text","role":"reference","id":"81cf0eab-18c0-4d14-a2ce-217e2939ff2f","x":78.4375,"y":358.40234375,"data":{"in":"","out":""}}],"operators":[{"id":"op-vlm","label":"Kimi-K3","role":"operator","kind":"model","model_id":"moonshotai/Kimi-K3","endpoint":"chat_completion","pipeline_tag":"image-text-to-text","inputs":[{"id":"image","label":"Image","type":"image","required":false},{"id":"text","label":"Prompt","type":"text","required":false}],"outputs":[{"id":"out_0","label":"Text","type":"text","output_index":0}],"width":260,"height":184,"x":380,"y":180,"data":{"out_0":"Looks like you're tucked in somewhere cozy — maybe a van or camper? I can see wood paneling, some overhead lights, and what looks like bedding. Gives off late-night, settled-in-for-the-evening vibes.\n\nWhat's up with you? Just chilling, or is something on your mind?","text":"what going on "}}],"subjects":[{"id":"subj-answer","label":"Answer","role":"subject","asset_type":"text","inputs":[{"id":"in","label":"Text","type":"text"}],"outputs":[{"id":"out","label":"Text","type":"text"}],"width":320,"height":182,"x":720,"y":180,"data":{"in":"Looks like you're tucked in somewhere cozy — maybe a van or camper? I can see wood paneling, some overhead lights, and what looks like bedding. Gives off late-night, settled-in-for-the-evening vibes.\n\nWhat's up with you? Just chilling, or is something on your mind?"}}],"edges":[{"id":"e3","from_node_id":"op-vlm","from_port_id":"out_0","to_node_id":"subj-answer","to_port_id":"in","type":"text"},{"from_node_id":"81cf0eab-18c0-4d14-a2ce-217e2939ff2f","from_port_id":"out","to_node_id":"op-vlm","to_port_id":"text","type":"text","id":"8e33634e-9bba-4457-8138-76dff0c5ef35"}],"view":{"default":"canvas"}}
\ No newline at end of file
diff --git a/scripts/copy_demos.py b/scripts/copy_demos.py
index f7e5502805a..54d18f94d8a 100644
--- a/scripts/copy_demos.py
+++ b/scripts/copy_demos.py
@@ -48,6 +48,8 @@ def copy_all_demos(source_dir: str, dest_dir: str):
         "stream_audio_out",
         "stream_frames",
         "video_component",
+        "workflow_image_generator_comparison",
+        "workflow_vlm_chat",
         "zip_files",
     ]
     for demo in demos_to_copy:

From c463e9a91cbc52c6e4c743765c354e9eb602d6ab Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Thu, 6 Aug 2026 14:54:53 +0100
Subject: [PATCH 05/10] add oauth to frontmatter

---
 scripts/upload_demo_to_space.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/upload_demo_to_space.py b/scripts/upload_demo_to_space.py
index 8d989b6c55d..42aa83721f5 100644
--- a/scripts/upload_demo_to_space.py
+++ b/scripts/upload_demo_to_space.py
@@ -66,6 +66,7 @@ def split_into_chunks(lst: list, n: int) -> list[list]:
             sdk_version: {gradio_version}
             app_file: run.py
             pinned: false
+            hf_oauth: true
             ---
             """
             readme.write_text(textwrap.dedent(readme_content))

From 40484cebc63a6a206ae6b965e631071032f29f4e Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Fri, 7 Aug 2026 14:05:12 +0100
Subject: [PATCH 06/10] image to text text prompt

---
 js/workflowcanvas/workflow/node-library.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/js/workflowcanvas/workflow/node-library.ts b/js/workflowcanvas/workflow/node-library.ts
index f29c6afed9d..ac0c48a212f 100644
--- a/js/workflowcanvas/workflow/node-library.ts
+++ b/js/workflowcanvas/workflow/node-library.ts
@@ -120,7 +120,10 @@ export const TASK_SCHEMAS: Record<string, TaskSchema> = {
 		outputs: [{ id: "out_0", label: "Segments", type: "json" }]
 	},
 	"image-to-text": {
-		inputs: [{ id: "in_0", label: "Image", type: "image" }],
+		inputs: [
+			{ id: "in_0", label: "Image", type: "image" },
+			{ id: "in_1", label: "Prompt", type: "text" }
+		],
 		outputs: [{ id: "out_0", label: "Text", type: "text" }]
 	},
 	"image-to-image": {

From ba449c3fe09d83250edad8bd738908cdb32f38da Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Mon, 10 Aug 2026 16:51:39 +0100
Subject: [PATCH 07/10] fix

---
 gradio/workflow.py | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 5a312271954..7670ffbb4eb 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -1033,7 +1033,34 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
             else:
                 raise
     else:
-        result = fn(**clean)
+        try:
+            result = fn(**clean)
+        except ValueError as exc:
+            # huggingface_hub's per-task client methods hardcode a task string,
+            # but a provider may register a model under a different task
+            # (e.g. fal-ai serves MiniMax-H3 as `image-text-to-video`, not
+            # `image-to-video`). Parse the mismatch out of the error and drive
+            # the same helper with the corrected task.
+            m = re.search(
+                r"is not supported for task (\S+) and provider (\S+)\. "
+                r"Supported task: ([^.\s]+)\.",
+                str(exc),
+            )
+            if not m:
+                raise
+            from huggingface_hub.inference._providers import get_provider_helper
+
+            helper = get_provider_helper(m.group(2), task=m.group(1), model=client.model)  # type: ignore[arg-type]
+            helper.task = m.group(3)
+            input_key = next(iter(inspect.signature(fn).parameters), "inputs")
+            req = helper.prepare_request(
+                inputs=clean.pop(input_key, None),
+                parameters=clean,
+                headers=client.headers,
+                model=client.model,
+                api_key=client.token,
+            )
+            result = helper.get_response(client._inner_post(req), req)
     ext = _ENDPOINT_OUTPUT_EXT.get(endpoint)
     if ext:
         return json.dumps([_save_tmp(result, ext)])

From 299511b215eec56c3a52d16f966bcc0ecd3fb0f1 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Mon, 10 Aug 2026 22:14:53 +0100
Subject: [PATCH 08/10] fal fix

---
 gradio/workflow.py                | 32 ++++++++----------
 gradio/workflow_provider_shims.py | 54 +++++++++++++++++++++++++++++++
 2 files changed, 67 insertions(+), 19 deletions(-)
 create mode 100644 gradio/workflow_provider_shims.py

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 7670ffbb4eb..5596017895f 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -32,6 +32,10 @@
 from gradio.context import Context
 from gradio.helpers import special_args as _special_args
 from gradio.oauth import OAuthProfile, OAuthToken
+from gradio.workflow_provider_shims import (
+    PROVIDER_TASK_MISMATCH_RE,
+    run_via_helper,
+)
 from gradio.route_utils import Request
 from gradio.utils import colab_check, get_space
 
@@ -1036,31 +1040,21 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
         try:
             result = fn(**clean)
         except ValueError as exc:
-            # huggingface_hub's per-task client methods hardcode a task string,
-            # but a provider may register a model under a different task
-            # (e.g. fal-ai serves MiniMax-H3 as `image-text-to-video`, not
-            # `image-to-video`). Parse the mismatch out of the error and drive
-            # the same helper with the corrected task.
-            m = re.search(
-                r"is not supported for task (\S+) and provider (\S+)\. "
-                r"Supported task: ([^.\s]+)\.",
-                str(exc),
-            )
+            m = PROVIDER_TASK_MISMATCH_RE.search(str(exc))
             if not m:
                 raise
             from huggingface_hub.inference._providers import get_provider_helper
 
             helper = get_provider_helper(m.group(2), task=m.group(1), model=client.model)  # type: ignore[arg-type]
             helper.task = m.group(3)
-            input_key = next(iter(inspect.signature(fn).parameters), "inputs")
-            req = helper.prepare_request(
-                inputs=clean.pop(input_key, None),
-                parameters=clean,
-                headers=client.headers,
-                model=client.model,
-                api_key=client.token,
-            )
-            result = helper.get_response(client._inner_post(req), req)
+            result = run_via_helper(client, helper, fn, endpoint, clean)
+        except KeyError:
+            if endpoint != "image_to_video":
+                raise
+            from huggingface_hub.inference._providers import get_provider_helper
+
+            helper = get_provider_helper(client.provider, task="image-to-video", model=client.model)
+            result = run_via_helper(client, helper, fn, endpoint, clean)
     ext = _ENDPOINT_OUTPUT_EXT.get(endpoint)
     if ext:
         return json.dumps([_save_tmp(result, ext)])
diff --git a/gradio/workflow_provider_shims.py b/gradio/workflow_provider_shims.py
new file mode 100644
index 00000000000..5bc4cfd8923
--- /dev/null
+++ b/gradio/workflow_provider_shims.py
@@ -0,0 +1,54 @@
+"""Shims for `huggingface_hub` inference-provider client fragility.
+
+The per-task client methods hardcode both the task string and the response key
+path, so any drift in a provider's registration or response shape breaks every
+caller until upstream cuts a release. Prefer fixing upstream over adding here.
+"""
+
+from __future__ import annotations
+
+import inspect
+import re
+
+import httpx
+
+
+# Raised by `TaskProviderHelper._prepare_mapping_info` when the client method's
+# hardcoded task doesn't match the model's provider registration.
+PROVIDER_TASK_MISMATCH_RE = re.compile(
+    r"is not supported for task (\S+) and provider (\S+)\. "
+    r"Supported task: ([^.\s]+)\."
+)
+
+
+def fal_ai_video_fallback(helper, response, request_params) -> bytes:
+    from huggingface_hub.inference._providers.fal_ai import FalAIQueueTask
+
+    output = FalAIQueueTask.get_response(helper, response, request_params)
+    d = output if isinstance(output, dict) else {}
+    videos = d.get("videos") if isinstance(d.get("videos"), list) else None
+    url = (
+        (d.get("video") or {}).get("url")
+        or d.get("video_url")
+        or (videos[0].get("url") if videos else None)
+    )
+    if not url:
+        raise ValueError(f"Unexpected fal-ai response shape: {output}")
+    return httpx.get(url, timeout=60).content
+
+
+def run_via_helper(client, helper, fn, endpoint: str, clean: dict):
+    from huggingface_hub.inference._providers.fal_ai import FalAIQueueTask
+
+    input_key = next(iter(inspect.signature(fn).parameters), "inputs")
+    req = helper.prepare_request(
+        inputs=clean.pop(input_key, None),
+        parameters=clean,
+        headers=client.headers,
+        model=client.model,
+        api_key=client.token,
+    )
+    response = client._inner_post(req)
+    if endpoint == "image_to_video" and isinstance(helper, FalAIQueueTask):
+        return fal_ai_video_fallback(helper, response, req)
+    return helper.get_response(response, req)

From 2a819904b5ceb252fb54dcbde57401f6d14f31d4 Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Mon, 10 Aug 2026 22:19:25 +0100
Subject: [PATCH 09/10] lint

---
 gradio/workflow.py                | 12 ++++++++----
 gradio/workflow_provider_shims.py |  1 -
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 5596017895f..99b40b5d0dd 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -32,12 +32,12 @@
 from gradio.context import Context
 from gradio.helpers import special_args as _special_args
 from gradio.oauth import OAuthProfile, OAuthToken
+from gradio.route_utils import Request
+from gradio.utils import colab_check, get_space
 from gradio.workflow_provider_shims import (
     PROVIDER_TASK_MISMATCH_RE,
     run_via_helper,
 )
-from gradio.route_utils import Request
-from gradio.utils import colab_check, get_space
 
 if TYPE_CHECKING:
     from gradio.workflow_api import WorkflowEndpointManager
@@ -1045,7 +1045,9 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
                 raise
             from huggingface_hub.inference._providers import get_provider_helper
 
-            helper = get_provider_helper(m.group(2), task=m.group(1), model=client.model)  # type: ignore[arg-type]
+            helper = get_provider_helper(
+                m.group(2), task=m.group(1), model=client.model
+            )  # type: ignore[arg-type]
             helper.task = m.group(3)
             result = run_via_helper(client, helper, fn, endpoint, clean)
         except KeyError:
@@ -1053,7 +1055,9 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
                 raise
             from huggingface_hub.inference._providers import get_provider_helper
 
-            helper = get_provider_helper(client.provider, task="image-to-video", model=client.model)
+            helper = get_provider_helper(
+                client.provider, task="image-to-video", model=client.model
+            )
             result = run_via_helper(client, helper, fn, endpoint, clean)
     ext = _ENDPOINT_OUTPUT_EXT.get(endpoint)
     if ext:
diff --git a/gradio/workflow_provider_shims.py b/gradio/workflow_provider_shims.py
index 5bc4cfd8923..8256a0694cb 100644
--- a/gradio/workflow_provider_shims.py
+++ b/gradio/workflow_provider_shims.py
@@ -12,7 +12,6 @@
 
 import httpx
 
-
 # Raised by `TaskProviderHelper._prepare_mapping_info` when the client method's
 # hardcoded task doesn't match the model's provider registration.
 PROVIDER_TASK_MISMATCH_RE = re.compile(

From 438c55ca3915ab19f6b1511a3277eb9e93f9e25e Mon Sep 17 00:00:00 2001
From: hannahblair <hannahblair@hotmail.co.uk>
Date: Mon, 10 Aug 2026 22:23:29 +0100
Subject: [PATCH 10/10] type

---
 gradio/workflow.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gradio/workflow.py b/gradio/workflow.py
index 99b40b5d0dd..e1b29d34353 100644
--- a/gradio/workflow.py
+++ b/gradio/workflow.py
@@ -1046,8 +1046,10 @@ def _dispatch_model_endpoint(client, endpoint: str, kwargs: dict) -> str:
             from huggingface_hub.inference._providers import get_provider_helper
 
             helper = get_provider_helper(
-                m.group(2), task=m.group(1), model=client.model
-            )  # type: ignore[arg-type]
+                m.group(2),  # ty: ignore[invalid-argument-type]
+                task=m.group(1),
+                model=client.model,
+            )
             helper.task = m.group(3)
             result = run_via_helper(client, helper, fn, endpoint, clean)
         except KeyError: