ai 3.1.2 → 3.1.3

package/dist/index.d.mts

@@ -788,6 +788,14 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
      */
     readonly warnings: CallWarning[] | undefined;
     /**
+  The token usage of the generated text. Resolved when the response is finished.
+     */
+    readonly usage: Promise<TokenUsage>;
+    /**
+  The reason why the generation finished. Resolved when the response is finished.
+     */
+    readonly finishReason: Promise<FinishReason>;
+    /**
   Optional raw response data.
      */
     rawResponse?: {

package/dist/index.d.ts

@@ -788,6 +788,14 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
      */
     readonly warnings: CallWarning[] | undefined;
     /**
+  The token usage of the generated text. Resolved when the response is finished.
+     */
+    readonly usage: Promise<TokenUsage>;
+    /**
+  The reason why the generation finished. Resolved when the response is finished.
+     */
+    readonly finishReason: Promise<FinishReason>;
+    /**
   Optional raw response data.
      */
     rawResponse?: {

package/dist/index.js

@@ -755,6 +755,7 @@ function fixJson(input) {
             break;
           }
           case "}": {
+            lastValidIndex = i;
             stack.pop();
             break;
           }
@@ -1486,9 +1487,27 @@ var StreamTextResult = class {
     warnings,
     rawResponse
   }) {
-    this.originalStream = stream;
     this.warnings = warnings;
     this.rawResponse = rawResponse;
+    let resolveUsage;
+    this.usage = new Promise((resolve) => {
+      resolveUsage = resolve;
+    });
+    let resolveFinishReason;
+    this.finishReason = new Promise((resolve) => {
+      resolveFinishReason = resolve;
+    });
+    this.originalStream = stream.pipeThrough(
+      new TransformStream({
+        async transform(chunk, controller) {
+          controller.enqueue(chunk);
+          if (chunk.type === "finish") {
+            resolveUsage(chunk.usage);
+            resolveFinishReason(chunk.finishReason);
+          }
+        }
+      })
+    );
   }
   /**
   Split out a new stream from the original stream.