ai 4.0.28 → 4.0.29

package/dist/index.mjs

@@ -810,7 +810,7 @@ import {
 async function generateImage({
   model,
   prompt,
-  n,
+  n = 1,
   size,
   aspectRatio,
   seed,
@@ -819,27 +819,46 @@ async function generateImage({
   abortSignal,
   headers
 }) {
+  var _a14;
   const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
-  return new DefaultGenerateImageResult(
-    await retry(
-      () => model.doGenerate({
-        prompt,
-        n: n != null ? n : 1,
-        abortSignal,
-        headers,
-        size,
-        aspectRatio,
-        seed,
-        providerOptions: providerOptions != null ? providerOptions : {}
-      })
+  const maxImagesPerCall = (_a14 = model.maxImagesPerCall) != null ? _a14 : 1;
+  const callCount = Math.ceil(n / maxImagesPerCall);
+  const callImageCounts = Array.from({ length: callCount }, (_, i) => {
+    if (i < callCount - 1) {
+      return maxImagesPerCall;
+    }
+    const remainder = n % maxImagesPerCall;
+    return remainder === 0 ? maxImagesPerCall : remainder;
+  });
+  const results = await Promise.all(
+    callImageCounts.map(
+      async (callImageCount) => retry(
+        () => model.doGenerate({
+          prompt,
+          n: callImageCount,
+          abortSignal,
+          headers,
+          size,
+          aspectRatio,
+          seed,
+          providerOptions: providerOptions != null ? providerOptions : {}
+        })
+      )
     )
   );
+  const images = [];
+  const warnings = [];
+  for (const result of results) {
+    images.push(
+      ...result.images.map((image) => new DefaultGeneratedImage({ image }))
+    );
+    warnings.push(...result.warnings);
+  }
+  return new DefaultGenerateImageResult({ images, warnings });
 }
 var DefaultGenerateImageResult = class {
   constructor(options) {
-    this.images = options.images.map(
-      (image) => new DefaultGeneratedImage({ imageData: image })
-    );
+    this.images = options.images;
     this.warnings = options.warnings;
   }
   get image() {
@@ -847,10 +866,10 @@ var DefaultGenerateImageResult = class {
   }
 };
 var DefaultGeneratedImage = class {
-  constructor({ imageData }) {
-    const isUint8Array = imageData instanceof Uint8Array;
-    this.base64Data = isUint8Array ? void 0 : imageData;
-    this.uint8ArrayData = isUint8Array ? imageData : void 0;
+  constructor({ image }) {
+    const isUint8Array = image instanceof Uint8Array;
+    this.base64Data = isUint8Array ? void 0 : image;
+    this.uint8ArrayData = isUint8Array ? image : void 0;
   }
   // lazy conversion with caching to avoid unnecessary conversion overhead:
   get base64() {