@ai-sdk/replicate 0.0.0-1c33ba03-20260114162300 → 0.0.0-4115c213-20260122152721

package/CHANGELOG.md

@@ -1,11 +1,40 @@
 # @ai-sdk/replicate
 
-## 0.0.0-1c33ba03-20260114162300
+## 0.0.0-4115c213-20260122152721
 
 ### Patch Changes
 
-- Updated dependencies [261c011]
-  - @ai-sdk/provider-utils@0.0.0-1c33ba03-20260114162300
+- 4caafb2: chore: excluded tests from src folder in npm package
+- Updated dependencies [4caafb2]
+  - @ai-sdk/provider@0.0.0-4115c213-20260122152721
+  - @ai-sdk/provider-utils@0.0.0-4115c213-20260122152721
+
+## 2.0.10
+
+### Patch Changes
+
+- 2b8369d: chore: add docs to package dist
+
+## 2.0.9
+
+### Patch Changes
+
+- 8dc54db: chore: add src folders to package bundle
+
+## 2.0.8
+
+### Patch Changes
+
+- Updated dependencies [5c090e7]
+  - @ai-sdk/provider@3.0.4
+  - @ai-sdk/provider-utils@4.0.8
+
+## 2.0.7
+
+### Patch Changes
+
+- Updated dependencies [46f46e4]
+  - @ai-sdk/provider-utils@4.0.7
 
 ## 2.0.6
 

package/dist/index.js

@@ -227,7 +227,7 @@ var replicateImageProviderOptionsSchema = (0, import_provider_utils2.lazySchema)
 );
 
 // src/version.ts
-var VERSION = true ? "0.0.0-1c33ba03-20260114162300" : "0.0.0-test";
+var VERSION = true ? "0.0.0-4115c213-20260122152721" : "0.0.0-test";
 
 // src/replicate-provider.ts
 function createReplicate(options = {}) {

package/dist/index.mjs

@@ -210,7 +210,7 @@ var replicateImageProviderOptionsSchema = lazySchema(
 );
 
 // src/version.ts
-var VERSION = true ? "0.0.0-1c33ba03-20260114162300" : "0.0.0-test";
+var VERSION = true ? "0.0.0-4115c213-20260122152721" : "0.0.0-test";
 
 // src/replicate-provider.ts
 function createReplicate(options = {}) {

package/docs/60-replicate.mdx

@@ -0,0 +1,241 @@
+---
+title: Replicate
+description: Learn how to use Replicate models with the AI SDK.
+---
+
+# Replicate Provider
+
+[Replicate](https://replicate.com/) is a platform for running open-source AI models.
+It is a popular choice for running image generation models.
+
+## Setup
+
+The Replicate provider is available via the `@ai-sdk/replicate` module. You can install it with
+
+<Tabs items={['pnpm', 'npm', 'yarn', 'bun']}>
+  <Tab>
+    <Snippet text="pnpm add @ai-sdk/replicate" dark />
+  </Tab>
+  <Tab>
+    <Snippet text="npm install @ai-sdk/replicate" dark />
+  </Tab>
+  <Tab>
+    <Snippet text="yarn add @ai-sdk/replicate" dark />
+  </Tab>
+
+  <Tab>
+    <Snippet text="bun add @ai-sdk/replicate" dark />
+  </Tab>
+</Tabs>
+
+## Provider Instance
+
+You can import the default provider instance `replicate` from `@ai-sdk/replicate`:
+
+```ts
+import { replicate } from '@ai-sdk/replicate';
+```
+
+If you need a customized setup, you can import `createReplicate` from `@ai-sdk/replicate`
+and create a provider instance with your settings:
+
+```ts
+import { createReplicate } from '@ai-sdk/replicate';
+
+const replicate = createReplicate({
+  apiToken: process.env.REPLICATE_API_TOKEN ?? '',
+});
+```
+
+You can use the following optional settings to customize the Replicate provider instance:
+
+- **baseURL** _string_
+
+  Use a different URL prefix for API calls, e.g. to use proxy servers.
+  The default prefix is `https://api.replicate.com/v1`.
+
+- **apiToken** _string_
+
+  API token that is being sent using the `Authorization` header. It defaults to
+  the `REPLICATE_API_TOKEN` environment variable.
+
+- **headers** _Record&lt;string,string&gt;_
+
+  Custom headers to include in the requests.
+
+- **fetch** _(input: RequestInfo, init?: RequestInit) => Promise&lt;Response&gt;_
+
+  Custom [fetch](https://developer.mozilla.org/en-US/docs/Web/API/fetch) implementation.
+
+## Image Models
+
+You can create Replicate image models using the `.image()` factory method.
+For more on image generation with the AI SDK see [generateImage()](/docs/reference/ai-sdk-core/generate-image).
+
+<Note>
+  Model support for `size` and other parameters varies by model. Check the
+  model's documentation on [Replicate](https://replicate.com/explore) for
+  supported options and additional parameters that can be passed via
+  `providerOptions.replicate`.
+</Note>
+
+### Supported Image Models
+
+The following image models are currently supported by the Replicate provider:
+
+- [black-forest-labs/flux-1.1-pro-ultra](https://replicate.com/black-forest-labs/flux-1.1-pro-ultra)
+- [black-forest-labs/flux-1.1-pro](https://replicate.com/black-forest-labs/flux-1.1-pro)
+- [black-forest-labs/flux-dev](https://replicate.com/black-forest-labs/flux-dev)
+- [black-forest-labs/flux-pro](https://replicate.com/black-forest-labs/flux-pro)
+- [black-forest-labs/flux-schnell](https://replicate.com/black-forest-labs/flux-schnell)
+- [bytedance/sdxl-lightning-4step](https://replicate.com/bytedance/sdxl-lightning-4step)
+- [fofr/aura-flow](https://replicate.com/fofr/aura-flow)
+- [fofr/latent-consistency-model](https://replicate.com/fofr/latent-consistency-model)
+- [fofr/realvisxl-v3-multi-controlnet-lora](https://replicate.com/fofr/realvisxl-v3-multi-controlnet-lora)
+- [fofr/sdxl-emoji](https://replicate.com/fofr/sdxl-emoji)
+- [fofr/sdxl-multi-controlnet-lora](https://replicate.com/fofr/sdxl-multi-controlnet-lora)
+- [ideogram-ai/ideogram-v2-turbo](https://replicate.com/ideogram-ai/ideogram-v2-turbo)
+- [ideogram-ai/ideogram-v2](https://replicate.com/ideogram-ai/ideogram-v2)
+- [lucataco/dreamshaper-xl-turbo](https://replicate.com/lucataco/dreamshaper-xl-turbo)
+- [lucataco/open-dalle-v1.1](https://replicate.com/lucataco/open-dalle-v1.1)
+- [lucataco/realvisxl-v2.0](https://replicate.com/lucataco/realvisxl-v2.0)
+- [lucataco/realvisxl2-lcm](https://replicate.com/lucataco/realvisxl2-lcm)
+- [luma/photon-flash](https://replicate.com/luma/photon-flash)
+- [luma/photon](https://replicate.com/luma/photon)
+- [nvidia/sana](https://replicate.com/nvidia/sana)
+- [playgroundai/playground-v2.5-1024px-aesthetic](https://replicate.com/playgroundai/playground-v2.5-1024px-aesthetic)
+- [recraft-ai/recraft-v3-svg](https://replicate.com/recraft-ai/recraft-v3-svg)
+- [recraft-ai/recraft-v3](https://replicate.com/recraft-ai/recraft-v3)
+- [stability-ai/stable-diffusion-3.5-large-turbo](https://replicate.com/stability-ai/stable-diffusion-3.5-large-turbo)
+- [stability-ai/stable-diffusion-3.5-large](https://replicate.com/stability-ai/stable-diffusion-3.5-large)
+- [stability-ai/stable-diffusion-3.5-medium](https://replicate.com/stability-ai/stable-diffusion-3.5-medium)
+- [tstramer/material-diffusion](https://replicate.com/tstramer/material-diffusion)
+
+You can also use [versioned models](https://replicate.com/docs/topics/models/versions).
+The id for versioned models is the Replicate model id followed by a colon and the version ID (`$modelId:$versionId`), e.g.
+`bytedance/sdxl-lightning-4step:5599ed30703defd1d160a25a63321b4dec97101d98b4674bcc56e41f62f35637`.
+
+<Note>
+  You can also pass any available Replicate model ID as a string if needed.
+</Note>
+
+### Basic Usage
+
+```ts
+import { replicate } from '@ai-sdk/replicate';
+import { generateImage } from 'ai';
+import { writeFile } from 'node:fs/promises';
+
+const { image } = await generateImage({
+  model: replicate.image('black-forest-labs/flux-schnell'),
+  prompt: 'The Loch Ness Monster getting a manicure',
+  aspectRatio: '16:9',
+});
+
+await writeFile('image.webp', image.uint8Array);
+
+console.log('Image saved as image.webp');
+```
+
+### Model-specific options
+
+```ts highlight="9-11"
+import { replicate } from '@ai-sdk/replicate';
+import { generateImage } from 'ai';
+
+const { image } = await generateImage({
+  model: replicate.image('recraft-ai/recraft-v3'),
+  prompt: 'The Loch Ness Monster getting a manicure',
+  size: '1365x1024',
+  providerOptions: {
+    replicate: {
+      style: 'realistic_image',
+    },
+  },
+});
+```
+
+### Versioned Models
+
+```ts
+import { replicate } from '@ai-sdk/replicate';
+import { generateImage } from 'ai';
+
+const { image } = await generateImage({
+  model: replicate.image(
+    'bytedance/sdxl-lightning-4step:5599ed30703defd1d160a25a63321b4dec97101d98b4674bcc56e41f62f35637',
+  ),
+  prompt: 'The Loch Ness Monster getting a manicure',
+});
+```
+
+### Image Editing
+
+Replicate supports image editing through various models. Pass input images via `prompt.images` to transform or edit existing images.
+
+#### Basic Image Editing
+
+Transform an existing image using text prompts:
+
+```ts
+const imageBuffer = readFileSync('./input-image.png');
+
+const { images } = await generateImage({
+  model: replicate.image('black-forest-labs/flux-kontext-dev'),
+  prompt: {
+    text: 'Turn the cat into a golden retriever dog',
+    images: [imageBuffer],
+  },
+  providerOptions: {
+    replicate: {
+      guidance_scale: 7.5,
+      num_inference_steps: 30,
+    },
+  },
+});
+```
+
+#### Inpainting with Mask
+
+Edit specific parts of an image using a mask. For FLUX Fill models, white areas in the mask indicate where the image should be edited:
+
+```ts
+const image = readFileSync('./input-image.png');
+const mask = readFileSync('./mask.png'); // White = inpaint, black = keep
+
+const { images } = await generateImage({
+  model: replicate.image('black-forest-labs/flux-fill-pro'),
+  prompt: {
+    text: 'A sunlit indoor lounge area with a pool containing a flamingo',
+    images: [image],
+    mask: mask,
+  },
+  providerOptions: {
+    replicate: {
+      guidance_scale: 7.5,
+      num_inference_steps: 30,
+    },
+  },
+});
+```
+
+<Note>
+  Input images can be provided as `Buffer`, `ArrayBuffer`, `Uint8Array`, or
+  base64-encoded strings. Different Replicate models have different parameter
+  names and capabilities — check the model's documentation on
+  [Replicate](https://replicate.com/explore) for details.
+</Note>
+
+### Provider Options
+
+Common provider options for image generation:
+
+- **maxWaitTimeInSeconds** _number_ - Maximum time in seconds to wait for the prediction to complete in sync mode. By default, Replicate uses [sync mode](https://replicate.com/docs/topics/predictions/create-a-prediction#timeout-duration) with a 60-second timeout. Set to a positive number to use a custom duration (e.g., `120` for 2 minutes). When not specified, uses the default 60-second wait.
+- **guidance_scale** _number_ - Guidance scale for classifier-free guidance. Higher values make the output more closely match the prompt.
+- **num_inference_steps** _number_ - Number of denoising steps. More steps = higher quality but slower.
+- **negative_prompt** _string_ - Negative prompt to guide what to avoid in the generation.
+- **output_format** _'png' | 'jpg' | 'webp'_ - Output image format.
+- **output_quality** _number (1-100)_ - Output image quality. Only applies to jpg and webp.
+- **strength** _number (0-1)_ - Strength of the transformation for img2img. Lower values keep more of the original image.
+
+For more details, see the [Replicate models page](https://replicate.com/explore).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/replicate",
-  "version": "0.0.0-1c33ba03-20260114162300",
+  "version": "0.0.0-4115c213-20260122152721",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -8,9 +8,18 @@
   "types": "./dist/index.d.ts",
   "files": [
     "dist/**/*",
+    "docs/**/*",
+    "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.test-d.ts",
+    "!src/**/__snapshots__",
+    "!src/**/__fixtures__",
     "CHANGELOG.md",
     "README.md"
   ],
+  "directories": {
+    "doc": "./docs"
+  },
   "exports": {
     "./package.json": "./package.json",
     ".": {
@@ -20,15 +29,15 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "3.0.3",
-    "@ai-sdk/provider-utils": "0.0.0-1c33ba03-20260114162300"
+    "@ai-sdk/provider": "0.0.0-4115c213-20260122152721",
+    "@ai-sdk/provider-utils": "0.0.0-4115c213-20260122152721"
   },
   "devDependencies": {
     "@types/node": "20.17.24",
     "tsup": "^8",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "1.0.1",
+    "@ai-sdk/test-server": "0.0.0-4115c213-20260122152721",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {
@@ -54,7 +63,7 @@
   "scripts": {
     "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
     "build:watch": "pnpm clean && tsup --watch",
-    "clean": "del-cli dist *.tsbuildinfo",
+    "clean": "del-cli dist docs *.tsbuildinfo",
     "lint": "eslint \"./**/*.ts*\"",
     "type-check": "tsc --build",
     "prettier-check": "prettier --check \"./**/*.ts*\"",

package/src/index.ts

@@ -0,0 +1,7 @@
+export { createReplicate, replicate } from './replicate-provider';
+export type {
+  ReplicateProvider,
+  ReplicateProviderSettings,
+} from './replicate-provider';
+export type { ReplicateImageProviderOptions } from './replicate-image-model';
+export { VERSION } from './version';

package/src/replicate-error.ts

@@ -0,0 +1,13 @@
+import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+const replicateErrorSchema = z.object({
+  detail: z.string().optional(),
+  error: z.string().optional(),
+});
+
+export const replicateFailedResponseHandler = createJsonErrorResponseHandler({
+  errorSchema: replicateErrorSchema,
+  errorToMessage: error =>
+    error.detail ?? error.error ?? 'Unknown Replicate error',
+});

package/src/replicate-image-model.ts

@@ -0,0 +1,268 @@
+import type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';
+import type { Resolvable } from '@ai-sdk/provider-utils';
+import {
+  combineHeaders,
+  convertImageModelFileToDataUri,
+  createBinaryResponseHandler,
+  createJsonResponseHandler,
+  FetchFunction,
+  getFromApi,
+  InferSchema,
+  lazySchema,
+  parseProviderOptions,
+  postJsonToApi,
+  resolve,
+  zodSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+import { replicateFailedResponseHandler } from './replicate-error';
+import { ReplicateImageModelId } from './replicate-image-settings';
+
+interface ReplicateImageModelConfig {
+  provider: string;
+  baseURL: string;
+  headers?: Resolvable<Record<string, string | undefined>>;
+  fetch?: FetchFunction;
+  _internal?: {
+    currentDate?: () => Date;
+  };
+}
+
+// Flux-2 models support up to 8 input images with input_image, input_image_2, etc.
+const FLUX_2_MODEL_PATTERN = /^black-forest-labs\/flux-2-/;
+const MAX_FLUX_2_INPUT_IMAGES = 8;
+
+export class ReplicateImageModel implements ImageModelV3 {
+  readonly specificationVersion = 'v3';
+
+  get maxImagesPerCall(): number {
+    // Flux-2 models support up to 8 input images
+    return this.isFlux2Model ? MAX_FLUX_2_INPUT_IMAGES : 1;
+  }
+
+  get provider(): string {
+    return this.config.provider;
+  }
+
+  private get isFlux2Model(): boolean {
+    return FLUX_2_MODEL_PATTERN.test(this.modelId);
+  }
+
+  constructor(
+    readonly modelId: ReplicateImageModelId,
+    private readonly config: ReplicateImageModelConfig,
+  ) {}
+
+  async doGenerate({
+    prompt,
+    n,
+    aspectRatio,
+    size,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal,
+    files,
+    mask,
+  }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<
+    Awaited<ReturnType<ImageModelV3['doGenerate']>>
+  > {
+    const warnings: Array<SharedV3Warning> = [];
+
+    const [modelId, version] = this.modelId.split(':');
+
+    const currentDate = this.config._internal?.currentDate?.() ?? new Date();
+
+    // Parse provider options
+    const replicateOptions = await parseProviderOptions({
+      provider: 'replicate',
+      providerOptions,
+      schema: replicateImageProviderOptionsSchema,
+    });
+
+    // Handle image input from files
+    let imageInputs: Record<string, string> = {};
+    if (files != null && files.length > 0) {
+      if (this.isFlux2Model) {
+        // Flux-2 models use input_image, input_image_2, input_image_3, etc.
+        for (
+          let i = 0;
+          i < Math.min(files.length, MAX_FLUX_2_INPUT_IMAGES);
+          i++
+        ) {
+          const key = i === 0 ? 'input_image' : `input_image_${i + 1}`;
+          imageInputs[key] = convertImageModelFileToDataUri(files[i]);
+        }
+        if (files.length > MAX_FLUX_2_INPUT_IMAGES) {
+          warnings.push({
+            type: 'other',
+            message: `Flux-2 models support up to ${MAX_FLUX_2_INPUT_IMAGES} input images. Additional images are ignored.`,
+          });
+        }
+      } else {
+        // Other models use single 'image' parameter
+        imageInputs = { image: convertImageModelFileToDataUri(files[0]) };
+        if (files.length > 1) {
+          warnings.push({
+            type: 'other',
+            message:
+              'This Replicate model only supports a single input image. Additional images are ignored.',
+          });
+        }
+      }
+    }
+
+    // Handle mask input (not supported by Flux-2 models)
+    let maskInput: string | undefined;
+    if (mask != null) {
+      if (this.isFlux2Model) {
+        warnings.push({
+          type: 'other',
+          message:
+            'Flux-2 models do not support mask input. The mask will be ignored.',
+        });
+      } else {
+        maskInput = convertImageModelFileToDataUri(mask);
+      }
+    }
+
+    // Extract maxWaitTimeInSeconds from provider options and prepare the rest for the request body
+    const { maxWaitTimeInSeconds, ...inputOptions } = replicateOptions ?? {};
+
+    // Build the prefer header based on maxWaitTimeInSeconds:
+    // - undefined/null: use default sync wait (prefer: wait)
+    // - positive number: use custom wait duration (prefer: wait=N)
+    const preferHeader: Record<string, string> =
+      maxWaitTimeInSeconds != null
+        ? { prefer: `wait=${maxWaitTimeInSeconds}` }
+        : { prefer: 'wait' };
+
+    const {
+      value: { output },
+      responseHeaders,
+    } = await postJsonToApi({
+      url:
+        // different endpoints for versioned vs unversioned models:
+        version != null
+          ? `${this.config.baseURL}/predictions`
+          : `${this.config.baseURL}/models/${modelId}/predictions`,
+
+      headers: combineHeaders(
+        await resolve(this.config.headers),
+        headers,
+        preferHeader,
+      ),
+
+      body: {
+        input: {
+          prompt,
+          aspect_ratio: aspectRatio,
+          size,
+          seed,
+          num_outputs: n,
+          ...imageInputs,
+          ...(maskInput != null ? { mask: maskInput } : {}),
+          ...inputOptions,
+        },
+        // for versioned models, include the version in the body:
+        ...(version != null ? { version } : {}),
+      },
+
+      successfulResponseHandler: createJsonResponseHandler(
+        replicateImageResponseSchema,
+      ),
+      failedResponseHandler: replicateFailedResponseHandler,
+      abortSignal,
+      fetch: this.config.fetch,
+    });
+
+    // download the images:
+    const outputArray = Array.isArray(output) ? output : [output];
+    const images = await Promise.all(
+      outputArray.map(async url => {
+        const { value: image } = await getFromApi({
+          url,
+          successfulResponseHandler: createBinaryResponseHandler(),
+          failedResponseHandler: replicateFailedResponseHandler,
+          abortSignal,
+          fetch: this.config.fetch,
+        });
+        return image;
+      }),
+    );
+
+    return {
+      images,
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders,
+      },
+    };
+  }
+}
+
+const replicateImageResponseSchema = z.object({
+  output: z.union([z.array(z.string()), z.string()]),
+});
+
+/**
+ * Provider options schema for Replicate image generation.
+ *
+ * Note: Different Replicate models support different parameters.
+ * This schema includes common parameters, but you can pass any
+ * model-specific parameters through the passthrough.
+ */
+export const replicateImageProviderOptionsSchema = lazySchema(() =>
+  zodSchema(
+    z
+      .object({
+        /**
+         * Maximum time in seconds to wait for the prediction to complete in sync mode.
+         * By default, Replicate uses sync mode with a 60-second timeout.
+         *
+         * - When not specified: Uses default 60-second sync wait (`prefer: wait`)
+         * - When set to a positive number: Uses that duration (`prefer: wait=N`)
+         */
+        maxWaitTimeInSeconds: z.number().positive().nullish(),
+
+        /**
+         * Guidance scale for classifier-free guidance.
+         * Higher values make the output more closely match the prompt.
+         */
+        guidance_scale: z.number().nullish(),
+
+        /**
+         * Number of denoising steps. More steps = higher quality but slower.
+         */
+        num_inference_steps: z.number().nullish(),
+
+        /**
+         * Negative prompt to guide what to avoid in the generation.
+         */
+        negative_prompt: z.string().nullish(),
+
+        /**
+         * Output image format.
+         */
+        output_format: z.enum(['png', 'jpg', 'webp']).nullish(),
+
+        /**
+         * Output image quality (1-100). Only applies to jpg and webp.
+         */
+        output_quality: z.number().min(1).max(100).nullish(),
+
+        /**
+         * Strength of the transformation for img2img (0-1).
+         * Lower values keep more of the original image.
+         */
+        strength: z.number().min(0).max(1).nullish(),
+      })
+      .passthrough(),
+  ),
+);
+
+export type ReplicateImageProviderOptions = InferSchema<
+  typeof replicateImageProviderOptionsSchema
+>;

package/src/replicate-image-settings.ts

@@ -0,0 +1,36 @@
+export type ReplicateImageModelId =
+  // Text-to-image models
+  | 'black-forest-labs/flux-1.1-pro'
+  | 'black-forest-labs/flux-1.1-pro-ultra'
+  | 'black-forest-labs/flux-dev'
+  | 'black-forest-labs/flux-pro'
+  | 'black-forest-labs/flux-schnell'
+  | 'bytedance/sdxl-lightning-4step'
+  | 'fofr/aura-flow'
+  | 'fofr/latent-consistency-model'
+  | 'fofr/realvisxl-v3-multi-controlnet-lora'
+  | 'fofr/sdxl-emoji'
+  | 'fofr/sdxl-multi-controlnet-lora'
+  | 'ideogram-ai/ideogram-v2'
+  | 'ideogram-ai/ideogram-v2-turbo'
+  | 'lucataco/dreamshaper-xl-turbo'
+  | 'lucataco/open-dalle-v1.1'
+  | 'lucataco/realvisxl-v2.0'
+  | 'lucataco/realvisxl2-lcm'
+  | 'luma/photon'
+  | 'luma/photon-flash'
+  | 'nvidia/sana'
+  | 'playgroundai/playground-v2.5-1024px-aesthetic'
+  | 'recraft-ai/recraft-v3'
+  | 'recraft-ai/recraft-v3-svg'
+  | 'stability-ai/stable-diffusion-3.5-large'
+  | 'stability-ai/stable-diffusion-3.5-large-turbo'
+  | 'stability-ai/stable-diffusion-3.5-medium'
+  | 'tstramer/material-diffusion'
+  // Inpainting and image editing models
+  | 'black-forest-labs/flux-fill-pro'
+  | 'black-forest-labs/flux-fill-dev'
+  // Flux-2 models (support up to 8 reference images)
+  | 'black-forest-labs/flux-2-pro'
+  | 'black-forest-labs/flux-2-dev'
+  | (string & {});

package/src/replicate-provider.ts

@@ -0,0 +1,99 @@
+import { NoSuchModelError, ProviderV3 } from '@ai-sdk/provider';
+import type { FetchFunction } from '@ai-sdk/provider-utils';
+import { loadApiKey, withUserAgentSuffix } from '@ai-sdk/provider-utils';
+import { ReplicateImageModel } from './replicate-image-model';
+import { ReplicateImageModelId } from './replicate-image-settings';
+import { VERSION } from './version';
+
+export interface ReplicateProviderSettings {
+  /**
+API token that is being send using the `Authorization` header.
+It defaults to the `REPLICATE_API_TOKEN` environment variable.
+   */
+  apiToken?: string;
+
+  /**
+Use a different URL prefix for API calls, e.g. to use proxy servers.
+The default prefix is `https://api.replicate.com/v1`.
+   */
+  baseURL?: string;
+
+  /**
+Custom headers to include in the requests.
+     */
+  headers?: Record<string, string>;
+
+  /**
+Custom fetch implementation. You can use it as a middleware to intercept requests,
+or to provide a custom fetch implementation for e.g. testing.
+    */
+  fetch?: FetchFunction;
+}
+
+export interface ReplicateProvider extends ProviderV3 {
+  /**
+   * Creates a Replicate image generation model.
+   */
+  image(modelId: ReplicateImageModelId): ReplicateImageModel;
+
+  /**
+   * Creates a Replicate image generation model.
+   */
+  imageModel(modelId: ReplicateImageModelId): ReplicateImageModel;
+
+  /**
+   * @deprecated Use `embeddingModel` instead.
+   */
+  textEmbeddingModel(modelId: string): never;
+}
+
+/**
+ * Create a Replicate provider instance.
+ */
+export function createReplicate(
+  options: ReplicateProviderSettings = {},
+): ReplicateProvider {
+  const createImageModel = (modelId: ReplicateImageModelId) =>
+    new ReplicateImageModel(modelId, {
+      provider: 'replicate',
+      baseURL: options.baseURL ?? 'https://api.replicate.com/v1',
+      headers: withUserAgentSuffix(
+        {
+          Authorization: `Bearer ${loadApiKey({
+            apiKey: options.apiToken,
+            environmentVariableName: 'REPLICATE_API_TOKEN',
+            description: 'Replicate',
+          })}`,
+          ...options.headers,
+        },
+        `ai-sdk/replicate/${VERSION}`,
+      ),
+      fetch: options.fetch,
+    });
+
+  const embeddingModel = (modelId: string) => {
+    throw new NoSuchModelError({
+      modelId,
+      modelType: 'embeddingModel',
+    });
+  };
+
+  return {
+    specificationVersion: 'v3' as const,
+    image: createImageModel,
+    imageModel: createImageModel,
+    languageModel: (modelId: string) => {
+      throw new NoSuchModelError({
+        modelId,
+        modelType: 'languageModel',
+      });
+    },
+    embeddingModel,
+    textEmbeddingModel: embeddingModel,
+  };
+}
+
+/**
+ * Default Replicate provider instance.
+ */
+export const replicate = createReplicate();

package/src/version.ts

@@ -0,0 +1,6 @@
+// Version string of this package injected at build time.
+declare const __PACKAGE_VERSION__: string | undefined;
+export const VERSION: string =
+  typeof __PACKAGE_VERSION__ !== 'undefined'
+    ? __PACKAGE_VERSION__
+    : '0.0.0-test';