ai 3.1.28 → 3.1.29

package/dist/index.mjs

@@ -993,6 +993,7 @@ async function streamObject({
   messages,
   maxRetries,
   abortSignal,
+  onFinish,
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
@@ -1108,7 +1109,8 @@ async function streamObject({
     stream: result.stream.pipeThrough(new TransformStream(transformer)),
     warnings: result.warnings,
     rawResponse: result.rawResponse,
-    schema
+    schema,
+    onFinish
   });
 }
 var StreamObjectResult = class {
@@ -1116,7 +1118,8 @@ var StreamObjectResult = class {
     stream,
     warnings,
     rawResponse,
-    schema
+    schema,
+    onFinish
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -1131,6 +1134,8 @@ var StreamObjectResult = class {
       resolveUsage = resolve;
     });
     let usage;
+    let object;
+    let error;
     let accumulatedText = "";
     let latestObject = void 0;
     this.originalStream = stream.pipeThrough(
@@ -1157,9 +1162,11 @@ var StreamObjectResult = class {
                 schema
               });
               if (validationResult.success) {
-                resolveObject(validationResult.value);
+                object = validationResult.value;
+                resolveObject(object);
               } else {
-                rejectObject(validationResult.error);
+                error = validationResult.error;
+                rejectObject(error);
               }
               break;
             }
@@ -1168,6 +1175,24 @@ var StreamObjectResult = class {
               break;
             }
           }
+        },
+        // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
+        async flush(controller) {
+          try {
+            await (onFinish == null ? void 0 : onFinish({
+              usage: usage != null ? usage : {
+                promptTokens: NaN,
+                completionTokens: NaN,
+                totalTokens: NaN
+              },
+              object,
+              error,
+              rawResponse,
+              warnings
+            }));
+          } catch (error2) {
+            controller.error(error2);
+          }
         }
       })
     );