ai 3.1.1 → 3.1.3

package/dist/index.mjs

@@ -658,6 +658,7 @@ function fixJson(input) {
             break;
           }
           case "}": {
+            lastValidIndex = i;
             stack.pop();
             break;
           }
@@ -1392,9 +1393,40 @@ 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.
+  The original stream is replaced to allow for further splitting,
+  since we do not know how many times the stream will be split.
+  
+  Note: this leads to buffering the stream content on the server.
+  However, the LLM results are expected to be small enough to not cause issues.
+     */
+  teeStream() {
+    const [stream1, stream2] = this.originalStream.tee();
+    this.originalStream = stream2;
+    return stream1;
   }
   /**
   A text stream that returns only the generated text deltas. You can use it
@@ -1402,7 +1434,7 @@ var StreamTextResult = class {
   stream will throw the error.
      */
   get textStream() {
-    return createAsyncIterableStream(this.originalStream, {
+    return createAsyncIterableStream(this.teeStream(), {
       transform(chunk, controller) {
         if (chunk.type === "text-delta") {
           if (chunk.textDelta.length > 0) {
@@ -1421,7 +1453,7 @@ var StreamTextResult = class {
   stream will throw the error.
      */
   get fullStream() {
-    return createAsyncIterableStream(this.originalStream, {
+    return createAsyncIterableStream(this.teeStream(), {
       transform(chunk, controller) {
         if (chunk.type === "text-delta") {
           if (chunk.textDelta.length > 0) {