@ai-sdk/replicate 0.0.1

package/.turbo/turbo-build.log

@@ -0,0 +1,21 @@
+
+> @ai-sdk/replicate@0.0.1 build /home/runner/work/ai/ai/packages/replicate
+> tsup
+
+CLI Building entry: src/index.ts
+CLI Using tsconfig: tsconfig.json
+CLI tsup v8.3.0
+CLI Using tsup config: /home/runner/work/ai/ai/packages/replicate/tsup.config.ts
+CLI Target: es2018
+CJS Build start
+ESM Build start
+ESM dist/index.mjs     3.13 KB
+ESM dist/index.mjs.map 7.22 KB
+ESM ⚡️ Build success in 74ms
+CJS dist/index.js     4.37 KB
+CJS dist/index.js.map 7.46 KB
+CJS ⚡️ Build success in 88ms
+DTS Build start
+DTS ⚡️ Build success in 18826ms
+DTS dist/index.d.ts  3.08 KB
+DTS dist/index.d.mts 3.08 KB

package/.turbo/turbo-clean.log

@@ -0,0 +1,4 @@
+
+> @ai-sdk/replicate@0.0.1 clean /home/runner/work/ai/ai/packages/replicate
+> rm -rf dist
+

package/CHANGELOG.md

@@ -0,0 +1,12 @@
+# @ai-sdk/replicate
+
+## 0.0.1
+
+### Patch Changes
+
+- 6636db6: feat (provider/replicate): add replicate image provider
+- 90fb95a: chore (provider-utils): switch to unified test server
+- Updated dependencies [90fb95a]
+- Updated dependencies [e6dfef4]
+- Updated dependencies [6636db6]
+  - @ai-sdk/provider-utils@2.0.7

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://sdk.vercel.ai/providers/ai-sdk-providers/replicate)** for the [AI SDK](https://sdk.vercel.ai/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 { experimental_generateImage as 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://sdk.vercel.ai/providers/ai-sdk-providers/replicate)** for more information.

package/dist/index.d.mts

@@ -0,0 +1,65 @@
+import { Resolvable, FetchFunction } from '@ai-sdk/provider-utils';
+import { ImageModelV1 } from '@ai-sdk/provider';
+
+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' | (string & {});
+interface ReplicateImageSettings {
+    /**
+  Override the maximum number of images per call (default 1)
+     */
+    maxImagesPerCall?: number;
+}
+
+interface ReplicateImageModelConfig {
+    provider: string;
+    baseURL: string;
+    headers?: Resolvable<Record<string, string | undefined>>;
+    fetch?: FetchFunction;
+}
+declare class ReplicateImageModel implements ImageModelV1 {
+    readonly specificationVersion = "v1";
+    readonly modelId: ReplicateImageModelId;
+    readonly settings: ReplicateImageSettings;
+    private readonly config;
+    get provider(): string;
+    get maxImagesPerCall(): number;
+    constructor(modelId: ReplicateImageModelId, settings: ReplicateImageSettings, config: ReplicateImageModelConfig);
+    doGenerate({ prompt, n, aspectRatio, size, seed, providerOptions, headers, abortSignal, }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV1['doGenerate']>>>;
+}
+
+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 {
+    /**
+     * Creates a Replicate image generation model.
+     */
+    image(modelId: ReplicateImageModelId, settings?: ReplicateImageSettings): ReplicateImageModel;
+}
+/**
+ * Create a Replicate provider instance.
+ */
+declare function createReplicate(options?: ReplicateProviderSettings): ReplicateProvider;
+/**
+ * Default Replicate provider instance.
+ */
+declare const replicate: ReplicateProvider;
+
+export { type ReplicateProvider, type ReplicateProviderSettings, createReplicate, replicate };

package/dist/index.d.ts

@@ -0,0 +1,65 @@
+import { Resolvable, FetchFunction } from '@ai-sdk/provider-utils';
+import { ImageModelV1 } from '@ai-sdk/provider';
+
+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' | (string & {});
+interface ReplicateImageSettings {
+    /**
+  Override the maximum number of images per call (default 1)
+     */
+    maxImagesPerCall?: number;
+}
+
+interface ReplicateImageModelConfig {
+    provider: string;
+    baseURL: string;
+    headers?: Resolvable<Record<string, string | undefined>>;
+    fetch?: FetchFunction;
+}
+declare class ReplicateImageModel implements ImageModelV1 {
+    readonly specificationVersion = "v1";
+    readonly modelId: ReplicateImageModelId;
+    readonly settings: ReplicateImageSettings;
+    private readonly config;
+    get provider(): string;
+    get maxImagesPerCall(): number;
+    constructor(modelId: ReplicateImageModelId, settings: ReplicateImageSettings, config: ReplicateImageModelConfig);
+    doGenerate({ prompt, n, aspectRatio, size, seed, providerOptions, headers, abortSignal, }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV1['doGenerate']>>>;
+}
+
+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 {
+    /**
+     * Creates a Replicate image generation model.
+     */
+    image(modelId: ReplicateImageModelId, settings?: ReplicateImageSettings): ReplicateImageModel;
+}
+/**
+ * Create a Replicate provider instance.
+ */
+declare function createReplicate(options?: ReplicateProviderSettings): ReplicateProvider;
+/**
+ * Default Replicate provider instance.
+ */
+declare const replicate: ReplicateProvider;
+
+export { type ReplicateProvider, type ReplicateProviderSettings, createReplicate, replicate };

package/dist/index.js

@@ -0,0 +1,142 @@
+"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, {
+  createReplicate: () => createReplicate,
+  replicate: () => replicate
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/replicate-provider.ts
+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_zod2 = require("zod");
+
+// src/replicate-error.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_zod = require("zod");
+var replicateErrorSchema = import_zod.z.object({
+  detail: import_zod.z.string().optional(),
+  error: import_zod.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 ReplicateImageModel = class {
+  constructor(modelId, settings, config) {
+    this.specificationVersion = "v1";
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 1;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    aspectRatio,
+    size,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    var _a;
+    const warnings = [];
+    const {
+      value: { output }
+    } = await (0, import_provider_utils2.postJsonToApi)({
+      url: `${this.config.baseURL}/models/${this.modelId}/predictions`,
+      headers: (0, import_provider_utils2.combineHeaders)(await (0, import_provider_utils2.resolve)(this.config.headers), headers, {
+        prefer: "wait"
+      }),
+      body: {
+        input: {
+          prompt,
+          aspect_ratio: aspectRatio,
+          size,
+          seed,
+          num_outputs: n,
+          ...(_a = providerOptions.replicate) != null ? _a : {}
+        }
+      },
+      failedResponseHandler: replicateFailedResponseHandler,
+      successfulResponseHandler: (0, import_provider_utils2.createJsonResponseHandler)(
+        replicateImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const outputArray = Array.isArray(output) ? output : [output];
+    const images = await Promise.all(
+      outputArray.map(async (url) => {
+        const response = await fetch(url);
+        return new Uint8Array(await response.arrayBuffer());
+      })
+    );
+    return { images, warnings };
+  }
+};
+var replicateImageResponseSchema = import_zod2.z.object({
+  output: import_zod2.z.union([import_zod2.z.array(import_zod2.z.string()), import_zod2.z.string()])
+});
+
+// src/replicate-provider.ts
+function createReplicate(options = {}) {
+  return {
+    image: (modelId, settings) => {
+      var _a;
+      return new ReplicateImageModel(modelId, settings != null ? settings : {}, {
+        provider: "replicate",
+        baseURL: (_a = options.baseURL) != null ? _a : "https://api.replicate.com/v1",
+        headers: {
+          Authorization: `Bearer ${(0, import_provider_utils3.loadApiKey)({
+            apiKey: options.apiToken,
+            environmentVariableName: "REPLICATE_API_TOKEN",
+            description: "Replicate"
+          })}`,
+          ...options.headers
+        },
+        fetch: options.fetch
+      });
+    }
+  };
+}
+var replicate = createReplicate();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  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"],"sourcesContent":["export { createReplicate, replicate } from './replicate-provider';\nexport type {\n  ReplicateProvider,\n  ReplicateProviderSettings,\n} from './replicate-provider';\n","import type { FetchFunction } from '@ai-sdk/provider-utils';\nimport { loadApiKey } from '@ai-sdk/provider-utils';\nimport { ReplicateImageModel } from './replicate-image-model';\nimport {\n  ReplicateImageModelId,\n  ReplicateImageSettings,\n} from './replicate-image-settings';\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 {\n  /**\n   * Creates a Replicate image generation model.\n   */\n  image(\n    modelId: ReplicateImageModelId,\n    settings?: ReplicateImageSettings,\n  ): ReplicateImageModel;\n}\n\n/**\n * Create a Replicate provider instance.\n */\nexport function createReplicate(\n  options: ReplicateProviderSettings = {},\n): ReplicateProvider {\n  return {\n    image: (\n      modelId: ReplicateImageModelId,\n      settings?: ReplicateImageSettings,\n    ) =>\n      new ReplicateImageModel(modelId, settings ?? {}, {\n        provider: 'replicate',\n        baseURL: options.baseURL ?? 'https://api.replicate.com/v1',\n        headers: {\n          Authorization: `Bearer ${loadApiKey({\n            apiKey: options.apiToken,\n            environmentVariableName: 'REPLICATE_API_TOKEN',\n            description: 'Replicate',\n          })}`,\n          ...options.headers,\n        },\n        fetch: options.fetch,\n      }),\n  };\n}\n\n/**\n * Default Replicate provider instance.\n */\nexport const replicate = createReplicate();\n","import type { ImageModelV1, ImageModelV1CallWarning } from '@ai-sdk/provider';\nimport type { Resolvable } from '@ai-sdk/provider-utils';\nimport {\n  FetchFunction,\n  combineHeaders,\n  createJsonResponseHandler,\n  postJsonToApi,\n  resolve,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport { replicateFailedResponseHandler } from './replicate-error';\nimport {\n  ReplicateImageModelId,\n  ReplicateImageSettings,\n} from './replicate-image-settings';\n\ninterface ReplicateImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers?: Resolvable<Record<string, string | undefined>>;\n  fetch?: FetchFunction;\n}\n\nexport class ReplicateImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  readonly modelId: ReplicateImageModelId;\n  readonly settings: ReplicateImageSettings;\n\n  private readonly config: ReplicateImageModelConfig;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  get maxImagesPerCall(): number {\n    return this.settings.maxImagesPerCall ?? 1;\n  }\n\n  constructor(\n    modelId: ReplicateImageModelId,\n    settings: ReplicateImageSettings,\n    config: ReplicateImageModelConfig,\n  ) {\n    this.modelId = modelId;\n    this.settings = settings;\n    this.config = config;\n  }\n\n  async doGenerate({\n    prompt,\n    n,\n    aspectRatio,\n    size,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const {\n      value: { output },\n    } = await postJsonToApi({\n      url: `${this.config.baseURL}/models/${this.modelId}/predictions`,\n      headers: combineHeaders(await resolve(this.config.headers), headers, {\n        prefer: 'wait',\n      }),\n      body: {\n        input: {\n          prompt,\n          aspect_ratio: aspectRatio,\n          size,\n          seed,\n          num_outputs: n,\n          ...(providerOptions.replicate ?? {}),\n        },\n      },\n      failedResponseHandler: replicateFailedResponseHandler,\n      successfulResponseHandler: createJsonResponseHandler(\n        replicateImageResponseSchema,\n      ),\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 response = await fetch(url);\n        return new Uint8Array(await response.arrayBuffer());\n      }),\n    );\n\n    return { images, warnings };\n  }\n}\n\nconst replicateImageResponseSchema = z.object({\n  output: z.union([z.array(z.string()), z.string()]),\n});\n","import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\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"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,yBAA2B;;;ACC3B,IAAAC,yBAMO;AACP,IAAAC,cAAkB;;;ACTlB,4BAA+C;AAC/C,iBAAkB;AAElB,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACpC,QAAQ,aAAE,OAAO,EAAE,SAAS;AAAA,EAC5B,OAAO,aAAE,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;;;ADWM,IAAM,sBAAN,MAAkD;AAAA,EAgBvD,YACE,SACA,UACA,QACA;AAnBF,SAAS,uBAAuB;AAoB9B,SAAK,UAAU;AACf,SAAK,WAAW;AAChB,SAAK,SAAS;AAAA,EAChB;AAAA,EAhBA,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AAnCjC;AAoCI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAYA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AA5DJ;AA6DI,UAAM,WAA2C,CAAC;AAElD,UAAM;AAAA,MACJ,OAAO,EAAE,OAAO;AAAA,IAClB,IAAI,UAAM,sCAAc;AAAA,MACtB,KAAK,GAAG,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO;AAAA,MAClD,aAAS,uCAAe,UAAM,gCAAQ,KAAK,OAAO,OAAO,GAAG,SAAS;AAAA,QACnE,QAAQ;AAAA,MACV,CAAC;AAAA,MACD,MAAM;AAAA,QACJ,OAAO;AAAA,UACL;AAAA,UACA,cAAc;AAAA,UACd;AAAA,UACA;AAAA,UACA,aAAa;AAAA,UACb,IAAI,qBAAgB,cAAhB,YAA6B,CAAC;AAAA,QACpC;AAAA,MACF;AAAA,MACA,uBAAuB;AAAA,MACvB,+BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;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,WAAW,MAAM,MAAM,GAAG;AAChC,eAAO,IAAI,WAAW,MAAM,SAAS,YAAY,CAAC;AAAA,MACpD,CAAC;AAAA,IACH;AAEA,WAAO,EAAE,QAAQ,SAAS;AAAA,EAC5B;AACF;AAEA,IAAM,+BAA+B,cAAE,OAAO;AAAA,EAC5C,QAAQ,cAAE,MAAM,CAAC,cAAE,MAAM,cAAE,OAAO,CAAC,GAAG,cAAE,OAAO,CAAC,CAAC;AACnD,CAAC;;;ADzDM,SAAS,gBACd,UAAqC,CAAC,GACnB;AACnB,SAAO;AAAA,IACL,OAAO,CACL,SACA,aACA;AArDN;AAsDM,iBAAI,oBAAoB,SAAS,8BAAY,CAAC,GAAG;AAAA,QAC/C,UAAU;AAAA,QACV,UAAS,aAAQ,YAAR,YAAmB;AAAA,QAC5B,SAAS;AAAA,UACP,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,OAAO,QAAQ;AAAA,MACjB,CAAC;AAAA;AAAA,EACL;AACF;AAKO,IAAM,YAAY,gBAAgB;","names":["import_provider_utils","import_provider_utils","import_zod"]}

package/dist/index.mjs

@@ -0,0 +1,119 @@
+// src/replicate-provider.ts
+import { loadApiKey } from "@ai-sdk/provider-utils";
+
+// src/replicate-image-model.ts
+import {
+  combineHeaders,
+  createJsonResponseHandler,
+  postJsonToApi,
+  resolve
+} from "@ai-sdk/provider-utils";
+import { z as z2 } from "zod";
+
+// src/replicate-error.ts
+import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
+import { z } from "zod";
+var replicateErrorSchema = z.object({
+  detail: z.string().optional(),
+  error: z.string().optional()
+});
+var replicateFailedResponseHandler = 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 ReplicateImageModel = class {
+  constructor(modelId, settings, config) {
+    this.specificationVersion = "v1";
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 1;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    aspectRatio,
+    size,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    var _a;
+    const warnings = [];
+    const {
+      value: { output }
+    } = await postJsonToApi({
+      url: `${this.config.baseURL}/models/${this.modelId}/predictions`,
+      headers: combineHeaders(await resolve(this.config.headers), headers, {
+        prefer: "wait"
+      }),
+      body: {
+        input: {
+          prompt,
+          aspect_ratio: aspectRatio,
+          size,
+          seed,
+          num_outputs: n,
+          ...(_a = providerOptions.replicate) != null ? _a : {}
+        }
+      },
+      failedResponseHandler: replicateFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        replicateImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    const outputArray = Array.isArray(output) ? output : [output];
+    const images = await Promise.all(
+      outputArray.map(async (url) => {
+        const response = await fetch(url);
+        return new Uint8Array(await response.arrayBuffer());
+      })
+    );
+    return { images, warnings };
+  }
+};
+var replicateImageResponseSchema = z2.object({
+  output: z2.union([z2.array(z2.string()), z2.string()])
+});
+
+// src/replicate-provider.ts
+function createReplicate(options = {}) {
+  return {
+    image: (modelId, settings) => {
+      var _a;
+      return new ReplicateImageModel(modelId, settings != null ? settings : {}, {
+        provider: "replicate",
+        baseURL: (_a = options.baseURL) != null ? _a : "https://api.replicate.com/v1",
+        headers: {
+          Authorization: `Bearer ${loadApiKey({
+            apiKey: options.apiToken,
+            environmentVariableName: "REPLICATE_API_TOKEN",
+            description: "Replicate"
+          })}`,
+          ...options.headers
+        },
+        fetch: options.fetch
+      });
+    }
+  };
+}
+var replicate = createReplicate();
+export {
+  createReplicate,
+  replicate
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/replicate-provider.ts","../src/replicate-image-model.ts","../src/replicate-error.ts"],"sourcesContent":["import type { FetchFunction } from '@ai-sdk/provider-utils';\nimport { loadApiKey } from '@ai-sdk/provider-utils';\nimport { ReplicateImageModel } from './replicate-image-model';\nimport {\n  ReplicateImageModelId,\n  ReplicateImageSettings,\n} from './replicate-image-settings';\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 {\n  /**\n   * Creates a Replicate image generation model.\n   */\n  image(\n    modelId: ReplicateImageModelId,\n    settings?: ReplicateImageSettings,\n  ): ReplicateImageModel;\n}\n\n/**\n * Create a Replicate provider instance.\n */\nexport function createReplicate(\n  options: ReplicateProviderSettings = {},\n): ReplicateProvider {\n  return {\n    image: (\n      modelId: ReplicateImageModelId,\n      settings?: ReplicateImageSettings,\n    ) =>\n      new ReplicateImageModel(modelId, settings ?? {}, {\n        provider: 'replicate',\n        baseURL: options.baseURL ?? 'https://api.replicate.com/v1',\n        headers: {\n          Authorization: `Bearer ${loadApiKey({\n            apiKey: options.apiToken,\n            environmentVariableName: 'REPLICATE_API_TOKEN',\n            description: 'Replicate',\n          })}`,\n          ...options.headers,\n        },\n        fetch: options.fetch,\n      }),\n  };\n}\n\n/**\n * Default Replicate provider instance.\n */\nexport const replicate = createReplicate();\n","import type { ImageModelV1, ImageModelV1CallWarning } from '@ai-sdk/provider';\nimport type { Resolvable } from '@ai-sdk/provider-utils';\nimport {\n  FetchFunction,\n  combineHeaders,\n  createJsonResponseHandler,\n  postJsonToApi,\n  resolve,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport { replicateFailedResponseHandler } from './replicate-error';\nimport {\n  ReplicateImageModelId,\n  ReplicateImageSettings,\n} from './replicate-image-settings';\n\ninterface ReplicateImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers?: Resolvable<Record<string, string | undefined>>;\n  fetch?: FetchFunction;\n}\n\nexport class ReplicateImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  readonly modelId: ReplicateImageModelId;\n  readonly settings: ReplicateImageSettings;\n\n  private readonly config: ReplicateImageModelConfig;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  get maxImagesPerCall(): number {\n    return this.settings.maxImagesPerCall ?? 1;\n  }\n\n  constructor(\n    modelId: ReplicateImageModelId,\n    settings: ReplicateImageSettings,\n    config: ReplicateImageModelConfig,\n  ) {\n    this.modelId = modelId;\n    this.settings = settings;\n    this.config = config;\n  }\n\n  async doGenerate({\n    prompt,\n    n,\n    aspectRatio,\n    size,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const {\n      value: { output },\n    } = await postJsonToApi({\n      url: `${this.config.baseURL}/models/${this.modelId}/predictions`,\n      headers: combineHeaders(await resolve(this.config.headers), headers, {\n        prefer: 'wait',\n      }),\n      body: {\n        input: {\n          prompt,\n          aspect_ratio: aspectRatio,\n          size,\n          seed,\n          num_outputs: n,\n          ...(providerOptions.replicate ?? {}),\n        },\n      },\n      failedResponseHandler: replicateFailedResponseHandler,\n      successfulResponseHandler: createJsonResponseHandler(\n        replicateImageResponseSchema,\n      ),\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 response = await fetch(url);\n        return new Uint8Array(await response.arrayBuffer());\n      }),\n    );\n\n    return { images, warnings };\n  }\n}\n\nconst replicateImageResponseSchema = z.object({\n  output: z.union([z.array(z.string()), z.string()]),\n});\n","import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\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"],"mappings":";AACA,SAAS,kBAAkB;;;ACC3B;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,KAAAA,UAAS;;;ACTlB,SAAS,sCAAsC;AAC/C,SAAS,SAAS;AAElB,IAAM,uBAAuB,EAAE,OAAO;AAAA,EACpC,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA,EAC5B,OAAO,EAAE,OAAO,EAAE,SAAS;AAC7B,CAAC;AAEM,IAAM,iCAAiC,+BAA+B;AAAA,EAC3E,aAAa;AAAA,EACb,gBAAgB,WAAM;AAVxB;AAWI,6BAAM,WAAN,YAAgB,MAAM,UAAtB,YAA+B;AAAA;AACnC,CAAC;;;ADWM,IAAM,sBAAN,MAAkD;AAAA,EAgBvD,YACE,SACA,UACA,QACA;AAnBF,SAAS,uBAAuB;AAoB9B,SAAK,UAAU;AACf,SAAK,WAAW;AAChB,SAAK,SAAS;AAAA,EAChB;AAAA,EAhBA,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AAnCjC;AAoCI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAYA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AA5DJ;AA6DI,UAAM,WAA2C,CAAC;AAElD,UAAM;AAAA,MACJ,OAAO,EAAE,OAAO;AAAA,IAClB,IAAI,MAAM,cAAc;AAAA,MACtB,KAAK,GAAG,KAAK,OAAO,OAAO,WAAW,KAAK,OAAO;AAAA,MAClD,SAAS,eAAe,MAAM,QAAQ,KAAK,OAAO,OAAO,GAAG,SAAS;AAAA,QACnE,QAAQ;AAAA,MACV,CAAC;AAAA,MACD,MAAM;AAAA,QACJ,OAAO;AAAA,UACL;AAAA,UACA,cAAc;AAAA,UACd;AAAA,UACA;AAAA,UACA,aAAa;AAAA,UACb,IAAI,qBAAgB,cAAhB,YAA6B,CAAC;AAAA,QACpC;AAAA,MACF;AAAA,MACA,uBAAuB;AAAA,MACvB,2BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;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,WAAW,MAAM,MAAM,GAAG;AAChC,eAAO,IAAI,WAAW,MAAM,SAAS,YAAY,CAAC;AAAA,MACpD,CAAC;AAAA,IACH;AAEA,WAAO,EAAE,QAAQ,SAAS;AAAA,EAC5B;AACF;AAEA,IAAM,+BAA+BC,GAAE,OAAO;AAAA,EAC5C,QAAQA,GAAE,MAAM,CAACA,GAAE,MAAMA,GAAE,OAAO,CAAC,GAAGA,GAAE,OAAO,CAAC,CAAC;AACnD,CAAC;;;ADzDM,SAAS,gBACd,UAAqC,CAAC,GACnB;AACnB,SAAO;AAAA,IACL,OAAO,CACL,SACA,aACA;AArDN;AAsDM,iBAAI,oBAAoB,SAAS,8BAAY,CAAC,GAAG;AAAA,QAC/C,UAAU;AAAA,QACV,UAAS,aAAQ,YAAR,YAAmB;AAAA,QAC5B,SAAS;AAAA,UACP,eAAe,UAAU,WAAW;AAAA,YAClC,QAAQ,QAAQ;AAAA,YAChB,yBAAyB;AAAA,YACzB,aAAa;AAAA,UACf,CAAC,CAAC;AAAA,UACF,GAAG,QAAQ;AAAA,QACb;AAAA,QACA,OAAO,QAAQ;AAAA,MACjB,CAAC;AAAA;AAAA,EACL;AACF;AAKO,IAAM,YAAY,gBAAgB;","names":["z","z"]}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@ai-sdk/replicate",
+  "version": "0.0.1",
+  "license": "Apache-2.0",
+  "sideEffects": false,
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "@ai-sdk/provider": "1.0.4",
+    "@ai-sdk/provider-utils": "2.0.7"
+  },
+  "devDependencies": {
+    "@types/node": "^18",
+    "tsup": "^8",
+    "typescript": "5.6.3",
+    "zod": "3.23.8",
+    "@vercel/ai-tsconfig": "0.0.0"
+  },
+  "peerDependencies": {
+    "zod": "^3.0.0"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsup",
+    "build:watch": "tsup --watch",
+    "clean": "rm -rf dist",
+    "lint": "eslint \"./**/*.ts*\"",
+    "type-check": "tsc --noEmit",
+    "prettier-check": "prettier --check \"./**/*.ts*\"",
+    "test": "pnpm test:node && pnpm test:edge",
+    "test:edge": "vitest --config vitest.edge.config.js --run",
+    "test:node": "vitest --config vitest.node.config.js --run"
+  }
+}

package/src/index.ts

@@ -0,0 +1,5 @@
+export { createReplicate, replicate } from './replicate-provider';
+export type {
+  ReplicateProvider,
+  ReplicateProviderSettings,
+} from './replicate-provider';

package/src/replicate-error.ts

@@ -0,0 +1,13 @@
+import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';
+import { z } from 'zod';
+
+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.test.ts

@@ -0,0 +1,177 @@
+import { createTestServer } from '@ai-sdk/provider-utils/test';
+import { createReplicate } from './replicate-provider';
+
+const prompt = 'The Loch Ness monster getting a manicure';
+
+const provider = createReplicate({ apiToken: 'test-api-token' });
+const model = provider.image('black-forest-labs/flux-schnell');
+
+describe('doGenerate', () => {
+  const server = createTestServer({
+    'https://api.replicate.com/*': {},
+    'https://replicate.delivery/*': {
+      response: {
+        type: 'binary',
+        body: Buffer.from('test-binary-content'),
+      },
+    },
+  });
+
+  function prepareResponse({
+    output = ['https://replicate.delivery/xezq/abc/out-0.webp'],
+  }: { output?: string | Array<string> } = {}) {
+    server.urls['https://api.replicate.com/*'].response = {
+      type: 'json-value',
+      body: {
+        id: 's7x1e3dcmhrmc0cm8rbatcneec',
+        model: 'black-forest-labs/flux-schnell',
+        version: 'dp-4d0bcc010b3049749a251855f12800be',
+        input: {
+          num_outputs: 1,
+          prompt: 'The Loch Ness Monster getting a manicure',
+        },
+        logs: '',
+        output,
+        data_removed: false,
+        error: null,
+        status: 'processing',
+        created_at: '2025-01-08T13:24:38.692Z',
+        urls: {
+          cancel:
+            'https://api.replicate.com/v1/predictions/s7x1e3dcmhrmc0cm8rbatcneec/cancel',
+          get: 'https://api.replicate.com/v1/predictions/s7x1e3dcmhrmc0cm8rbatcneec',
+          stream:
+            'https://stream.replicate.com/v1/files/bcwr-3okdfv3o2wehstv5f2okyftwxy57hhypqsi6osiim5iaq5k7u24a',
+        },
+      },
+    };
+  }
+
+  it('should pass the model and the settings', async () => {
+    prepareResponse();
+
+    await model.doGenerate({
+      prompt,
+      n: 1,
+      size: '1024x768',
+      aspectRatio: '3:4',
+      seed: 123,
+      providerOptions: {
+        replicate: {
+          style: 'realistic_image',
+        },
+        other: {
+          something: 'else',
+        },
+      },
+    });
+
+    expect(await server.calls[0].requestBody).toStrictEqual({
+      input: {
+        prompt,
+        num_outputs: 1,
+        aspect_ratio: '3:4',
+        size: '1024x768',
+        seed: 123,
+        style: 'realistic_image',
+      },
+    });
+  });
+
+  it('should call the correct url', async () => {
+    prepareResponse();
+
+    await model.doGenerate({
+      prompt,
+      n: 1,
+      size: undefined,
+      aspectRatio: undefined,
+      seed: undefined,
+      providerOptions: {},
+    });
+
+    expect(server.calls[0].requestMethod).toStrictEqual('POST');
+    expect(server.calls[0].requestUrl).toStrictEqual(
+      'https://api.replicate.com/v1/models/black-forest-labs/flux-schnell/predictions',
+    );
+  });
+
+  it('should pass headers and set the prefer header', async () => {
+    prepareResponse();
+
+    const provider = createReplicate({
+      apiToken: 'test-api-token',
+      headers: {
+        'Custom-Provider-Header': 'provider-header-value',
+      },
+    });
+
+    await provider.image('black-forest-labs/flux-schnell').doGenerate({
+      prompt,
+      n: 1,
+      size: undefined,
+      aspectRatio: undefined,
+      seed: undefined,
+      providerOptions: {},
+      headers: {
+        'Custom-Request-Header': 'request-header-value',
+      },
+    });
+
+    expect(server.calls[0].requestHeaders).toStrictEqual({
+      authorization: 'Bearer test-api-token',
+      'content-type': 'application/json',
+      'custom-provider-header': 'provider-header-value',
+      'custom-request-header': 'request-header-value',
+      prefer: 'wait',
+    });
+  });
+
+  it('should extract the generated image from array response', async () => {
+    prepareResponse({
+      output: ['https://replicate.delivery/xezq/abc/out-0.webp'],
+    });
+
+    const result = await model.doGenerate({
+      prompt,
+      n: 1,
+      size: undefined,
+      aspectRatio: undefined,
+      seed: undefined,
+      providerOptions: {},
+    });
+
+    expect(result.images).toStrictEqual([
+      new Uint8Array(Buffer.from('test-binary-content')),
+    ]);
+
+    expect(server.calls[1].requestMethod).toStrictEqual('GET');
+    expect(server.calls[1].requestUrl).toStrictEqual(
+      'https://replicate.delivery/xezq/abc/out-0.webp',
+    );
+  });
+
+  it('should extract the generated image from string response', async () => {
+    prepareResponse({
+      output: 'https://replicate.delivery/xezq/abc/out-0.webp',
+    });
+
+    const result = await model.doGenerate({
+      prompt,
+      n: 1,
+      size: undefined,
+      aspectRatio: undefined,
+      seed: undefined,
+      providerOptions: {},
+    });
+
+    expect(result.images).toStrictEqual([
+      new Uint8Array(Buffer.from('test-binary-content')),
+    ]);
+
+    expect(server.calls[1].requestMethod).toStrictEqual('GET');
+    expect(server.calls[1].requestUrl).toStrictEqual(
+      'https://replicate.delivery/xezq/abc/out-0.webp',
+    );
+  });
+});

package/src/replicate-image-model.ts

@@ -0,0 +1,104 @@
+import type { ImageModelV1, ImageModelV1CallWarning } from '@ai-sdk/provider';
+import type { Resolvable } from '@ai-sdk/provider-utils';
+import {
+  FetchFunction,
+  combineHeaders,
+  createJsonResponseHandler,
+  postJsonToApi,
+  resolve,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod';
+import { replicateFailedResponseHandler } from './replicate-error';
+import {
+  ReplicateImageModelId,
+  ReplicateImageSettings,
+} from './replicate-image-settings';
+
+interface ReplicateImageModelConfig {
+  provider: string;
+  baseURL: string;
+  headers?: Resolvable<Record<string, string | undefined>>;
+  fetch?: FetchFunction;
+}
+
+export class ReplicateImageModel implements ImageModelV1 {
+  readonly specificationVersion = 'v1';
+
+  readonly modelId: ReplicateImageModelId;
+  readonly settings: ReplicateImageSettings;
+
+  private readonly config: ReplicateImageModelConfig;
+
+  get provider(): string {
+    return this.config.provider;
+  }
+
+  get maxImagesPerCall(): number {
+    return this.settings.maxImagesPerCall ?? 1;
+  }
+
+  constructor(
+    modelId: ReplicateImageModelId,
+    settings: ReplicateImageSettings,
+    config: ReplicateImageModelConfig,
+  ) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+  }
+
+  async doGenerate({
+    prompt,
+    n,
+    aspectRatio,
+    size,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal,
+  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<
+    Awaited<ReturnType<ImageModelV1['doGenerate']>>
+  > {
+    const warnings: Array<ImageModelV1CallWarning> = [];
+
+    const {
+      value: { output },
+    } = await postJsonToApi({
+      url: `${this.config.baseURL}/models/${this.modelId}/predictions`,
+      headers: combineHeaders(await resolve(this.config.headers), headers, {
+        prefer: 'wait',
+      }),
+      body: {
+        input: {
+          prompt,
+          aspect_ratio: aspectRatio,
+          size,
+          seed,
+          num_outputs: n,
+          ...(providerOptions.replicate ?? {}),
+        },
+      },
+      failedResponseHandler: replicateFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        replicateImageResponseSchema,
+      ),
+      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 response = await fetch(url);
+        return new Uint8Array(await response.arrayBuffer());
+      }),
+    );
+
+    return { images, warnings };
+  }
+}
+
+const replicateImageResponseSchema = z.object({
+  output: z.union([z.array(z.string()), z.string()]),
+});

package/src/replicate-image-settings.ts

@@ -0,0 +1,36 @@
+export 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'
+  | (string & {});
+
+export interface ReplicateImageSettings {
+  /**
+Override the maximum number of images per call (default 1)
+   */
+  maxImagesPerCall?: number;
+}

package/src/replicate-provider.test.ts

@@ -0,0 +1,24 @@
+import { describe, it, expect } from 'vitest';
+import { createReplicate } from './replicate-provider';
+import { ReplicateImageModel } from './replicate-image-model';
+
+describe('createReplicate', () => {
+  it('creates a provider with required settings', () => {
+    const provider = createReplicate({ apiToken: 'test-token' });
+    expect(provider.image).toBeDefined();
+  });
+
+  it('creates a provider with custom settings', () => {
+    const provider = createReplicate({
+      apiToken: 'test-token',
+      baseURL: 'https://custom.replicate.com',
+    });
+    expect(provider.image).toBeDefined();
+  });
+
+  it('creates an image model instance', () => {
+    const provider = createReplicate({ apiToken: 'test-token' });
+    const model = provider.image('black-forest-labs/flux-schnell');
+    expect(model).toBeInstanceOf(ReplicateImageModel);
+  });
+});

package/src/replicate-provider.ts

@@ -0,0 +1,74 @@
+import type { FetchFunction } from '@ai-sdk/provider-utils';
+import { loadApiKey } from '@ai-sdk/provider-utils';
+import { ReplicateImageModel } from './replicate-image-model';
+import {
+  ReplicateImageModelId,
+  ReplicateImageSettings,
+} from './replicate-image-settings';
+
+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 {
+  /**
+   * Creates a Replicate image generation model.
+   */
+  image(
+    modelId: ReplicateImageModelId,
+    settings?: ReplicateImageSettings,
+  ): ReplicateImageModel;
+}
+
+/**
+ * Create a Replicate provider instance.
+ */
+export function createReplicate(
+  options: ReplicateProviderSettings = {},
+): ReplicateProvider {
+  return {
+    image: (
+      modelId: ReplicateImageModelId,
+      settings?: ReplicateImageSettings,
+    ) =>
+      new ReplicateImageModel(modelId, settings ?? {}, {
+        provider: 'replicate',
+        baseURL: options.baseURL ?? 'https://api.replicate.com/v1',
+        headers: {
+          Authorization: `Bearer ${loadApiKey({
+            apiKey: options.apiToken,
+            environmentVariableName: 'REPLICATE_API_TOKEN',
+            description: 'Replicate',
+          })}`,
+          ...options.headers,
+        },
+        fetch: options.fetch,
+      }),
+  };
+}
+
+/**
+ * Default Replicate provider instance.
+ */
+export const replicate = createReplicate();

package/tsconfig.json

@@ -0,0 +1,5 @@
+{
+  "extends": "./node_modules/@vercel/ai-tsconfig/ts-library.json",
+  "include": ["."],
+  "exclude": ["*/dist", "dist", "build", "node_modules"]
+}

package/tsup.config.ts

@@ -0,0 +1,10 @@
+import { defineConfig } from 'tsup';
+
+export default defineConfig([
+  {
+    entry: ['src/index.ts'],
+    format: ['cjs', 'esm'],
+    dts: true,
+    sourcemap: true,
+  },
+]);

package/turbo.json

@@ -0,0 +1,12 @@
+{
+  "extends": [
+    "//"
+  ],
+  "tasks": {
+    "build": {
+      "outputs": [
+        "**/dist/**"
+      ]
+    }
+  }
+}

package/vitest.edge.config.js

@@ -0,0 +1,10 @@
+import { defineConfig } from 'vite';
+
+// https://vitejs.dev/config/
+export default defineConfig({
+  test: {
+    environment: 'edge-runtime',
+    globals: true,
+    include: ['**/*.test.ts', '**/*.test.tsx'],
+  },
+});

package/vitest.node.config.js

@@ -0,0 +1,10 @@
+import { defineConfig } from 'vite';
+
+// https://vitejs.dev/config/
+export default defineConfig({
+  test: {
+    environment: 'node',
+    globals: true,
+    include: ['**/*.test.ts', '**/*.test.tsx'],
+  },
+});