@llmgateway/ai-sdk-provider 3.3.0 → 3.5.0

package/dist/index.mjs

@@ -2515,10 +2515,7 @@ var LLMGatewayChatCompletionBaseResponseSchema = z6.object({
       reasoning_tokens: z6.number()
     }).nullish(),
     total_tokens: z6.number(),
-    cost: z6.union([
-      z6.number(),
-      z6.object({ total_cost: z6.number() }).passthrough()
-    ]).optional(),
+    cost: z6.union([z6.number(), z6.object({ total_cost: z6.number() }).passthrough()]).optional(),
     cost_details: z6.object({
       upstream_inference_cost: z6.number().nullish()
     }).nullish()
@@ -3607,30 +3604,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),