@ai-sdk/replicate 0.1.1 → 0.1.3

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @ai-sdk/replicate
 
+## 0.1.3
+
+### Patch Changes
+
+- Updated dependencies [39e5c1f]
+  - @ai-sdk/provider-utils@2.1.3
+
+## 0.1.2
+
+### Patch Changes
+
+- 3a58a2e: feat (ai/core): throw NoImageGeneratedError from generateImage when no predictions are returned.
+- Updated dependencies [ed012d2]
+- Updated dependencies [3a58a2e]
+  - @ai-sdk/provider-utils@2.1.2
+  - @ai-sdk/provider@1.0.6
+
 ## 0.1.1
 
 ### Patch Changes

package/dist/index.d.mts

@@ -14,6 +14,9 @@ interface ReplicateImageModelConfig {
     baseURL: string;
     headers?: Resolvable<Record<string, string | undefined>>;
     fetch?: FetchFunction;
+    _internal?: {
+        currentDate?: () => Date;
+    };
 }
 declare class ReplicateImageModel implements ImageModelV1 {
     readonly modelId: ReplicateImageModelId;

package/dist/index.d.ts

@@ -14,6 +14,9 @@ interface ReplicateImageModelConfig {
     baseURL: string;
     headers?: Resolvable<Record<string, string | undefined>>;
     fetch?: FetchFunction;
+    _internal?: {
+        currentDate?: () => Date;
+    };
 }
 declare class ReplicateImageModel implements ImageModelV1 {
     readonly modelId: ReplicateImageModelId;

package/dist/index.js

@@ -72,10 +72,12 @@ var ReplicateImageModel = class {
     headers,
     abortSignal
   }) {
-    var _a;
+    var _a, _b, _c, _d;
     const warnings = [];
+    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 {
-      value: { output }
+      value: { output },
+      responseHeaders
     } = 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, {
@@ -88,7 +90,7 @@ var ReplicateImageModel = class {
           size,
           seed,
           num_outputs: n,
-          ...(_a = providerOptions.replicate) != null ? _a : {}
+          ...(_d = providerOptions.replicate) != null ? _d : {}
         }
       },
       failedResponseHandler: replicateFailedResponseHandler,
@@ -105,7 +107,15 @@ var ReplicateImageModel = class {
         return new Uint8Array(await response.arrayBuffer());
       })
     );
-    return { images, warnings };
+    return {
+      images,
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
   }
 };
 var replicateImageResponseSchema = import_zod2.z.object({

package/dist/index.js.map

@@ -1 +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  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    readonly modelId: ReplicateImageModelId,\n    private readonly settings: ReplicateImageSettings,\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  }: 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,EAWvD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAbnB,SAAS,uBAAuB;AAAA,EAc7B;AAAA,EAZH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AA9BjC;AA+BI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAQA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAnDJ;AAoDI,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;;;ADhDM,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"]}
+{"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  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nexport class ReplicateImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\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    readonly modelId: ReplicateImageModelId,\n    private readonly settings: ReplicateImageSettings,\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  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const {\n      value: { output },\n      responseHeaders,\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 {\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","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;;;ADcM,IAAM,sBAAN,MAAkD;AAAA,EAWvD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAbnB,SAAS,uBAAuB;AAAA,EAc7B;AAAA,EAZH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AAjCjC;AAkCI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAQA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAtDJ;AAuDI,UAAM,WAA2C,CAAC;AAElD,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM;AAAA,MACJ,OAAO,EAAE,OAAO;AAAA,MAChB;AAAA,IACF,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;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,cAAE,OAAO;AAAA,EAC5C,QAAQ,cAAE,MAAM,CAAC,cAAE,MAAM,cAAE,OAAO,CAAC,GAAG,cAAE,OAAO,CAAC,CAAC;AACnD,CAAC;;;AD7DM,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

@@ -50,10 +50,12 @@ var ReplicateImageModel = class {
     headers,
     abortSignal
   }) {
-    var _a;
+    var _a, _b, _c, _d;
     const warnings = [];
+    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 {
-      value: { output }
+      value: { output },
+      responseHeaders
     } = await postJsonToApi({
       url: `${this.config.baseURL}/models/${this.modelId}/predictions`,
       headers: combineHeaders(await resolve(this.config.headers), headers, {
@@ -66,7 +68,7 @@ var ReplicateImageModel = class {
           size,
           seed,
           num_outputs: n,
-          ...(_a = providerOptions.replicate) != null ? _a : {}
+          ...(_d = providerOptions.replicate) != null ? _d : {}
         }
       },
       failedResponseHandler: replicateFailedResponseHandler,
@@ -83,7 +85,15 @@ var ReplicateImageModel = class {
         return new Uint8Array(await response.arrayBuffer());
       })
     );
-    return { images, warnings };
+    return {
+      images,
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
   }
 };
 var replicateImageResponseSchema = z2.object({

package/dist/index.mjs.map

@@ -1 +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  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    readonly modelId: ReplicateImageModelId,\n    private readonly settings: ReplicateImageSettings,\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  }: 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,EAWvD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAbnB,SAAS,uBAAuB;AAAA,EAc7B;AAAA,EAZH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AA9BjC;AA+BI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAQA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAnDJ;AAoDI,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;;;ADhDM,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"]}
+{"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  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nexport class ReplicateImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\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    readonly modelId: ReplicateImageModelId,\n    private readonly settings: ReplicateImageSettings,\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  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const {\n      value: { output },\n      responseHeaders,\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 {\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","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;;;ADcM,IAAM,sBAAN,MAAkD;AAAA,EAWvD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAbnB,SAAS,uBAAuB;AAAA,EAc7B;AAAA,EAZH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AAjCjC;AAkCI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAQA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAtDJ;AAuDI,UAAM,WAA2C,CAAC;AAElD,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM;AAAA,MACJ,OAAO,EAAE,OAAO;AAAA,MAChB;AAAA,IACF,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;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+BC,GAAE,OAAO;AAAA,EAC5C,QAAQA,GAAE,MAAM,CAACA,GAAE,MAAMA,GAAE,OAAO,CAAC,GAAGA,GAAE,OAAO,CAAC,CAAC;AACnD,CAAC;;;AD7DM,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

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/replicate",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -19,8 +19,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "1.0.5",
-    "@ai-sdk/provider-utils": "2.1.1"
+    "@ai-sdk/provider": "1.0.6",
+    "@ai-sdk/provider-utils": "2.1.3"
   },
   "devDependencies": {
     "@types/node": "^18",