@ai-sdk/replicate 3.0.0-beta.3 → 3.0.0-beta.30

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@ai-sdk/replicate",
-  "version": "3.0.0-beta.3",
+  "version": "3.0.0-beta.30",
+  "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
-  "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",
   "files": [
     "dist/**/*",
@@ -24,20 +24,20 @@
     "./package.json": "./package.json",
     ".": {
       "types": "./dist/index.d.ts",
-      "import": "./dist/index.mjs",
-      "require": "./dist/index.js"
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "4.0.0-beta.1",
-    "@ai-sdk/provider-utils": "5.0.0-beta.2"
+    "@ai-sdk/provider": "4.0.0-beta.13",
+    "@ai-sdk/provider-utils": "5.0.0-beta.28"
   },
   "devDependencies": {
     "@types/node": "20.17.24",
     "tsup": "^8",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "2.0.0-beta.0",
+    "@ai-sdk/test-server": "2.0.0-beta.2",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {
@@ -47,12 +47,14 @@
     "node": ">=18"
   },
   "publishConfig": {
-    "access": "public"
+    "access": "public",
+    "provenance": true
   },
   "homepage": "https://ai-sdk.dev/docs",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/vercel/ai.git"
+    "url": "https://github.com/vercel/ai",
+    "directory": "packages/replicate"
   },
   "bugs": {
     "url": "https://github.com/vercel/ai/issues"
@@ -64,9 +66,7 @@
     "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
     "build:watch": "pnpm clean && tsup --watch",
     "clean": "del-cli dist docs *.tsbuildinfo",
-    "lint": "eslint \"./**/*.ts*\"",
     "type-check": "tsc --build",
-    "prettier-check": "prettier --check \"./**/*.ts*\"",
     "test": "pnpm test:node && pnpm test:edge",
     "test:update": "pnpm test:node -u",
     "test:watch": "vitest --config vitest.node.config.js",

package/src/replicate-image-model.ts

@@ -12,6 +12,9 @@ import {
   parseProviderOptions,
   postJsonToApi,
   resolve,
+  serializeModelOptions,
+  WORKFLOW_SERIALIZE,
+  WORKFLOW_DESERIALIZE,
   zodSchema,
 } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
@@ -48,6 +51,20 @@ export class ReplicateImageModel implements ImageModelV4 {
     return FLUX_2_MODEL_PATTERN.test(this.modelId);
   }
 
+  static [WORKFLOW_SERIALIZE](model: ReplicateImageModel) {
+    return serializeModelOptions({
+      modelId: model.modelId,
+      config: model.config,
+    });
+  }
+
+  static [WORKFLOW_DESERIALIZE](options: {
+    modelId: ReplicateImageModelId;
+    config: ReplicateImageModelConfig;
+  }) {
+    return new ReplicateImageModel(options.modelId, options.config);
+  }
+
   constructor(
     readonly modelId: ReplicateImageModelId,
     private readonly config: ReplicateImageModelConfig,
@@ -148,7 +165,7 @@ export class ReplicateImageModel implements ImageModelV4 {
           : `${this.config.baseURL}/models/${modelId}/predictions`,
 
       headers: combineHeaders(
-        await resolve(this.config.headers),
+        this.config.headers ? await resolve(this.config.headers) : undefined,
         headers,
         preferHeader,
       ),

package/dist/index.d.mts

@@ -1,117 +0,0 @@
-import { ImageModelV4, ProviderV4, Experimental_VideoModelV4 } 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 ImageModelV4 {
-    readonly modelId: ReplicateImageModelId;
-    private readonly config;
-    readonly specificationVersion = "v4";
-    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<ImageModelV4['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV4['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 replicateImageModelOptionsSchema: _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 ReplicateImageModelOptions = InferSchema<typeof replicateImageModelOptionsSchema>;
-
-type ReplicateVideoModelId = 'minimax/video-01' | 'minimax/video-01:6c1e4171-288a-4ca2-a738-894f0e87699d' | 'stability-ai/stable-video-diffusion:3f0457e4619daac51203dedb472816fd4af51f3149fa7a9e0b5ffcf1b8172438' | (string & {});
-
-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 ProviderV4 {
-    /**
-     * 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;
-    /**
-     * Creates a Replicate video generation model.
-     */
-    video(modelId: ReplicateVideoModelId): Experimental_VideoModelV4;
-    /**
-     * Creates a Replicate video generation model.
-     */
-    videoModel(modelId: ReplicateVideoModelId): Experimental_VideoModelV4;
-}
-/**
- * Create a Replicate provider instance.
- */
-declare function createReplicate(options?: ReplicateProviderSettings): ReplicateProvider;
-/**
- * Default Replicate provider instance.
- */
-declare const replicate: ReplicateProvider;
-
-type ReplicateVideoModelOptions = {
-    pollIntervalMs?: number | null;
-    pollTimeoutMs?: number | null;
-    maxWaitTimeInSeconds?: number | null;
-    guidance_scale?: number | null;
-    num_inference_steps?: number | null;
-    motion_bucket_id?: number | null;
-    cond_aug?: number | null;
-    decoding_t?: number | null;
-    video_length?: string | null;
-    sizing_strategy?: string | null;
-    frames_per_second?: number | null;
-    prompt_optimizer?: boolean | null;
-    [key: string]: unknown;
-};
-
-declare const VERSION: string;
-
-export { type ReplicateImageModelOptions, type ReplicateImageModelOptions as ReplicateImageProviderOptions, type ReplicateProvider, type ReplicateProviderSettings, type ReplicateVideoModelId, type ReplicateVideoModelOptions, type ReplicateVideoModelOptions as ReplicateVideoProviderOptions, VERSION, createReplicate, replicate };