@llmgateway/ai-sdk-provider 3.6.0 → 3.8.0

package/dist/index.d.mts

@@ -133,7 +133,9 @@ type LLMGatewayCompletionSettings = {
 } & LLMGatewaySharedSettings;
 
 type LLMGatewayImageModelId = string;
-type LLMGatewayImageSettings = {};
+type LLMGatewayImageSettings = {
+    extraBody?: Record<string, unknown>;
+};
 
 type LLMGatewayChatConfig = {
     provider: string;
@@ -191,6 +193,10 @@ type LLMGatewayImageConfig = {
         path: string;
     }) => string;
     fetch?: typeof fetch;
+    extraBody?: Record<string, unknown>;
+};
+type LLMGatewayImageModelCallOptions = ImageModelV3CallOptions & {
+    quality?: string;
 };
 declare class LLMGatewayImageModel implements ImageModelV3 {
     readonly specificationVersion: "v3";
@@ -200,7 +206,7 @@ declare class LLMGatewayImageModel implements ImageModelV3 {
     readonly settings: LLMGatewayImageSettings;
     private readonly config;
     constructor(modelId: LLMGatewayImageModelId, settings: LLMGatewayImageSettings, config: LLMGatewayImageConfig);
-    doGenerate(options: ImageModelV3CallOptions): Promise<{
+    doGenerate(options: LLMGatewayImageModelCallOptions): Promise<{
         images: Array<string>;
         warnings: Array<SharedV3Warning>;
         response: {

package/dist/index.d.ts

@@ -133,7 +133,9 @@ type LLMGatewayCompletionSettings = {
 } & LLMGatewaySharedSettings;
 
 type LLMGatewayImageModelId = string;
-type LLMGatewayImageSettings = {};
+type LLMGatewayImageSettings = {
+    extraBody?: Record<string, unknown>;
+};
 
 type LLMGatewayChatConfig = {
     provider: string;
@@ -191,6 +193,10 @@ type LLMGatewayImageConfig = {
         path: string;
     }) => string;
     fetch?: typeof fetch;
+    extraBody?: Record<string, unknown>;
+};
+type LLMGatewayImageModelCallOptions = ImageModelV3CallOptions & {
+    quality?: string;
 };
 declare class LLMGatewayImageModel implements ImageModelV3 {
     readonly specificationVersion: "v3";
@@ -200,7 +206,7 @@ declare class LLMGatewayImageModel implements ImageModelV3 {
     readonly settings: LLMGatewayImageSettings;
     private readonly config;
     constructor(modelId: LLMGatewayImageModelId, settings: LLMGatewayImageSettings, config: LLMGatewayImageConfig);
-    doGenerate(options: ImageModelV3CallOptions): Promise<{
+    doGenerate(options: LLMGatewayImageModelCallOptions): Promise<{
         images: Array<string>;
         warnings: Array<SharedV3Warning>;
         response: {

package/dist/index.js

@@ -2308,6 +2308,29 @@ function getBase64FromDataUrl(dataUrl) {
 }
 
 // src/chat/convert-to-llmgateway-chat-messages.ts
+var AUDIO_FORMAT_BY_MIME = {
+  "audio/wav": "wav",
+  "audio/wave": "wav",
+  "audio/x-wav": "wav",
+  "audio/mpeg": "mp3",
+  "audio/mp3": "mp3",
+  "audio/mp4": "mp4",
+  "audio/m4a": "m4a",
+  "audio/x-m4a": "m4a",
+  "audio/aac": "aac",
+  "audio/ogg": "ogg",
+  "audio/flac": "flac",
+  "audio/aiff": "aiff",
+  "audio/x-aiff": "aiff",
+  "audio/webm": "webm"
+};
+function audioFormatFromMime(mime) {
+  var _a16;
+  if (!mime) return void 0;
+  const lower = (_a16 = mime.toLowerCase().split(";")[0]) == null ? void 0 : _a16.trim();
+  if (!lower) return void 0;
+  return AUDIO_FORMAT_BY_MIME[lower];
+}
 function getCacheControl(providerOptions) {
   var _a16, _b16, _c;
   const anthropic = providerOptions == null ? void 0 : providerOptions.anthropic;
@@ -2346,7 +2369,7 @@ function convertToLLMGatewayChatMessages(prompt) {
         const messageCacheControl = getCacheControl(providerOptions);
         const contentParts = content.map(
           (part) => {
-            var _a17, _b17, _c2, _d, _e, _f;
+            var _a17, _b17, _c2, _d, _e, _f, _g;
             const cacheControl = (_a17 = getCacheControl(part.providerOptions)) != null ? _a17 : messageCacheControl;
             switch (part.type) {
               case "text":
@@ -2371,8 +2394,27 @@ function convertToLLMGatewayChatMessages(prompt) {
                     cache_control: cacheControl
                   };
                 }
+                if ((_c2 = part.mediaType) == null ? void 0 : _c2.startsWith("audio/")) {
+                  const format = audioFormatFromMime(part.mediaType);
+                  if (format) {
+                    const fileUrl = getFileUrl({
+                      part,
+                      defaultMediaType: part.mediaType
+                    });
+                    if (fileUrl.startsWith("data:")) {
+                      return {
+                        type: "input_audio",
+                        input_audio: {
+                          data: getBase64FromDataUrl(fileUrl),
+                          format
+                        },
+                        cache_control: cacheControl
+                      };
+                    }
+                  }
+                }
                 const fileName = String(
-                  (_f = (_e = (_d = (_c2 = part.providerOptions) == null ? void 0 : _c2.llmgateway) == null ? void 0 : _d.filename) != null ? _e : part.filename) != null ? _f : ""
+                  (_g = (_f = (_e = (_d = part.providerOptions) == null ? void 0 : _d.llmgateway) == null ? void 0 : _e.filename) != null ? _f : part.filename) != null ? _g : ""
                 );
                 const fileData = getFileUrl({
                   part,
@@ -3700,13 +3742,19 @@ var LLMGatewayImageModel = class {
     if (options.aspectRatio != null) {
       body.aspect_ratio = options.aspectRatio;
     }
+    if (options.quality != null) {
+      body.quality = options.quality;
+    }
+    const providerOptions = options.providerOptions || {};
+    const llmgatewayOptions = providerOptions.llmgateway || {};
+    const requestBody = __spreadValues(__spreadValues(__spreadValues(__spreadValues({}, body), this.config.extraBody), this.settings.extraBody), llmgatewayOptions);
     const { value: response, responseHeaders } = await postJsonToApi({
       url: this.config.url({
         path: hasFiles ? "/images/edits" : "/images/generations",
         modelId: this.modelId
       }),
       headers: combineHeaders(this.config.headers(), options.headers),
-      body,
+      body: requestBody,
       failedResponseHandler: llmgatewayFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
         LLMGatewayImageResponseSchema
@@ -3765,7 +3813,8 @@ function createLLMGateway(options = {}) {
     provider: "llmgateway.image",
     url: ({ path }) => `${baseURL}${path}`,
     headers: getHeaders,
-    fetch: options.fetch
+    fetch: options.fetch,
+    extraBody: options.extraBody
   });
   const createLanguageModel = (modelId, settings) => {
     if (new.target) {