ai 5.0.278 → 5.0.280

package/dist/index.mjs

@@ -727,7 +727,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.278" : "0.0.0-test";
+var VERSION = true ? "5.0.280" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5349,12 +5349,23 @@ var DefaultStreamTextResult = class {
         return stitchableStream.stream.cancel(reason);
       }
     });
+    let isRunning = true;
+    stream = stream.pipeThrough(
+      new TransformStream({
+        async transform(chunk, controller) {
+          if (isRunning) {
+            controller.enqueue(chunk);
+          }
+        }
+      })
+    );
     for (const transform of transforms) {
       stream = stream.pipeThrough(
         transform({
           tools,
           stopStream() {
             stitchableStream.terminate();
+            isRunning = false;
           }
         })
       );
@@ -8873,21 +8884,52 @@ function pruneMessages({
       if (message.role !== "assistant" && message.role !== "tool" || typeof message.content === "string" || keepLastMessagesCount && messageIndex >= messages.length - keepLastMessagesCount) {
         return message;
       }
-      const toolCallIdToToolName = {};
-      return {
-        ...message,
-        content: message.content.filter((part) => {
-          if (part.type !== "tool-call" && part.type !== "tool-result") {
-            return true;
+      const keepParts = message.content.map((part) => {
+        if (part.type !== "tool-call" && part.type !== "tool-result") {
+          return true;
+        }
+        if ((part.type === "tool-call" || part.type === "tool-result") && keptToolCallIds.has(part.toolCallId)) {
+          return true;
+        }
+        return toolCall.tools != null && !toolCall.tools.includes(part.toolName);
+      });
+      if (message.role === "assistant") {
+        for (let reasoningIndex = 0; reasoningIndex < message.content.length; reasoningIndex++) {
+          const part = message.content[reasoningIndex];
+          if (part.type !== "reasoning") {
+            continue;
           }
-          if (part.type === "tool-call") {
-            toolCallIdToToolName[part.toolCallId] = part.toolName;
+          let reasoningEndIndex = reasoningIndex + 1;
+          while (reasoningEndIndex < message.content.length) {
+            const nextPart = message.content[reasoningEndIndex];
+            if (nextPart.type !== "reasoning") {
+              break;
+            }
+            reasoningEndIndex++;
           }
-          if ((part.type === "tool-call" || part.type === "tool-result") && keptToolCallIds.has(part.toolCallId)) {
-            return true;
+          let followingPartIndex = reasoningEndIndex;
+          let hasFollowingPart = false;
+          let hasKeptFollowingPart = false;
+          while (followingPartIndex < message.content.length) {
+            const followingPart = message.content[followingPartIndex];
+            if (followingPart.type === "reasoning") {
+              break;
+            }
+            hasFollowingPart = true;
+            hasKeptFollowingPart || (hasKeptFollowingPart = keepParts[followingPartIndex]);
+            followingPartIndex++;
           }
-          return toolCall.tools != null && !toolCall.tools.includes(part.toolName);
-        })
+          if (hasFollowingPart && !hasKeptFollowingPart) {
+            for (let index = reasoningIndex; index < reasoningEndIndex; index++) {
+              keepParts[index] = false;
+            }
+          }
+          reasoningIndex = followingPartIndex - 1;
+        }
+      }
+      return {
+        ...message,
+        content: message.content.filter((_, index) => keepParts[index])
       };
     });
   }