ai 3.1.27 → 3.1.29

package/dist/index.mjs

@@ -983,6 +983,7 @@ function parsePartialJson(jsonText) {
 }
 
 // core/generate-object/stream-object.ts
+import { safeValidateTypes } from "@ai-sdk/provider-utils";
 async function streamObject({
   model,
   schema,
@@ -992,6 +993,7 @@ async function streamObject({
   messages,
   maxRetries,
   abortSignal,
+  onFinish,
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
@@ -1106,22 +1108,34 @@ async function streamObject({
   return new StreamObjectResult({
     stream: result.stream.pipeThrough(new TransformStream(transformer)),
     warnings: result.warnings,
-    rawResponse: result.rawResponse
+    rawResponse: result.rawResponse,
+    schema,
+    onFinish
   });
 }
 var StreamObjectResult = class {
   constructor({
     stream,
     warnings,
-    rawResponse
+    rawResponse,
+    schema,
+    onFinish
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
+    let resolveObject;
+    let rejectObject;
+    this.object = new Promise((resolve, reject) => {
+      resolveObject = resolve;
+      rejectObject = reject;
+    });
     let resolveUsage;
     this.usage = new Promise((resolve) => {
       resolveUsage = resolve;
     });
     let usage;
+    let object;
+    let error;
     let accumulatedText = "";
     let latestObject = void 0;
     this.originalStream = stream.pipeThrough(
@@ -1141,11 +1155,19 @@ var StreamObjectResult = class {
           switch (chunk.type) {
             case "finish": {
               usage = calculateTokenUsage(chunk.usage);
-              controller.enqueue({
-                ...chunk,
-                usage
-              });
+              controller.enqueue({ ...chunk, usage });
               resolveUsage(usage);
+              const validationResult = safeValidateTypes({
+                value: latestObject,
+                schema
+              });
+              if (validationResult.success) {
+                object = validationResult.value;
+                resolveObject(object);
+              } else {
+                error = validationResult.error;
+                rejectObject(error);
+              }
               break;
             }
             default: {
@@ -1153,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);
+          }
         }
       })
     );