ai 6.0.80 → 6.0.82

package/dist/index.mjs

@@ -1100,7 +1100,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.80" : "0.0.0-test";
+var VERSION = true ? "6.0.82" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -5490,6 +5490,7 @@ function processUIMessageStream({
 function handleUIMessageStreamFinish({
   messageId,
   originalMessages = [],
+  onStepFinish,
   onFinish,
   onError,
   stream
@@ -5517,7 +5518,7 @@ function handleUIMessageStreamFinish({
       }
     })
   );
-  if (onFinish == null) {
+  if (onFinish == null && onStepFinish == null) {
     return idInjectedStream;
   }
   const state = createStreamingUIMessageState({
@@ -5547,13 +5548,34 @@ function handleUIMessageStreamFinish({
       finishReason: state.finishReason
     });
   };
+  const callOnStepFinish = async () => {
+    if (!onStepFinish) {
+      return;
+    }
+    const isContinuation = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
+    try {
+      await onStepFinish({
+        isContinuation,
+        responseMessage: structuredClone(state.message),
+        messages: [
+          ...isContinuation ? originalMessages.slice(0, -1) : originalMessages,
+          structuredClone(state.message)
+        ]
+      });
+    } catch (error) {
+      onError(error);
+    }
+  };
   return processUIMessageStream({
     stream: idInjectedStream,
     runUpdateMessageJob,
     onError
   }).pipeThrough(
     new TransformStream({
-      transform(chunk, controller) {
+      async transform(chunk, controller) {
+        if (chunk.type === "finish-step") {
+          await callOnStepFinish();
+        }
         controller.enqueue(chunk);
       },
       // @ts-expect-error cancel is still new and missing from types https://developer.mozilla.org/en-US/docs/Web/API/TransformStream#browser_compatibility
@@ -7696,6 +7718,7 @@ function createUIMessageStream({
   execute,
   onError = getErrorMessage8,
   originalMessages,
+  onStepFinish,
   onFinish,
   generateId: generateId2 = generateIdFunc
 }) {
@@ -7771,6 +7794,7 @@ function createUIMessageStream({
     stream,
     messageId: generateId2(),
     originalMessages,
+    onStepFinish,
     onFinish,
     onError
   });