add support for image-to-image models on Replicate

packages/inference/src/lib/getProviderHelper.ts

@@ -133,6 +133,7 @@ export const PROVIDERS: Record<InferenceProvider, Partial<Record<InferenceTask,
 		"text-generation": new OvhCloud.OvhCloudTextGenerationTask(),
 	},
 	replicate: {
+		"image-to-image": new Replicate.ReplicateImageToImageTask(),
 		"text-to-image": new Replicate.ReplicateTextToImageTask(),
 		"text-to-speech": new Replicate.ReplicateTextToSpeechTask(),
 		"text-to-video": new Replicate.ReplicateTextToVideoTask(),

packages/inference/src/providers/replicate.ts

@@ -18,7 +18,12 @@ import { InferenceOutputError } from "../lib/InferenceOutputError";
 import { isUrl } from "../lib/isUrl";
 import type { BodyParams, HeaderParams, UrlParams } from "../types";
 import { omit } from "../utils/omit";
-import { TaskProviderHelper, type TextToImageTaskHelper, type TextToVideoTaskHelper } from "./providerHelper";
+import {
+	TaskProviderHelper,
+	type TextToImageTaskHelper,
+	type TextToVideoTaskHelper,
+	type ImageToImageTaskHelper,
+} from "./providerHelper";
 export interface ReplicateOutput {
 	output?: string | string[];
 }
@@ -152,3 +157,20 @@ export class ReplicateTextToVideoTask extends ReplicateTask implements TextToVid
 		throw new InferenceOutputError("Expected { output: string }");
 	}
 }
+
+export class ReplicateImageToImageTask extends ReplicateTask implements ImageToImageTaskHelper {
+	override async getResponse(response: ReplicateOutput): Promise<Blob> {
+		if (
+			typeof response === "object" &&
+			!!response &&
+			"output" in response &&
+			typeof response.output === "string" &&
+			isUrl(response.output)
+		) {
+			const urlResponse = await fetch(response.output);
+			return await urlResponse.blob();
+		}
+
+		throw new InferenceOutputError("Expected { output: string }");
+	}
+}