ai 4.0.28 → 4.0.30

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() {
@@ -4606,17 +4625,18 @@ var DefaultStreamTextResult = class {
       async flush(controller) {
         var _a15;
         try {
-          const lastStep = recordedSteps[recordedSteps.length - 1];
-          if (lastStep) {
-            self.warningsPromise.resolve(lastStep.warnings);
-            self.requestPromise.resolve(lastStep.request);
-            self.responsePromise.resolve(lastStep.response);
-            self.toolCallsPromise.resolve(lastStep.toolCalls);
-            self.toolResultsPromise.resolve(lastStep.toolResults);
-            self.providerMetadataPromise.resolve(
-              lastStep.experimental_providerMetadata
-            );
+          if (recordedSteps.length === 0) {
+            return;
           }
+          const lastStep = recordedSteps[recordedSteps.length - 1];
+          self.warningsPromise.resolve(lastStep.warnings);
+          self.requestPromise.resolve(lastStep.request);
+          self.responsePromise.resolve(lastStep.response);
+          self.toolCallsPromise.resolve(lastStep.toolCalls);
+          self.toolResultsPromise.resolve(lastStep.toolResults);
+          self.providerMetadataPromise.resolve(
+            lastStep.experimental_providerMetadata
+          );
           const finishReason = recordedFinishReason != null ? recordedFinishReason : "unknown";
           const usage = recordedUsage != null ? recordedUsage : {
             completionTokens: NaN,