ai 5.0.22 → 5.0.23

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({
@@ -3816,12 +3825,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 +9304,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 +9493,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"
       )
     }));
   }
@@ -9646,7 +9700,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 +10165,10 @@ export {
   generateText,
   getTextFromDataUrl,
   getToolName,
+  getToolOrDynamicToolName,
   hasToolCall,
   isDeepEqualData,
+  isToolOrDynamicToolUIPart,
   isToolUIPart,
   jsonSchema2 as jsonSchema,
   lastAssistantMessageIsCompleteWithToolCalls,