@llmgateway/ai-sdk-provider 3.3.0 → 3.5.0

package/dist/index.js

@@ -2548,10 +2548,7 @@ var LLMGatewayChatCompletionBaseResponseSchema = import_v45.z.object({
       reasoning_tokens: import_v45.z.number()
     }).nullish(),
     total_tokens: import_v45.z.number(),
-    cost: import_v45.z.union([
-      import_v45.z.number(),
-      import_v45.z.object({ total_cost: import_v45.z.number() }).passthrough()
-    ]).optional(),
+    cost: import_v45.z.union([import_v45.z.number(), import_v45.z.object({ total_cost: import_v45.z.number() }).passthrough()]).optional(),
     cost_details: import_v45.z.object({
       upstream_inference_cost: import_v45.z.number().nullish()
     }).nullish()
@@ -3640,30 +3637,39 @@ var LLMGatewayImageModel = class {
   }
   async doGenerate(options) {
     const warnings = [];
-    if (options.aspectRatio != null) {
-      warnings.push({
-        type: "unsupported",
-        feature: "aspectRatio"
-      });
-    }
     if (options.seed != null) {
       warnings.push({
         type: "unsupported",
         feature: "seed"
       });
     }
+    const hasFiles = options.files != null && options.files.length > 0;
     const body = {
       model: this.modelId,
       prompt: options.prompt,
       n: options.n,
       response_format: "b64_json"
     };
+    if (hasFiles) {
+      body.images = options.files.map((file) => {
+        var _a16;
+        if (file.type === "url") {
+          return { image_url: file.url };
+        }
+        const base64 = typeof file.data === "string" ? file.data : Buffer.from(file.data).toString("base64");
+        const mediaType = (_a16 = file.mediaType) != null ? _a16 : "image/png";
+        return { image_url: `data:${mediaType};base64,${base64}` };
+      });
+    }
     if (options.size != null) {
       body.size = options.size;
     }
+    if (options.aspectRatio != null) {
+      body.aspect_ratio = options.aspectRatio;
+    }
     const { value: response, responseHeaders } = await postJsonToApi({
       url: this.config.url({
-        path: "/images/generations",
+        path: hasFiles ? "/images/edits" : "/images/generations",
         modelId: this.modelId
       }),
       headers: combineHeaders(this.config.headers(), options.headers),