ai 5.0.23 → 5.0.24

package/dist/index.mjs

@@ -3771,6 +3771,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,
@@ -3780,17 +3797,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();
       }
     })
   );
@@ -9572,12 +9584,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,
@@ -9588,12 +9595,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,
@@ -9648,13 +9650,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 = [];