ai 6.0.0-beta.97 → 6.0.0-beta.99

package/dist/index.mjs

@@ -757,7 +757,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.97" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.99" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -1862,6 +1862,13 @@ function addLanguageModelUsage(usage1, usage2) {
 function addTokenCounts(tokenCount1, tokenCount2) {
   return tokenCount1 == null && tokenCount2 == null ? void 0 : (tokenCount1 != null ? tokenCount1 : 0) + (tokenCount2 != null ? tokenCount2 : 0);
 }
+function addImageModelUsage(usage1, usage2) {
+  return {
+    inputTokens: addTokenCounts(usage1.inputTokens, usage2.inputTokens),
+    outputTokens: addTokenCounts(usage1.outputTokens, usage2.outputTokens),
+    totalTokens: addTokenCounts(usage1.totalTokens, usage2.totalTokens)
+  };
+}
 
 // src/util/as-array.ts
 function asArray(value) {
@@ -6627,14 +6634,26 @@ var ToolLoopAgent = class {
   /**
    * Generates an output from the agent (non-streaming).
    */
-  async generate(options) {
-    return generateText(await this.prepareCall(options));
+  async generate({
+    abortSignal,
+    ...options
+  }) {
+    return generateText({
+      ...await this.prepareCall(options),
+      abortSignal
+    });
   }
   /**
    * Streams an output from the agent (streaming).
    */
-  async stream(options) {
-    return streamText(await this.prepareCall(options));
+  async stream({
+    abortSignal,
+    ...options
+  }) {
+    return streamText({
+      ...await this.prepareCall(options),
+      abortSignal
+    });
   }
 };
 
@@ -7861,6 +7880,11 @@ async function generateImage({
   const warnings = [];
   const responses = [];
   const providerMetadata = {};
+  let totalUsage = {
+    inputTokens: void 0,
+    outputTokens: void 0,
+    totalTokens: void 0
+  };
   for (const result of results) {
     images.push(
       ...result.images.map(
@@ -7877,6 +7901,9 @@ async function generateImage({
       )
     );
     warnings.push(...result.warnings);
+    if (result.usage != null) {
+      totalUsage = addImageModelUsage(totalUsage, result.usage);
+    }
     if (result.providerMetadata) {
       for (const [providerName, metadata] of Object.entries(result.providerMetadata)) {
         (_b = providerMetadata[providerName]) != null ? _b : providerMetadata[providerName] = { images: [] };
@@ -7895,7 +7922,8 @@ async function generateImage({
     images,
     warnings,
     responses,
-    providerMetadata
+    providerMetadata,
+    usage: totalUsage
   });
 }
 var DefaultGenerateImageResult = class {
@@ -7904,6 +7932,7 @@ var DefaultGenerateImageResult = class {
     this.warnings = options.warnings;
     this.responses = options.responses;
     this.providerMetadata = options.providerMetadata;
+    this.usage = options.usage;
   }
   get image() {
     return this.images[0];