@ai-sdk/openai 1.1.15 → 1.2.1

package/internal/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV1, EmbeddingModelV1 } from '@ai-sdk/provider';
+import { LanguageModelV1, EmbeddingModelV1, ImageModelV1 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 
 type OpenAIChatModelId = 'o1' | 'o1-2024-12-17' | 'o1-mini' | 'o1-mini-2024-09-12' | 'o1-preview' | 'o1-preview-2024-09-12' | 'o3-mini' | 'o3-mini-2025-01-31' | 'gpt-4o' | 'gpt-4o-2024-05-13' | 'gpt-4o-2024-08-06' | 'gpt-4o-2024-11-20' | 'gpt-4o-audio-preview' | 'gpt-4o-audio-preview-2024-10-01' | 'gpt-4o-audio-preview-2024-12-17' | 'gpt-4o-mini' | 'gpt-4o-mini-2024-07-18' | 'gpt-4-turbo' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-turbo-preview' | 'gpt-4-0125-preview' | 'gpt-4-1106-preview' | 'gpt-4' | 'gpt-4-0613' | 'gpt-4.5-preview' | 'gpt-4.5-preview-2025-02-27' | 'gpt-3.5-turbo-0125' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-1106' | (string & {});
@@ -214,4 +214,30 @@ declare class OpenAIEmbeddingModel implements EmbeddingModelV1<string> {
     doEmbed({ values, headers, abortSignal, }: Parameters<EmbeddingModelV1<string>['doEmbed']>[0]): Promise<Awaited<ReturnType<EmbeddingModelV1<string>['doEmbed']>>>;
 }
 
-export { OpenAIChatLanguageModel, type OpenAIChatModelId, type OpenAIChatSettings, OpenAICompletionLanguageModel, type OpenAICompletionModelId, type OpenAICompletionSettings, OpenAIEmbeddingModel, type OpenAIEmbeddingModelId, type OpenAIEmbeddingSettings };
+type OpenAIImageModelId = 'dall-e-3' | 'dall-e-2' | (string & {});
+declare const modelMaxImagesPerCall: Record<OpenAIImageModelId, number>;
+interface OpenAIImageSettings {
+    /**
+  Override the maximum number of images per call (default is dependent on the
+  model, or 1 for an unknown model).
+     */
+    maxImagesPerCall?: number;
+}
+
+interface OpenAIImageModelConfig extends OpenAIConfig {
+    _internal?: {
+        currentDate?: () => Date;
+    };
+}
+declare class OpenAIImageModel implements ImageModelV1 {
+    readonly modelId: OpenAIImageModelId;
+    private readonly settings;
+    private readonly config;
+    readonly specificationVersion = "v1";
+    get maxImagesPerCall(): number;
+    get provider(): string;
+    constructor(modelId: OpenAIImageModelId, settings: OpenAIImageSettings, config: OpenAIImageModelConfig);
+    doGenerate({ prompt, n, size, aspectRatio, seed, providerOptions, headers, abortSignal, }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV1['doGenerate']>>>;
+}
+
+export { OpenAIChatLanguageModel, type OpenAIChatModelId, type OpenAIChatSettings, OpenAICompletionLanguageModel, type OpenAICompletionModelId, type OpenAICompletionSettings, OpenAIEmbeddingModel, type OpenAIEmbeddingModelId, type OpenAIEmbeddingSettings, OpenAIImageModel, type OpenAIImageModelId, type OpenAIImageSettings, modelMaxImagesPerCall };

package/internal/dist/index.js

@@ -22,7 +22,9 @@ var internal_exports = {};
 __export(internal_exports, {
   OpenAIChatLanguageModel: () => OpenAIChatLanguageModel,
   OpenAICompletionLanguageModel: () => OpenAICompletionLanguageModel,
-  OpenAIEmbeddingModel: () => OpenAIEmbeddingModel
+  OpenAIEmbeddingModel: () => OpenAIEmbeddingModel,
+  OpenAIImageModel: () => OpenAIImageModel,
+  modelMaxImagesPerCall: () => modelMaxImagesPerCall
 });
 module.exports = __toCommonJS(internal_exports);
 
@@ -610,7 +612,11 @@ var OpenAIChatLanguageModel = class {
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h;
     const { args: body, warnings } = this.getArgs(options);
-    const { responseHeaders, value: response } = await (0, import_provider_utils3.postJsonToApi)({
+    const {
+      responseHeaders,
+      value: response,
+      rawValue: rawResponse
+    } = await (0, import_provider_utils3.postJsonToApi)({
       url: this.config.url({
         path: "/chat/completions",
         modelId: this.modelId
@@ -665,7 +671,7 @@ var OpenAIChatLanguageModel = class {
         completionTokens: (_h = (_g = response.usage) == null ? void 0 : _g.completion_tokens) != null ? _h : NaN
       },
       rawCall: { rawPrompt, rawSettings },
-      rawResponse: { headers: responseHeaders },
+      rawResponse: { headers: responseHeaders, body: rawResponse },
       request: { body: JSON.stringify(body) },
       response: getResponseMetadata(response),
       warnings,
@@ -1267,7 +1273,11 @@ var OpenAICompletionLanguageModel = class {
   }
   async doGenerate(options) {
     const { args, warnings } = this.getArgs(options);
-    const { responseHeaders, value: response } = await (0, import_provider_utils4.postJsonToApi)({
+    const {
+      responseHeaders,
+      value: response,
+      rawValue: rawResponse
+    } = await (0, import_provider_utils4.postJsonToApi)({
       url: this.config.url({
         path: "/completions",
         modelId: this.modelId
@@ -1292,7 +1302,7 @@ var OpenAICompletionLanguageModel = class {
       finishReason: mapOpenAIFinishReason(choice.finish_reason),
       logprobs: mapOpenAICompletionLogProbs(choice.logprobs),
       rawCall: { rawPrompt, rawSettings },
-      rawResponse: { headers: responseHeaders },
+      rawResponse: { headers: responseHeaders, body: rawResponse },
       response: getResponseMetadata(response),
       warnings,
       request: { body: JSON.stringify(args) }
@@ -1502,10 +1512,96 @@ var openaiTextEmbeddingResponseSchema = import_zod4.z.object({
   data: import_zod4.z.array(import_zod4.z.object({ embedding: import_zod4.z.array(import_zod4.z.number()) })),
   usage: import_zod4.z.object({ prompt_tokens: import_zod4.z.number() }).nullish()
 });
+
+// src/openai-image-model.ts
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_zod5 = require("zod");
+
+// src/openai-image-settings.ts
+var modelMaxImagesPerCall = {
+  "dall-e-3": 1,
+  "dall-e-2": 10
+};
+
+// src/openai-image-model.ts
+var OpenAIImageModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.specificationVersion = "v1";
+  }
+  get maxImagesPerCall() {
+    var _a, _b;
+    return (_b = (_a = this.settings.maxImagesPerCall) != null ? _a : modelMaxImagesPerCall[this.modelId]) != null ? _b : 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    var _a, _b, _c, _d;
+    const warnings = [];
+    if (aspectRatio != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "aspectRatio",
+        details: "This model does not support aspect ratio. Use `size` instead."
+      });
+    }
+    if (seed != null) {
+      warnings.push({ type: "unsupported-setting", setting: "seed" });
+    }
+    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: response, responseHeaders } = await (0, import_provider_utils6.postJsonToApi)({
+      url: this.config.url({
+        path: "/images/generations",
+        modelId: this.modelId
+      }),
+      headers: (0, import_provider_utils6.combineHeaders)(this.config.headers(), headers),
+      body: {
+        model: this.modelId,
+        prompt,
+        n,
+        size,
+        ...(_d = providerOptions.openai) != null ? _d : {},
+        response_format: "b64_json"
+      },
+      failedResponseHandler: openaiFailedResponseHandler,
+      successfulResponseHandler: (0, import_provider_utils6.createJsonResponseHandler)(
+        openaiImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return {
+      images: response.data.map((item) => item.b64_json),
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
+  }
+};
+var openaiImageResponseSchema = import_zod5.z.object({
+  data: import_zod5.z.array(import_zod5.z.object({ b64_json: import_zod5.z.string() }))
+});
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   OpenAIChatLanguageModel,
   OpenAICompletionLanguageModel,
-  OpenAIEmbeddingModel
+  OpenAIEmbeddingModel,
+  OpenAIImageModel,
+  modelMaxImagesPerCall
 });
 //# sourceMappingURL=index.js.map