ai 5.0.23 → 5.0.24

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # ai
 
+## 5.0.24
+
+### Patch Changes
+
+- f8f3682: fix: call onFinish when stream is cancelled in toUIMessageStream
+
+  Previously, onFinish was only called on normal stream completion. Now it's also called when the reader is cancelled (e.g., browser close, navigation), ensuring partial messages are persisted.
+
+- Updated dependencies [1b5a3d3]
+- Updated dependencies [c9994f9]
+  - @ai-sdk/provider-utils@3.0.6
+  - @ai-sdk/gateway@1.0.13
+
 ## 5.0.23
 
 ### Patch Changes

package/dist/index.js

@@ -3838,6 +3838,23 @@ function handleUIMessageStreamFinish({
     await job({ state, write: () => {
     } });
   };
+  let finishCalled = false;
+  const callOnFinish = async () => {
+    if (finishCalled || !onFinish) {
+      return;
+    }
+    finishCalled = true;
+    const isContinuation = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
+    await onFinish({
+      isAborted,
+      isContinuation,
+      responseMessage: state.message,
+      messages: [
+        ...isContinuation ? originalMessages.slice(0, -1) : originalMessages,
+        state.message
+      ]
+    });
+  };
   return processUIMessageStream({
     stream: idInjectedStream,
     runUpdateMessageJob,
@@ -3847,17 +3864,12 @@ function handleUIMessageStreamFinish({
       transform(chunk, controller) {
         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
+      async cancel() {
+        await callOnFinish();
+      },
       async flush() {
-        const isContinuation = state.message.id === (lastMessage == null ? void 0 : lastMessage.id);
-        await onFinish({
-          isAborted,
-          isContinuation,
-          responseMessage: state.message,
-          messages: [
-            ...isContinuation ? originalMessages.slice(0, -1) : originalMessages,
-            state.message
-          ]
-        });
+        await callOnFinish();
       }
     })
   );
@@ -9609,12 +9621,7 @@ function convertToModelMessages(messages, options) {
                 });
               } else if (part.type === "dynamic-tool") {
                 const toolName = part.toolName;
-                if (part.state === "input-streaming") {
-                  throw new MessageConversionError({
-                    originalMessage: message,
-                    message: `incomplete tool input is not supported: ${part.toolCallId}`
-                  });
-                } else {
+                if (part.state !== "input-streaming") {
                   content.push({
                     type: "tool-call",
                     toolCallId: part.toolCallId,
@@ -9625,12 +9632,7 @@ function convertToModelMessages(messages, options) {
                 }
               } else if (isToolUIPart(part)) {
                 const toolName = getToolName(part);
-                if (part.state === "input-streaming") {
-                  throw new MessageConversionError({
-                    originalMessage: message,
-                    message: `incomplete tool input is not supported: ${part.toolCallId}`
-                  });
-                } else {
+                if (part.state !== "input-streaming") {
                   content.push({
                     type: "tool-call",
                     toolCallId: part.toolCallId,
@@ -9685,13 +9687,12 @@ function convertToModelMessages(messages, options) {
                       };
                     }
                     default: {
-                      throw new MessageConversionError({
-                        originalMessage: message,
-                        message: `Unsupported tool part state: ${toolPart.state}`
-                      });
+                      return null;
                     }
                   }
-                })
+                }).filter(
+                  (output) => output != null
+                )
               });
             }
             block = [];