@ai-sdk/replicate 3.0.0-beta.5 → 3.0.0-beta.51

package/dist/index.js

@@ -1,47 +1,36 @@
-"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 index_exports = {};
-__export(index_exports, {
-  VERSION: () => VERSION,
-  createReplicate: () => createReplicate,
-  replicate: () => replicate
-});
-module.exports = __toCommonJS(index_exports);
-
 // src/replicate-provider.ts
-var import_provider2 = require("@ai-sdk/provider");
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+import {
+  NoSuchModelError
+} from "@ai-sdk/provider";
+import {
+  loadApiKey,
+  withUserAgentSuffix
+} from "@ai-sdk/provider-utils";
 
 // src/replicate-image-model.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
-var import_v42 = require("zod/v4");
+import {
+  combineHeaders,
+  convertImageModelFileToDataUri,
+  createBinaryResponseHandler,
+  createJsonResponseHandler,
+  getFromApi,
+  parseProviderOptions,
+  postJsonToApi,
+  resolve,
+  serializeModelOptions,
+  WORKFLOW_SERIALIZE,
+  WORKFLOW_DESERIALIZE
+} from "@ai-sdk/provider-utils";
+import { z as z3 } from "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()
+import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
+import { z } from "zod/v4";
+var replicateErrorSchema = z.object({
+  detail: z.string().optional(),
+  error: z.string().optional()
 });
-var replicateFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
+var replicateFailedResponseHandler = createJsonErrorResponseHandler({
   errorSchema: replicateErrorSchema,
   errorToMessage: (error) => {
     var _a, _b;
@@ -49,10 +38,57 @@ var replicateFailedResponseHandler = (0, import_provider_utils.createJsonErrorRe
   }
 });
 
+// src/replicate-image-model-options.ts
+import {
+  lazySchema,
+  zodSchema
+} from "@ai-sdk/provider-utils";
+import { z as z2 } from "zod/v4";
+var replicateImageModelOptionsSchema = lazySchema(
+  () => zodSchema(
+    z2.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: z2.number().positive().nullish(),
+      /**
+       * Guidance scale for classifier-free guidance.
+       * Higher values make the output more closely match the prompt.
+       */
+      guidance_scale: z2.number().nullish(),
+      /**
+       * Number of denoising steps. More steps = higher quality but slower.
+       */
+      num_inference_steps: z2.number().nullish(),
+      /**
+       * Negative prompt to guide what to avoid in the generation.
+       */
+      negative_prompt: z2.string().nullish(),
+      /**
+       * Output image format.
+       */
+      output_format: z2.enum(["png", "jpg", "webp"]).nullish(),
+      /**
+       * Output image quality (1-100). Only applies to jpg and webp.
+       */
+      output_quality: z2.number().min(1).max(100).nullish(),
+      /**
+       * Strength of the transformation for img2img (0-1).
+       * Lower values keep more of the original image.
+       */
+      strength: z2.number().min(0).max(1).nullish()
+    }).passthrough()
+  )
+);
+
 // 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 {
+var ReplicateImageModel = class _ReplicateImageModel {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
@@ -67,6 +103,15 @@ var ReplicateImageModel = class {
   get isFlux2Model() {
     return FLUX_2_MODEL_PATTERN.test(this.modelId);
   }
+  static [WORKFLOW_SERIALIZE](model) {
+    return serializeModelOptions({
+      modelId: model.modelId,
+      config: model.config
+    });
+  }
+  static [WORKFLOW_DESERIALIZE](options) {
+    return new _ReplicateImageModel(options.modelId, options.config);
+  }
   async doGenerate({
     prompt,
     n,
@@ -83,7 +128,7 @@ var ReplicateImageModel = class {
     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)({
+    const replicateOptions = await parseProviderOptions({
       provider: "replicate",
       providerOptions,
       schema: replicateImageModelOptionsSchema
@@ -93,7 +138,7 @@ var ReplicateImageModel = class {
       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]);
+          imageInputs[key] = convertImageModelFileToDataUri(files[i]);
         }
         if (files.length > MAX_FLUX_2_INPUT_IMAGES) {
           warnings.push({
@@ -102,7 +147,7 @@ var ReplicateImageModel = class {
           });
         }
       } else {
-        imageInputs = { image: (0, import_provider_utils2.convertImageModelFileToDataUri)(files[0]) };
+        imageInputs = { image: convertImageModelFileToDataUri(files[0]) };
         if (files.length > 1) {
           warnings.push({
             type: "other",
@@ -119,7 +164,7 @@ var ReplicateImageModel = class {
           message: "Flux-2 models do not support mask input. The mask will be ignored."
         });
       } else {
-        maskInput = (0, import_provider_utils2.convertImageModelFileToDataUri)(mask);
+        maskInput = convertImageModelFileToDataUri(mask);
       }
     }
     const { maxWaitTimeInSeconds, ...inputOptions } = replicateOptions != null ? replicateOptions : {};
@@ -127,13 +172,13 @@ var ReplicateImageModel = class {
     const {
       value: { output },
       responseHeaders
-    } = await (0, import_provider_utils2.postJsonToApi)({
+    } = await 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: combineHeaders(
+        this.config.headers ? await resolve(this.config.headers) : void 0,
         headers,
         preferHeader
       ),
@@ -151,7 +196,7 @@ var ReplicateImageModel = class {
         // for versioned models, include the version in the body:
         ...version != null ? { version } : {}
       },
-      successfulResponseHandler: (0, import_provider_utils2.createJsonResponseHandler)(
+      successfulResponseHandler: createJsonResponseHandler(
         replicateImageResponseSchema
       ),
       failedResponseHandler: replicateFailedResponseHandler,
@@ -161,9 +206,9 @@ var ReplicateImageModel = class {
     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)({
+        const { value: image } = await getFromApi({
           url,
-          successfulResponseHandler: (0, import_provider_utils2.createBinaryResponseHandler)(),
+          successfulResponseHandler: createBinaryResponseHandler(),
           failedResponseHandler: replicateFailedResponseHandler,
           abortSignal,
           fetch: this.config.fetch
@@ -182,54 +227,50 @@ var ReplicateImageModel = class {
     };
   }
 };
-var replicateImageResponseSchema = import_v42.z.object({
-  output: import_v42.z.union([import_v42.z.array(import_v42.z.string()), import_v42.z.string()])
+var replicateImageResponseSchema = z3.object({
+  output: z3.union([z3.array(z3.string()), z3.string()])
 });
-var replicateImageModelOptionsSchema = (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/replicate-video-model.ts
+import {
+  AISDKError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders as combineHeaders2,
+  convertImageModelFileToDataUri as convertImageModelFileToDataUri2,
+  createJsonResponseHandler as createJsonResponseHandler2,
+  delay,
+  getFromApi as getFromApi2,
+  isSameOrigin,
+  parseProviderOptions as parseProviderOptions2,
+  postJsonToApi as postJsonToApi2,
+  resolve as resolve2
+} from "@ai-sdk/provider-utils";
+import { z as z5 } from "zod/v4";
+
+// src/replicate-video-model-options.ts
+import { lazySchema as lazySchema2, zodSchema as zodSchema2 } from "@ai-sdk/provider-utils";
+import { z as z4 } from "zod/v4";
+var replicateVideoModelOptionsSchema = lazySchema2(
+  () => zodSchema2(
+    z4.object({
+      pollIntervalMs: z4.number().positive().nullish(),
+      pollTimeoutMs: z4.number().positive().nullish(),
+      maxWaitTimeInSeconds: z4.number().positive().nullish(),
+      guidance_scale: z4.number().nullish(),
+      num_inference_steps: z4.number().nullish(),
+      motion_bucket_id: z4.number().nullish(),
+      cond_aug: z4.number().nullish(),
+      decoding_t: z4.number().nullish(),
+      video_length: z4.string().nullish(),
+      sizing_strategy: z4.string().nullish(),
+      frames_per_second: z4.number().nullish(),
+      prompt_optimizer: z4.boolean().nullish()
+    }).loose()
   )
 );
 
 // src/replicate-video-model.ts
-var import_provider = require("@ai-sdk/provider");
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
-var import_v43 = require("zod/v4");
 var ReplicateVideoModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -245,7 +286,7 @@ var ReplicateVideoModel = class {
     var _a, _b, _c, _d, _e, _f, _g;
     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 warnings = [];
-    const replicateOptions = await (0, import_provider_utils3.parseProviderOptions)({
+    const replicateOptions = await parseProviderOptions2({
       provider: "replicate",
       providerOptions: options.providerOptions,
       schema: replicateVideoModelOptionsSchema
@@ -259,7 +300,7 @@ var ReplicateVideoModel = class {
       if (options.image.type === "url") {
         input.image = options.image.url;
       } else {
-        input.image = (0, import_provider_utils3.convertImageModelFileToDataUri)(options.image);
+        input.image = convertImageModelFileToDataUri2(options.image);
       }
     }
     if (options.aspectRatio) {
@@ -328,10 +369,10 @@ var ReplicateVideoModel = class {
     const maxWaitTimeInSeconds = replicateOptions == null ? void 0 : replicateOptions.maxWaitTimeInSeconds;
     const preferHeader = maxWaitTimeInSeconds != null ? { prefer: `wait=${maxWaitTimeInSeconds}` } : { prefer: "wait" };
     const predictionUrl = version != null ? `${this.config.baseURL}/predictions` : `${this.config.baseURL}/models/${modelId}/predictions`;
-    const { value: prediction, responseHeaders } = await (0, import_provider_utils3.postJsonToApi)({
+    const { value: prediction, responseHeaders } = await postJsonToApi2({
       url: predictionUrl,
-      headers: (0, import_provider_utils3.combineHeaders)(
-        await (0, import_provider_utils3.resolve)(this.config.headers),
+      headers: combineHeaders2(
+        await resolve2(this.config.headers),
         options.headers,
         preferHeader
       ),
@@ -339,7 +380,7 @@ var ReplicateVideoModel = class {
         input,
         ...version != null ? { version } : {}
       },
-      successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
+      successfulResponseHandler: createJsonResponseHandler2(
         replicatePredictionSchema
       ),
       failedResponseHandler: replicateFailedResponseHandler,
@@ -353,22 +394,25 @@ var ReplicateVideoModel = class {
       const startTime = Date.now();
       while (finalPrediction.status === "starting" || finalPrediction.status === "processing") {
         if (Date.now() - startTime > pollTimeoutMs) {
-          throw new import_provider.AISDKError({
+          throw new AISDKError({
             name: "REPLICATE_VIDEO_GENERATION_TIMEOUT",
             message: `Video generation timed out after ${pollTimeoutMs}ms`
           });
         }
-        await (0, import_provider_utils3.delay)(pollIntervalMs);
+        await delay(pollIntervalMs);
         if ((_f = options.abortSignal) == null ? void 0 : _f.aborted) {
-          throw new import_provider.AISDKError({
+          throw new AISDKError({
             name: "REPLICATE_VIDEO_GENERATION_ABORTED",
             message: "Video generation request was aborted"
           });
         }
-        const { value: statusPrediction } = await (0, import_provider_utils3.getFromApi)({
-          url: finalPrediction.urls.get,
-          headers: await (0, import_provider_utils3.resolve)(this.config.headers),
-          successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
+        const pollUrl = finalPrediction.urls.get;
+        const { value: statusPrediction } = await getFromApi2({
+          url: pollUrl,
+          // The polling URL comes from the provider response; only send
+          // credentials when it stays on the provider's own origin.
+          headers: isSameOrigin(pollUrl, this.config.baseURL) ? await resolve2(this.config.headers) : void 0,
+          successfulResponseHandler: createJsonResponseHandler2(
             replicatePredictionSchema
           ),
           failedResponseHandler: replicateFailedResponseHandler,
@@ -379,20 +423,20 @@ var ReplicateVideoModel = class {
       }
     }
     if (finalPrediction.status === "failed") {
-      throw new import_provider.AISDKError({
+      throw new AISDKError({
         name: "REPLICATE_VIDEO_GENERATION_FAILED",
         message: `Video generation failed: ${(_g = finalPrediction.error) != null ? _g : "Unknown error"}`
       });
     }
     if (finalPrediction.status === "canceled") {
-      throw new import_provider.AISDKError({
+      throw new AISDKError({
         name: "REPLICATE_VIDEO_GENERATION_CANCELED",
         message: "Video generation was canceled"
       });
     }
     const videoUrl = finalPrediction.output;
     if (!videoUrl) {
-      throw new import_provider.AISDKError({
+      throw new AISDKError({
         name: "REPLICATE_VIDEO_GENERATION_ERROR",
         message: "No video URL in response"
       });
@@ -425,45 +469,27 @@ var ReplicateVideoModel = class {
     };
   }
 };
-var replicatePredictionSchema = import_v43.z.object({
-  id: import_v43.z.string(),
-  status: import_v43.z.enum(["starting", "processing", "succeeded", "failed", "canceled"]),
-  output: import_v43.z.string().nullish(),
-  error: import_v43.z.string().nullish(),
-  urls: import_v43.z.object({
-    get: import_v43.z.string()
+var replicatePredictionSchema = z5.object({
+  id: z5.string(),
+  status: z5.enum(["starting", "processing", "succeeded", "failed", "canceled"]),
+  output: z5.string().nullish(),
+  error: z5.string().nullish(),
+  urls: z5.object({
+    get: z5.string()
   }),
-  metrics: import_v43.z.object({
-    predict_time: import_v43.z.number().nullish()
+  metrics: z5.object({
+    predict_time: z5.number().nullish()
   }).nullish()
 });
-var replicateVideoModelOptionsSchema = (0, import_provider_utils3.lazySchema)(
-  () => (0, import_provider_utils3.zodSchema)(
-    import_v43.z.object({
-      pollIntervalMs: import_v43.z.number().positive().nullish(),
-      pollTimeoutMs: import_v43.z.number().positive().nullish(),
-      maxWaitTimeInSeconds: import_v43.z.number().positive().nullish(),
-      guidance_scale: import_v43.z.number().nullish(),
-      num_inference_steps: import_v43.z.number().nullish(),
-      motion_bucket_id: import_v43.z.number().nullish(),
-      cond_aug: import_v43.z.number().nullish(),
-      decoding_t: import_v43.z.number().nullish(),
-      video_length: import_v43.z.string().nullish(),
-      sizing_strategy: import_v43.z.string().nullish(),
-      frames_per_second: import_v43.z.number().nullish(),
-      prompt_optimizer: import_v43.z.boolean().nullish()
-    }).loose()
-  )
-);
 
 // src/version.ts
-var VERSION = true ? "3.0.0-beta.5" : "0.0.0-test";
+var VERSION = true ? "3.0.0-beta.51" : "0.0.0-test";
 
 // src/replicate-provider.ts
 function createReplicate(options = {}) {
-  const getHeaders = () => (0, import_provider_utils4.withUserAgentSuffix)(
+  const getHeaders = () => withUserAgentSuffix(
     {
-      Authorization: `Bearer ${(0, import_provider_utils4.loadApiKey)({
+      Authorization: `Bearer ${loadApiKey({
         apiKey: options.apiToken,
         environmentVariableName: "REPLICATE_API_TOKEN",
         description: "Replicate"
@@ -491,7 +517,7 @@ function createReplicate(options = {}) {
     });
   };
   const embeddingModel = (modelId) => {
-    throw new import_provider2.NoSuchModelError({
+    throw new NoSuchModelError({
       modelId,
       modelType: "embeddingModel"
     });
@@ -501,7 +527,7 @@ function createReplicate(options = {}) {
     image: createImageModel,
     imageModel: createImageModel,
     languageModel: (modelId) => {
-      throw new import_provider2.NoSuchModelError({
+      throw new NoSuchModelError({
         modelId,
         modelType: "languageModel"
       });
@@ -513,10 +539,9 @@ function createReplicate(options = {}) {
   };
 }
 var replicate = createReplicate();
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+export {
   VERSION,
   createReplicate,
   replicate
-});
+};
 //# sourceMappingURL=index.js.map