ai 3.1.23 → 3.1.24

package/dist/index.mjs

@@ -1115,57 +1115,72 @@ var StreamObjectResult = class {
     warnings,
     rawResponse
   }) {
-    this.originalStream = stream;
     this.warnings = warnings;
     this.rawResponse = rawResponse;
-  }
-  get partialObjectStream() {
-    let accumulatedText = "";
-    let latestObject = void 0;
-    return createAsyncIterableStream(this.originalStream, {
-      transform(chunk, controller) {
-        if (typeof chunk === "string") {
-          accumulatedText += chunk;
-          const currentObject = parsePartialJson(
-            accumulatedText
-          );
-          if (!isDeepEqualData(latestObject, currentObject)) {
-            latestObject = currentObject;
-            controller.enqueue(currentObject);
-          }
-        } else if (chunk.type === "error") {
-          throw chunk.error;
-        }
-      }
+    let resolveUsage;
+    this.usage = new Promise((resolve) => {
+      resolveUsage = resolve;
     });
-  }
-  get fullStream() {
+    let usage;
     let accumulatedText = "";
     let latestObject = void 0;
-    return createAsyncIterableStream(this.originalStream, {
-      transform(chunk, controller) {
-        if (typeof chunk === "string") {
-          accumulatedText += chunk;
-          const currentObject = parsePartialJson(
-            accumulatedText
-          );
-          if (!isDeepEqualData(latestObject, currentObject)) {
-            latestObject = currentObject;
-            controller.enqueue({ type: "object", object: currentObject });
+    this.originalStream = stream.pipeThrough(
+      new TransformStream({
+        async transform(chunk, controller) {
+          if (typeof chunk === "string") {
+            accumulatedText += chunk;
+            const currentObject = parsePartialJson(
+              accumulatedText
+            );
+            if (!isDeepEqualData(latestObject, currentObject)) {
+              latestObject = currentObject;
+              controller.enqueue({ type: "object", object: currentObject });
+            }
+            return;
           }
-        } else {
           switch (chunk.type) {
-            case "finish":
+            case "finish": {
+              usage = calculateTokenUsage(chunk.usage);
               controller.enqueue({
                 ...chunk,
-                usage: calculateTokenUsage(chunk.usage)
+                usage
               });
+              resolveUsage(usage);
               break;
-            default:
+            }
+            default: {
               controller.enqueue(chunk);
               break;
+            }
           }
         }
+      })
+    );
+  }
+  get partialObjectStream() {
+    return createAsyncIterableStream(this.originalStream, {
+      transform(chunk, controller) {
+        switch (chunk.type) {
+          case "object":
+            controller.enqueue(chunk.object);
+            break;
+          case "finish":
+            break;
+          case "error":
+            controller.error(chunk.error);
+            break;
+          default: {
+            const _exhaustiveCheck = chunk;
+            throw new Error(`Unsupported chunk type: ${_exhaustiveCheck}`);
+          }
+        }
+      }
+    });
+  }
+  get fullStream() {
+    return createAsyncIterableStream(this.originalStream, {
+      transform(chunk, controller) {
+        controller.enqueue(chunk);
       }
     });
   }