@ai-sdk/replicate 0.0.0-64aae7dd-20260114144918

package/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2023 Vercel, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/README.md

@@ -0,0 +1,46 @@
+# AI SDK - Replicate Provider
+
+The **[Replicate provider](https://ai-sdk.dev/providers/ai-sdk-providers/replicate)** for the [AI SDK](https://ai-sdk.dev/docs) contains image model support for the Replicate API.
+
+## Setup
+
+The Replicate provider is available in the `@ai-sdk/replicate` module. You can install it with
+
+```bash
+npm i @ai-sdk/replicate
+```
+
+## Usage
+
+```ts
+import { replicate } from '@ai-sdk/replicate';
+import { generateImage } from 'ai';
+
+const { image } = await generateImage({
+  model: replicate.image('black-forest-labs/flux-schnell'),
+  prompt: 'The Loch Ness Monster getting a manicure',
+});
+
+const filename = `image-${Date.now()}.png`;
+fs.writeFileSync(filename, image.uint8Array);
+console.log(`Image saved to ${filename}`);
+```
+
+If you want to pass additional inputs to the model besides the prompt, use the `providerOptions.replicate` property:
+
+```ts
+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',
+    },
+  },
+});
+```
+
+## Documentation
+
+Please check out the **[Replicate provider](https://ai-sdk.dev/providers/ai-sdk-providers/replicate)** for more information.

package/dist/index.d.mts

@@ -0,0 +1,91 @@
+import { ImageModelV3, ProviderV3 } from '@ai-sdk/provider';
+import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
+import { Resolvable, FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
+
+type ReplicateImageModelId = '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' | 'black-forest-labs/flux-fill-pro' | 'black-forest-labs/flux-fill-dev' | 'black-forest-labs/flux-2-pro' | 'black-forest-labs/flux-2-dev' | (string & {});
+
+interface ReplicateImageModelConfig {
+    provider: string;
+    baseURL: string;
+    headers?: Resolvable<Record<string, string | undefined>>;
+    fetch?: FetchFunction;
+    _internal?: {
+        currentDate?: () => Date;
+    };
+}
+declare class ReplicateImageModel implements ImageModelV3 {
+    readonly modelId: ReplicateImageModelId;
+    private readonly config;
+    readonly specificationVersion = "v3";
+    get maxImagesPerCall(): number;
+    get provider(): string;
+    private get isFlux2Model();
+    constructor(modelId: ReplicateImageModelId, config: ReplicateImageModelConfig);
+    doGenerate({ prompt, n, aspectRatio, size, seed, providerOptions, headers, abortSignal, files, mask, }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV3['doGenerate']>>>;
+}
+/**
+ * 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.
+ */
+declare const replicateImageProviderOptionsSchema: _ai_sdk_provider_utils.LazySchema<{
+    [x: string]: unknown;
+    maxWaitTimeInSeconds?: number | null | undefined;
+    guidance_scale?: number | null | undefined;
+    num_inference_steps?: number | null | undefined;
+    negative_prompt?: string | null | undefined;
+    output_format?: "png" | "jpg" | "webp" | null | undefined;
+    output_quality?: number | null | undefined;
+    strength?: number | null | undefined;
+}>;
+type ReplicateImageProviderOptions = InferSchema<typeof replicateImageProviderOptionsSchema>;
+
+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;
+}
+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.
+ */
+declare function createReplicate(options?: ReplicateProviderSettings): ReplicateProvider;
+/**
+ * Default Replicate provider instance.
+ */
+declare const replicate: ReplicateProvider;
+
+declare const VERSION: string;
+
+export { type ReplicateImageProviderOptions, type ReplicateProvider, type ReplicateProviderSettings, VERSION, createReplicate, replicate };

package/dist/index.d.ts

@@ -0,0 +1,91 @@
+import { ImageModelV3, ProviderV3 } from '@ai-sdk/provider';
+import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
+import { Resolvable, FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
+
+type ReplicateImageModelId = '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' | 'black-forest-labs/flux-fill-pro' | 'black-forest-labs/flux-fill-dev' | 'black-forest-labs/flux-2-pro' | 'black-forest-labs/flux-2-dev' | (string & {});
+
+interface ReplicateImageModelConfig {
+    provider: string;
+    baseURL: string;
+    headers?: Resolvable<Record<string, string | undefined>>;
+    fetch?: FetchFunction;
+    _internal?: {
+        currentDate?: () => Date;
+    };
+}
+declare class ReplicateImageModel implements ImageModelV3 {
+    readonly modelId: ReplicateImageModelId;
+    private readonly config;
+    readonly specificationVersion = "v3";
+    get maxImagesPerCall(): number;
+    get provider(): string;
+    private get isFlux2Model();
+    constructor(modelId: ReplicateImageModelId, config: ReplicateImageModelConfig);
+    doGenerate({ prompt, n, aspectRatio, size, seed, providerOptions, headers, abortSignal, files, mask, }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV3['doGenerate']>>>;
+}
+/**
+ * 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.
+ */
+declare const replicateImageProviderOptionsSchema: _ai_sdk_provider_utils.LazySchema<{
+    [x: string]: unknown;
+    maxWaitTimeInSeconds?: number | null | undefined;
+    guidance_scale?: number | null | undefined;
+    num_inference_steps?: number | null | undefined;
+    negative_prompt?: string | null | undefined;
+    output_format?: "png" | "jpg" | "webp" | null | undefined;
+    output_quality?: number | null | undefined;
+    strength?: number | null | undefined;
+}>;
+type ReplicateImageProviderOptions = InferSchema<typeof replicateImageProviderOptionsSchema>;
+
+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;
+}
+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.
+ */
+declare function createReplicate(options?: ReplicateProviderSettings): ReplicateProvider;
+/**
+ * Default Replicate provider instance.
+ */
+declare const replicate: ReplicateProvider;
+
+declare const VERSION: string;
+
+export { type ReplicateImageProviderOptions, type ReplicateProvider, type ReplicateProviderSettings, VERSION, createReplicate, replicate };

package/dist/index.js

@@ -0,0 +1,280 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  VERSION: () => VERSION,
+  createReplicate: () => createReplicate,
+  replicate: () => replicate
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/replicate-provider.ts
+var import_provider = require("@ai-sdk/provider");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+
+// src/replicate-image-model.ts
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_v42 = require("zod/v4");
+
+// src/replicate-error.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_v4 = require("zod/v4");
+var replicateErrorSchema = import_v4.z.object({
+  detail: import_v4.z.string().optional(),
+  error: import_v4.z.string().optional()
+});
+var replicateFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
+  errorSchema: replicateErrorSchema,
+  errorToMessage: (error) => {
+    var _a, _b;
+    return (_b = (_a = error.detail) != null ? _a : error.error) != null ? _b : "Unknown Replicate error";
+  }
+});
+
+// src/replicate-image-model.ts
+var FLUX_2_MODEL_PATTERN = /^black-forest-labs\/flux-2-/;
+var MAX_FLUX_2_INPUT_IMAGES = 8;
+var ReplicateImageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v3";
+  }
+  get maxImagesPerCall() {
+    return this.isFlux2Model ? MAX_FLUX_2_INPUT_IMAGES : 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  get isFlux2Model() {
+    return FLUX_2_MODEL_PATTERN.test(this.modelId);
+  }
+  async doGenerate({
+    prompt,
+    n,
+    aspectRatio,
+    size,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal,
+    files,
+    mask
+  }) {
+    var _a, _b, _c;
+    const warnings = [];
+    const [modelId, version] = this.modelId.split(":");
+    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+    const replicateOptions = await (0, import_provider_utils2.parseProviderOptions)({
+      provider: "replicate",
+      providerOptions,
+      schema: replicateImageProviderOptionsSchema
+    });
+    let imageInputs = {};
+    if (files != null && files.length > 0) {
+      if (this.isFlux2Model) {
+        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] = (0, import_provider_utils2.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 {
+        imageInputs = { image: (0, import_provider_utils2.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."
+          });
+        }
+      }
+    }
+    let maskInput;
+    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 = (0, import_provider_utils2.convertImageModelFileToDataUri)(mask);
+      }
+    }
+    const { maxWaitTimeInSeconds, ...inputOptions } = replicateOptions != null ? replicateOptions : {};
+    const preferHeader = maxWaitTimeInSeconds != null ? { prefer: `wait=${maxWaitTimeInSeconds}` } : { prefer: "wait" };
+    const {
+      value: { output },
+      responseHeaders
+    } = await (0, import_provider_utils2.postJsonToApi)({
+      url: (
+        // different endpoints for versioned vs unversioned models:
+        version != null ? `${this.config.baseURL}/predictions` : `${this.config.baseURL}/models/${modelId}/predictions`
+      ),
+      headers: (0, import_provider_utils2.combineHeaders)(
+        await (0, import_provider_utils2.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: (0, import_provider_utils2.createJsonResponseHandler)(
+        replicateImageResponseSchema
+      ),
+      failedResponseHandler: replicateFailedResponseHandler,
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const outputArray = Array.isArray(output) ? output : [output];
+    const images = await Promise.all(
+      outputArray.map(async (url) => {
+        const { value: image } = await (0, import_provider_utils2.getFromApi)({
+          url,
+          successfulResponseHandler: (0, import_provider_utils2.createBinaryResponseHandler)(),
+          failedResponseHandler: replicateFailedResponseHandler,
+          abortSignal,
+          fetch: this.config.fetch
+        });
+        return image;
+      })
+    );
+    return {
+      images,
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
+  }
+};
+var replicateImageResponseSchema = import_v42.z.object({
+  output: import_v42.z.union([import_v42.z.array(import_v42.z.string()), import_v42.z.string()])
+});
+var replicateImageProviderOptionsSchema = (0, import_provider_utils2.lazySchema)(
+  () => (0, import_provider_utils2.zodSchema)(
+    import_v42.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: import_v42.z.number().positive().nullish(),
+      /**
+       * Guidance scale for classifier-free guidance.
+       * Higher values make the output more closely match the prompt.
+       */
+      guidance_scale: import_v42.z.number().nullish(),
+      /**
+       * Number of denoising steps. More steps = higher quality but slower.
+       */
+      num_inference_steps: import_v42.z.number().nullish(),
+      /**
+       * Negative prompt to guide what to avoid in the generation.
+       */
+      negative_prompt: import_v42.z.string().nullish(),
+      /**
+       * Output image format.
+       */
+      output_format: import_v42.z.enum(["png", "jpg", "webp"]).nullish(),
+      /**
+       * Output image quality (1-100). Only applies to jpg and webp.
+       */
+      output_quality: import_v42.z.number().min(1).max(100).nullish(),
+      /**
+       * Strength of the transformation for img2img (0-1).
+       * Lower values keep more of the original image.
+       */
+      strength: import_v42.z.number().min(0).max(1).nullish()
+    }).passthrough()
+  )
+);
+
+// src/version.ts
+var VERSION = true ? "0.0.0-64aae7dd-20260114144918" : "0.0.0-test";
+
+// src/replicate-provider.ts
+function createReplicate(options = {}) {
+  const createImageModel = (modelId) => {
+    var _a;
+    return new ReplicateImageModel(modelId, {
+      provider: "replicate",
+      baseURL: (_a = options.baseURL) != null ? _a : "https://api.replicate.com/v1",
+      headers: (0, import_provider_utils3.withUserAgentSuffix)(
+        {
+          Authorization: `Bearer ${(0, import_provider_utils3.loadApiKey)({
+            apiKey: options.apiToken,
+            environmentVariableName: "REPLICATE_API_TOKEN",
+            description: "Replicate"
+          })}`,
+          ...options.headers
+        },
+        `ai-sdk/replicate/${VERSION}`
+      ),
+      fetch: options.fetch
+    });
+  };
+  const embeddingModel = (modelId) => {
+    throw new import_provider.NoSuchModelError({
+      modelId,
+      modelType: "embeddingModel"
+    });
+  };
+  return {
+    specificationVersion: "v3",
+    image: createImageModel,
+    imageModel: createImageModel,
+    languageModel: (modelId) => {
+      throw new import_provider.NoSuchModelError({
+        modelId,
+        modelType: "languageModel"
+      });
+    },
+    embeddingModel,
+    textEmbeddingModel: embeddingModel
+  };
+}
+var replicate = createReplicate();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  VERSION,
+  createReplicate,
+  replicate
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/replicate-provider.ts","../src/replicate-image-model.ts","../src/replicate-error.ts","../src/version.ts"],"sourcesContent":["export { createReplicate, replicate } from './replicate-provider';\nexport type {\n  ReplicateProvider,\n  ReplicateProviderSettings,\n} from './replicate-provider';\nexport type { ReplicateImageProviderOptions } from './replicate-image-model';\nexport { VERSION } from './version';\n","import { NoSuchModelError, ProviderV3 } from '@ai-sdk/provider';\nimport type { FetchFunction } from '@ai-sdk/provider-utils';\nimport { loadApiKey, withUserAgentSuffix } from '@ai-sdk/provider-utils';\nimport { ReplicateImageModel } from './replicate-image-model';\nimport { ReplicateImageModelId } from './replicate-image-settings';\nimport { VERSION } from './version';\n\nexport interface ReplicateProviderSettings {\n  /**\nAPI token that is being send using the `Authorization` header.\nIt defaults to the `REPLICATE_API_TOKEN` environment variable.\n   */\n  apiToken?: string;\n\n  /**\nUse a different URL prefix for API calls, e.g. to use proxy servers.\nThe default prefix is `https://api.replicate.com/v1`.\n   */\n  baseURL?: string;\n\n  /**\nCustom headers to include in the requests.\n     */\n  headers?: Record<string, string>;\n\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n    */\n  fetch?: FetchFunction;\n}\n\nexport interface ReplicateProvider extends ProviderV3 {\n  /**\n   * Creates a Replicate image generation model.\n   */\n  image(modelId: ReplicateImageModelId): ReplicateImageModel;\n\n  /**\n   * Creates a Replicate image generation model.\n   */\n  imageModel(modelId: ReplicateImageModelId): ReplicateImageModel;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: string): never;\n}\n\n/**\n * Create a Replicate provider instance.\n */\nexport function createReplicate(\n  options: ReplicateProviderSettings = {},\n): ReplicateProvider {\n  const createImageModel = (modelId: ReplicateImageModelId) =>\n    new ReplicateImageModel(modelId, {\n      provider: 'replicate',\n      baseURL: options.baseURL ?? 'https://api.replicate.com/v1',\n      headers: withUserAgentSuffix(\n        {\n          Authorization: `Bearer ${loadApiKey({\n            apiKey: options.apiToken,\n            environmentVariableName: 'REPLICATE_API_TOKEN',\n            description: 'Replicate',\n          })}`,\n          ...options.headers,\n        },\n        `ai-sdk/replicate/${VERSION}`,\n      ),\n      fetch: options.fetch,\n    });\n\n  const embeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({\n      modelId,\n      modelType: 'embeddingModel',\n    });\n  };\n\n  return {\n    specificationVersion: 'v3' as const,\n    image: createImageModel,\n    imageModel: createImageModel,\n    languageModel: (modelId: string) => {\n      throw new NoSuchModelError({\n        modelId,\n        modelType: 'languageModel',\n      });\n    },\n    embeddingModel,\n    textEmbeddingModel: embeddingModel,\n  };\n}\n\n/**\n * Default Replicate provider instance.\n */\nexport const replicate = createReplicate();\n","import type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';\nimport type { Resolvable } from '@ai-sdk/provider-utils';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  FetchFunction,\n  getFromApi,\n  InferSchema,\n  lazySchema,\n  parseProviderOptions,\n  postJsonToApi,\n  resolve,\n  zodSchema,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { replicateFailedResponseHandler } from './replicate-error';\nimport { ReplicateImageModelId } from './replicate-image-settings';\n\ninterface ReplicateImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers?: Resolvable<Record<string, string | undefined>>;\n  fetch?: FetchFunction;\n  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\n// Flux-2 models support up to 8 input images with input_image, input_image_2, etc.\nconst FLUX_2_MODEL_PATTERN = /^black-forest-labs\\/flux-2-/;\nconst MAX_FLUX_2_INPUT_IMAGES = 8;\n\nexport class ReplicateImageModel implements ImageModelV3 {\n  readonly specificationVersion = 'v3';\n\n  get maxImagesPerCall(): number {\n    // Flux-2 models support up to 8 input images\n    return this.isFlux2Model ? MAX_FLUX_2_INPUT_IMAGES : 1;\n  }\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  private get isFlux2Model(): boolean {\n    return FLUX_2_MODEL_PATTERN.test(this.modelId);\n  }\n\n  constructor(\n    readonly modelId: ReplicateImageModelId,\n    private readonly config: ReplicateImageModelConfig,\n  ) {}\n\n  async doGenerate({\n    prompt,\n    n,\n    aspectRatio,\n    size,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n    files,\n    mask,\n  }: Parameters<ImageModelV3['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV3['doGenerate']>>\n  > {\n    const warnings: Array<SharedV3Warning> = [];\n\n    const [modelId, version] = this.modelId.split(':');\n\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n\n    // Parse provider options\n    const replicateOptions = await parseProviderOptions({\n      provider: 'replicate',\n      providerOptions,\n      schema: replicateImageProviderOptionsSchema,\n    });\n\n    // Handle image input from files\n    let imageInputs: Record<string, string> = {};\n    if (files != null && files.length > 0) {\n      if (this.isFlux2Model) {\n        // Flux-2 models use input_image, input_image_2, input_image_3, etc.\n        for (\n          let i = 0;\n          i < Math.min(files.length, MAX_FLUX_2_INPUT_IMAGES);\n          i++\n        ) {\n          const key = i === 0 ? 'input_image' : `input_image_${i + 1}`;\n          imageInputs[key] = convertImageModelFileToDataUri(files[i]);\n        }\n        if (files.length > MAX_FLUX_2_INPUT_IMAGES) {\n          warnings.push({\n            type: 'other',\n            message: `Flux-2 models support up to ${MAX_FLUX_2_INPUT_IMAGES} input images. Additional images are ignored.`,\n          });\n        }\n      } else {\n        // Other models use single 'image' parameter\n        imageInputs = { image: convertImageModelFileToDataUri(files[0]) };\n        if (files.length > 1) {\n          warnings.push({\n            type: 'other',\n            message:\n              'This Replicate model only supports a single input image. Additional images are ignored.',\n          });\n        }\n      }\n    }\n\n    // Handle mask input (not supported by Flux-2 models)\n    let maskInput: string | undefined;\n    if (mask != null) {\n      if (this.isFlux2Model) {\n        warnings.push({\n          type: 'other',\n          message:\n            'Flux-2 models do not support mask input. The mask will be ignored.',\n        });\n      } else {\n        maskInput = convertImageModelFileToDataUri(mask);\n      }\n    }\n\n    // Extract maxWaitTimeInSeconds from provider options and prepare the rest for the request body\n    const { maxWaitTimeInSeconds, ...inputOptions } = replicateOptions ?? {};\n\n    // Build the prefer header based on maxWaitTimeInSeconds:\n    // - undefined/null: use default sync wait (prefer: wait)\n    // - positive number: use custom wait duration (prefer: wait=N)\n    const preferHeader: Record<string, string> =\n      maxWaitTimeInSeconds != null\n        ? { prefer: `wait=${maxWaitTimeInSeconds}` }\n        : { prefer: 'wait' };\n\n    const {\n      value: { output },\n      responseHeaders,\n    } = await postJsonToApi({\n      url:\n        // different endpoints for versioned vs unversioned models:\n        version != null\n          ? `${this.config.baseURL}/predictions`\n          : `${this.config.baseURL}/models/${modelId}/predictions`,\n\n      headers: combineHeaders(\n        await resolve(this.config.headers),\n        headers,\n        preferHeader,\n      ),\n\n      body: {\n        input: {\n          prompt,\n          aspect_ratio: aspectRatio,\n          size,\n          seed,\n          num_outputs: n,\n          ...imageInputs,\n          ...(maskInput != null ? { mask: maskInput } : {}),\n          ...inputOptions,\n        },\n        // for versioned models, include the version in the body:\n        ...(version != null ? { version } : {}),\n      },\n\n      successfulResponseHandler: createJsonResponseHandler(\n        replicateImageResponseSchema,\n      ),\n      failedResponseHandler: replicateFailedResponseHandler,\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    // download the images:\n    const outputArray = Array.isArray(output) ? output : [output];\n    const images = await Promise.all(\n      outputArray.map(async url => {\n        const { value: image } = await getFromApi({\n          url,\n          successfulResponseHandler: createBinaryResponseHandler(),\n          failedResponseHandler: replicateFailedResponseHandler,\n          abortSignal,\n          fetch: this.config.fetch,\n        });\n        return image;\n      }),\n    );\n\n    return {\n      images,\n      warnings,\n      response: {\n        timestamp: currentDate,\n        modelId: this.modelId,\n        headers: responseHeaders,\n      },\n    };\n  }\n}\n\nconst replicateImageResponseSchema = z.object({\n  output: z.union([z.array(z.string()), z.string()]),\n});\n\n/**\n * Provider options schema for Replicate image generation.\n *\n * Note: Different Replicate models support different parameters.\n * This schema includes common parameters, but you can pass any\n * model-specific parameters through the passthrough.\n */\nexport const replicateImageProviderOptionsSchema = lazySchema(() =>\n  zodSchema(\n    z\n      .object({\n        /**\n         * Maximum time in seconds to wait for the prediction to complete in sync mode.\n         * By default, Replicate uses sync mode with a 60-second timeout.\n         *\n         * - When not specified: Uses default 60-second sync wait (`prefer: wait`)\n         * - When set to a positive number: Uses that duration (`prefer: wait=N`)\n         */\n        maxWaitTimeInSeconds: z.number().positive().nullish(),\n\n        /**\n         * Guidance scale for classifier-free guidance.\n         * Higher values make the output more closely match the prompt.\n         */\n        guidance_scale: z.number().nullish(),\n\n        /**\n         * Number of denoising steps. More steps = higher quality but slower.\n         */\n        num_inference_steps: z.number().nullish(),\n\n        /**\n         * Negative prompt to guide what to avoid in the generation.\n         */\n        negative_prompt: z.string().nullish(),\n\n        /**\n         * Output image format.\n         */\n        output_format: z.enum(['png', 'jpg', 'webp']).nullish(),\n\n        /**\n         * Output image quality (1-100). Only applies to jpg and webp.\n         */\n        output_quality: z.number().min(1).max(100).nullish(),\n\n        /**\n         * Strength of the transformation for img2img (0-1).\n         * Lower values keep more of the original image.\n         */\n        strength: z.number().min(0).max(1).nullish(),\n      })\n      .passthrough(),\n  ),\n);\n\nexport type ReplicateImageProviderOptions = InferSchema<\n  typeof replicateImageProviderOptionsSchema\n>;\n","import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\nconst replicateErrorSchema = z.object({\n  detail: z.string().optional(),\n  error: z.string().optional(),\n});\n\nexport const replicateFailedResponseHandler = createJsonErrorResponseHandler({\n  errorSchema: replicateErrorSchema,\n  errorToMessage: error =>\n    error.detail ?? error.error ?? 'Unknown Replicate error',\n});\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n  typeof __PACKAGE_VERSION__ !== 'undefined'\n    ? __PACKAGE_VERSION__\n    : '0.0.0-test';\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,sBAA6C;AAE7C,IAAAA,yBAAgD;;;ACAhD,IAAAC,yBAaO;AACP,IAAAC,aAAkB;;;AChBlB,4BAA+C;AAC/C,gBAAkB;AAElB,IAAM,uBAAuB,YAAE,OAAO;AAAA,EACpC,QAAQ,YAAE,OAAO,EAAE,SAAS;AAAA,EAC5B,OAAO,YAAE,OAAO,EAAE,SAAS;AAC7B,CAAC;AAEM,IAAM,qCAAiC,sDAA+B;AAAA,EAC3E,aAAa;AAAA,EACb,gBAAgB,WAAM;AAVxB;AAWI,6BAAM,WAAN,YAAgB,MAAM,UAAtB,YAA+B;AAAA;AACnC,CAAC;;;ADmBD,IAAM,uBAAuB;AAC7B,IAAM,0BAA0B;AAEzB,IAAM,sBAAN,MAAkD;AAAA,EAgBvD,YACW,SACQ,QACjB;AAFS;AACQ;AAjBnB,SAAS,uBAAuB;AAAA,EAkB7B;AAAA,EAhBH,IAAI,mBAA2B;AAE7B,WAAO,KAAK,eAAe,0BAA0B;AAAA,EACvD;AAAA,EAEA,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAY,eAAwB;AAClC,WAAO,qBAAqB,KAAK,KAAK,OAAO;AAAA,EAC/C;AAAA,EAOA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AApEJ;AAqEI,UAAM,WAAmC,CAAC;AAE1C,UAAM,CAAC,SAAS,OAAO,IAAI,KAAK,QAAQ,MAAM,GAAG;AAEjD,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AAGvE,UAAM,mBAAmB,UAAM,6CAAqB;AAAA,MAClD,UAAU;AAAA,MACV;AAAA,MACA,QAAQ;AAAA,IACV,CAAC;AAGD,QAAI,cAAsC,CAAC;AAC3C,QAAI,SAAS,QAAQ,MAAM,SAAS,GAAG;AACrC,UAAI,KAAK,cAAc;AAErB,iBACM,IAAI,GACR,IAAI,KAAK,IAAI,MAAM,QAAQ,uBAAuB,GAClD,KACA;AACA,gBAAM,MAAM,MAAM,IAAI,gBAAgB,eAAe,IAAI,CAAC;AAC1D,sBAAY,GAAG,QAAI,uDAA+B,MAAM,CAAC,CAAC;AAAA,QAC5D;AACA,YAAI,MAAM,SAAS,yBAAyB;AAC1C,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,SAAS,+BAA+B,uBAAuB;AAAA,UACjE,CAAC;AAAA,QACH;AAAA,MACF,OAAO;AAEL,sBAAc,EAAE,WAAO,uDAA+B,MAAM,CAAC,CAAC,EAAE;AAChE,YAAI,MAAM,SAAS,GAAG;AACpB,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,SACE;AAAA,UACJ,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAGA,QAAI;AACJ,QAAI,QAAQ,MAAM;AAChB,UAAI,KAAK,cAAc;AACrB,iBAAS,KAAK;AAAA,UACZ,MAAM;AAAA,UACN,SACE;AAAA,QACJ,CAAC;AAAA,MACH,OAAO;AACL,wBAAY,uDAA+B,IAAI;AAAA,MACjD;AAAA,IACF;AAGA,UAAM,EAAE,sBAAsB,GAAG,aAAa,IAAI,8CAAoB,CAAC;AAKvE,UAAM,eACJ,wBAAwB,OACpB,EAAE,QAAQ,QAAQ,oBAAoB,GAAG,IACzC,EAAE,QAAQ,OAAO;AAEvB,UAAM;AAAA,MACJ,OAAO,EAAE,OAAO;AAAA,MAChB;AAAA,IACF,IAAI,UAAM,sCAAc;AAAA,MACtB;AAAA;AAAA,QAEE,WAAW,OACP,GAAG,KAAK,OAAO,OAAO,iBACtB,GAAG,KAAK,OAAO,OAAO,WAAW,OAAO;AAAA;AAAA,MAE9C,aAAS;AAAA,QACP,UAAM,gCAAQ,KAAK,OAAO,OAAO;AAAA,QACjC;AAAA,QACA;AAAA,MACF;AAAA,MAEA,MAAM;AAAA,QACJ,OAAO;AAAA,UACL;AAAA,UACA,cAAc;AAAA,UACd;AAAA,UACA;AAAA,UACA,aAAa;AAAA,UACb,GAAG;AAAA,UACH,GAAI,aAAa,OAAO,EAAE,MAAM,UAAU,IAAI,CAAC;AAAA,UAC/C,GAAG;AAAA,QACL;AAAA;AAAA,QAEA,GAAI,WAAW,OAAO,EAAE,QAAQ,IAAI,CAAC;AAAA,MACvC;AAAA,MAEA,+BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA,uBAAuB;AAAA,MACvB;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAGD,UAAM,cAAc,MAAM,QAAQ,MAAM,IAAI,SAAS,CAAC,MAAM;AAC5D,UAAM,SAAS,MAAM,QAAQ;AAAA,MAC3B,YAAY,IAAI,OAAM,QAAO;AAC3B,cAAM,EAAE,OAAO,MAAM,IAAI,UAAM,mCAAW;AAAA,UACxC;AAAA,UACA,+BAA2B,oDAA4B;AAAA,UACvD,uBAAuB;AAAA,UACvB;AAAA,UACA,OAAO,KAAK,OAAO;AAAA,QACrB,CAAC;AACD,eAAO;AAAA,MACT,CAAC;AAAA,IACH;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA,UAAU;AAAA,QACR,WAAW;AAAA,QACX,SAAS,KAAK;AAAA,QACd,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAM,+BAA+B,aAAE,OAAO;AAAA,EAC5C,QAAQ,aAAE,MAAM,CAAC,aAAE,MAAM,aAAE,OAAO,CAAC,GAAG,aAAE,OAAO,CAAC,CAAC;AACnD,CAAC;AASM,IAAM,0CAAsC;AAAA,EAAW,UAC5D;AAAA,IACE,aACG,OAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAQN,sBAAsB,aAAE,OAAO,EAAE,SAAS,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA,MAMpD,gBAAgB,aAAE,OAAO,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA,MAKnC,qBAAqB,aAAE,OAAO,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA,MAKxC,iBAAiB,aAAE,OAAO,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA,MAKpC,eAAe,aAAE,KAAK,CAAC,OAAO,OAAO,MAAM,CAAC,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA,MAKtD,gBAAgB,aAAE,OAAO,EAAE,IAAI,CAAC,EAAE,IAAI,GAAG,EAAE,QAAQ;AAAA;AAAA;AAAA;AAAA;AAAA,MAMnD,UAAU,aAAE,OAAO,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,QAAQ;AAAA,IAC7C,CAAC,EACA,YAAY;AAAA,EACjB;AACF;;;AErQO,IAAM,UACX,OACI,kCACA;;;AH+CC,SAAS,gBACd,UAAqC,CAAC,GACnB;AACnB,QAAM,mBAAmB,CAAC,YAAgC;AAvD5D;AAwDI,eAAI,oBAAoB,SAAS;AAAA,MAC/B,UAAU;AAAA,MACV,UAAS,aAAQ,YAAR,YAAmB;AAAA,MAC5B,aAAS;AAAA,QACP;AAAA,UACE,eAAe,cAAU,mCAAW;AAAA,YAClC,QAAQ,QAAQ;AAAA,YAChB,yBAAyB;AAAA,YACzB,aAAa;AAAA,UACf,CAAC,CAAC;AAAA,UACF,GAAG,QAAQ;AAAA,QACb;AAAA,QACA,oBAAoB,OAAO;AAAA,MAC7B;AAAA,MACA,OAAO,QAAQ;AAAA,IACjB,CAAC;AAAA;AAEH,QAAM,iBAAiB,CAAC,YAAoB;AAC1C,UAAM,IAAI,iCAAiB;AAAA,MACzB;AAAA,MACA,WAAW;AAAA,IACb,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,OAAO;AAAA,IACP,YAAY;AAAA,IACZ,eAAe,CAAC,YAAoB;AAClC,YAAM,IAAI,iCAAiB;AAAA,QACzB;AAAA,QACA,WAAW;AAAA,MACb,CAAC;AAAA,IACH;AAAA,IACA;AAAA,IACA,oBAAoB;AAAA,EACtB;AACF;AAKO,IAAM,YAAY,gBAAgB;","names":["import_provider_utils","import_provider_utils","import_v4"]}