[Inference Providers] wavespeed supoort image-text-to-image type

packages/inference/src/lib/getProviderHelper.ts

@@ -181,6 +181,8 @@ export const PROVIDERS: Record<InferenceProvider, Partial<Record<InferenceTask,
 		"text-to-video": new Wavespeed.WavespeedAITextToVideoTask(),
 		"image-to-image": new Wavespeed.WavespeedAIImageToImageTask(),
 		"image-to-video": new Wavespeed.WavespeedAIImageToVideoTask(),
+		"image-text-to-image": new Wavespeed.WavespeedAIImageTextToImageTask(),
+		"image-text-to-video": new Wavespeed.WavespeedAIImageTextToVideoTask(),
 	},
 	"zai-org": {
 		conversational: new Zai.ZaiConversationalTask(),

packages/inference/src/providers/wavespeed.ts

@@ -2,6 +2,8 @@ import type { TextToImageArgs } from "../tasks/cv/textToImage.js";
 import type { ImageToImageArgs } from "../tasks/cv/imageToImage.js";
 import type { TextToVideoArgs } from "../tasks/cv/textToVideo.js";
 import type { ImageToVideoArgs } from "../tasks/cv/imageToVideo.js";
+import type { ImageTextToImageArgs } from "../tasks/cv/imageTextToImage.js";
+import type { ImageTextToVideoArgs } from "../tasks/cv/imageTextToVideo.js";
 import type { BodyParams, RequestArgs, UrlParams } from "../types.js";
 import { delay } from "../utils/delay.js";
 import { omit } from "../utils/omit.js";
@@ -11,6 +13,8 @@ import type {
 	TextToVideoTaskHelper,
 	ImageToImageTaskHelper,
 	ImageToVideoTaskHelper,
+	ImageTextToImageTaskHelper,
+	ImageTextToVideoTaskHelper,
 } from "./providerHelper.js";
 import { TaskProviderHelper } from "./providerHelper.js";
 import {
@@ -93,7 +97,14 @@ abstract class WavespeedAITask extends TaskProviderHelper {
 	}
 
 	preparePayload(
-		params: BodyParams<ImageToImageArgs | TextToImageArgs | TextToVideoArgs | ImageToVideoArgs>
+		params: BodyParams<
+			| ImageToImageArgs
+			| TextToImageArgs
+			| TextToVideoArgs
+			| ImageToVideoArgs
+			| ImageTextToImageArgs
+			| ImageTextToVideoArgs
+		>
 	): Record<string, unknown> {
 		const payload: Record<string, unknown> = {
 			...omit(params.args, ["inputs", "parameters"]),
@@ -221,3 +232,29 @@ export class WavespeedAIImageToVideoTask extends WavespeedAITask implements Imag
 		return { ...args, inputs: args.parameters?.prompt, image: base, images };
 	}
 }
+
+export class WavespeedAIImageTextToImageTask extends WavespeedAITask implements ImageTextToImageTaskHelper {
+	constructor() {
+		super(WAVESPEEDAI_API_BASE_URL);
+	}
+
+	async preparePayloadAsync(args: ImageTextToImageArgs): Promise<RequestArgs> {
+		const hasImages =
+			(args as { images?: unknown }).images ?? (args.parameters as Record<string, unknown> | undefined)?.images;
+		const { base, images } = await buildImagesField(args.inputs as Blob | ArrayBuffer, hasImages);
+		return { ...args, inputs: args.parameters?.prompt, image: base, images };
+	}
+}
+
+export class WavespeedAIImageTextToVideoTask extends WavespeedAITask implements ImageTextToVideoTaskHelper {
+	constructor() {
+		super(WAVESPEEDAI_API_BASE_URL);
+	}
+
+	async preparePayloadAsync(args: ImageTextToVideoArgs): Promise<RequestArgs> {
+		const hasImages =
+			(args as { images?: unknown }).images ?? (args.parameters as Record<string, unknown> | undefined)?.images;
+		const { base, images } = await buildImagesField(args.inputs as Blob | ArrayBuffer, hasImages);
+		return { ...args, inputs: args.parameters?.prompt, image: base, images };
+	}
+}

packages/inference/test/InferenceClient.spec.ts

@@ -2297,6 +2297,13 @@ describe.skip("InferenceClient", () => {
 			const client = new InferenceClient(env.HF_WAVESPEED_KEY ?? "dummy");
 
 			HARDCODED_MODEL_INFERENCE_MAPPING["wavespeed"] = {
+				"Qwen/Qwen-Image-Layered": {
+					provider: "wavespeed",
+					hfModelId: "Qwen/Qwen-Image-Layered",
+					providerId: "wavespeed-ai/qwen-image/layered",
+					status: "live",
+					task: "image-text-to-image",
+				},
 				"Qwen/Qwen-Image-Edit-2509": {
 					provider: "wavespeed",
 					hfModelId: "Qwen/Qwen-Image-Edit-2509",
@@ -2463,6 +2470,20 @@ describe.skip("InferenceClient", () => {
 				});
 				expect(res).toBeInstanceOf(Blob);
 			});
+			it(`imageTextToImage - Qwen/Qwen-Image-Layered`, async () => {
+				const res = await client.imageToImage({
+					model: "Qwen/Qwen-Image-Layered",
+					provider: "wavespeed",
+					inputs: new Blob([readTestFile("cheetah.png")], { type: "image/png" }),
+					parameters: {
+						prompt: "The leopard chases its prey",
+						guidance_scale: 5,
+						num_inference_steps: 30,
+						seed: -1,
+					},
+				});
+				expect(res).toBeInstanceOf(Blob);
+			});
 		},
 		TIMEOUT
 	);