ai 5.0.22 → 5.0.24

package/dist/index.mjs

@@ -3267,9 +3267,18 @@ async function parsePartialJson(jsonText) {
 function isToolUIPart(part) {
   return part.type.startsWith("tool-");
 }
+function isDynamicToolUIPart(part) {
+  return part.type === "dynamic-tool";
+}
+function isToolOrDynamicToolUIPart(part) {
+  return isToolUIPart(part) || isDynamicToolUIPart(part);
+}
 function getToolName(part) {
   return part.type.split("-").slice(1).join("-");
 }
+function getToolOrDynamicToolName(part) {
+  return isDynamicToolUIPart(part) ? part.toolName : getToolName(part);
+}
 
 // src/ui/process-ui-message-stream.ts
 function createStreamingUIMessageState({
@@ -3762,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,
@@ -3771,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();
       }
     })
   );
@@ -3816,12 +3837,57 @@ function pipeUIMessageStreamToResponse({
 // src/util/async-iterable-stream.ts
 function createAsyncIterableStream(source) {
   const stream = source.pipeThrough(new TransformStream());
-  stream[Symbol.asyncIterator] = () => {
-    const reader = stream.getReader();
+  stream[Symbol.asyncIterator] = function() {
+    const reader = this.getReader();
+    let finished = false;
+    async function cleanup(cancelStream) {
+      var _a17;
+      finished = true;
+      try {
+        if (cancelStream) {
+          await ((_a17 = reader.cancel) == null ? void 0 : _a17.call(reader));
+        }
+      } finally {
+        try {
+          reader.releaseLock();
+        } catch (e) {
+        }
+      }
+    }
     return {
+      /**
+       * Reads the next chunk from the stream.
+       * @returns A promise resolving to the next IteratorResult.
+       */
       async next() {
+        if (finished) {
+          return { done: true, value: void 0 };
+        }
         const { done, value } = await reader.read();
-        return done ? { done: true, value: void 0 } : { done: false, value };
+        if (done) {
+          await cleanup(true);
+          return { done: true, value: void 0 };
+        }
+        return { done: false, value };
+      },
+      /**
+       * Called on early exit (e.g., break from for-await).
+       * Ensures the stream is cancelled and resources are released.
+       * @returns A promise resolving to a completed IteratorResult.
+       */
+      async return() {
+        await cleanup(true);
+        return { done: true, value: void 0 };
+      },
+      /**
+       * Called on early exit with error.
+       * Ensures the stream is cancelled and resources are released, then rethrows the error.
+       * @param err The error to throw.
+       * @returns A promise that rejects with the provided error.
+       */
+      async throw(err) {
+        await cleanup(true);
+        throw err;
       }
     };
   };
@@ -9250,12 +9316,12 @@ var AbstractChat = class {
       this.state.replaceMessage(messages.length - 1, {
         ...lastMessage,
         parts: lastMessage.parts.map(
-          (part) => isToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state: "output-available", output } : part
+          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state: "output-available", output } : part
         )
       });
       if (this.activeResponse) {
         this.activeResponse.state.message.parts = this.activeResponse.state.message.parts.map(
-          (part) => isToolUIPart(part) && part.toolCallId === toolCallId ? {
+          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? {
             ...part,
             state: "output-available",
             output,
@@ -9439,7 +9505,7 @@ function convertToModelMessages(messages, options) {
     messages = messages.map((message) => ({
       ...message,
       parts: message.parts.filter(
-        (part) => !isToolUIPart(part) || part.state !== "input-streaming" && part.state !== "input-available"
+        (part) => !isToolOrDynamicToolUIPart(part) || part.state !== "input-streaming" && part.state !== "input-available"
       )
     }));
   }
@@ -9518,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,
@@ -9534,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,
@@ -9594,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 = [];
@@ -9646,7 +9701,7 @@ function lastAssistantMessageIsCompleteWithToolCalls({
   const lastStepStartIndex = message.parts.reduce((lastIndex, part, index) => {
     return part.type === "step-start" ? index : lastIndex;
   }, -1);
-  const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter((part) => isToolUIPart(part) || part.type === "dynamic-tool");
+  const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter(isToolOrDynamicToolUIPart);
   return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => part.state === "output-available");
 }
 
@@ -10111,8 +10166,10 @@ export {
   generateText,
   getTextFromDataUrl,
   getToolName,
+  getToolOrDynamicToolName,
   hasToolCall,
   isDeepEqualData,
+  isToolOrDynamicToolUIPart,
   isToolUIPart,
   jsonSchema2 as jsonSchema,
   lastAssistantMessageIsCompleteWithToolCalls,