ai 6.0.312 → 6.0.313

package/dist/index.mjs

@@ -1225,7 +1225,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.312" : "0.0.0-test";
+var VERSION = true ? "6.0.313" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2876,6 +2876,7 @@ function collectToolApprovals({
   }
   const approvedToolApprovals = [];
   const deniedToolApprovals = [];
+  const collectedToolCallIds = /* @__PURE__ */ new Set();
   const approvalResponses = lastMessage.content.filter(
     (part) => part.type === "tool-approval-response"
   );
@@ -2896,6 +2897,10 @@ function collectToolApprovals({
         approvalId: approvalRequest.approvalId
       });
     }
+    if (collectedToolCallIds.has(toolCall.toolCallId)) {
+      continue;
+    }
+    collectedToolCallIds.add(toolCall.toolCallId);
     const approval = {
       approvalRequest,
       approvalResponse,
@@ -6075,6 +6080,7 @@ function processUIMessageStream({
             const anyPart = part;
             if (part != null) {
               part.state = options.state;
+              anyPart.type = `tool-${options.toolName}`;
               anyPart.input = anyOptions.input;
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
@@ -12805,18 +12811,53 @@ function pruneMessages({
       if (message.role !== "assistant" && message.role !== "tool" || typeof message.content === "string" || keepLastMessagesCount && messageIndex >= messages.length - keepLastMessagesCount) {
         return message;
       }
-      return {
-        ...message,
-        content: message.content.filter((part) => {
-          if (part.type !== "tool-call" && part.type !== "tool-result" && part.type !== "tool-approval-request" && part.type !== "tool-approval-response") {
-            return true;
+      const keepParts = message.content.map((part) => {
+        if (part.type !== "tool-call" && part.type !== "tool-result" && part.type !== "tool-approval-request" && part.type !== "tool-approval-response") {
+          return true;
+        }
+        if ((part.type === "tool-call" || part.type === "tool-result") && keptToolCallIds.has(part.toolCallId) || (part.type === "tool-approval-request" || part.type === "tool-approval-response") && keptApprovalIds.has(part.approvalId)) {
+          return true;
+        }
+        const partToolName = part.type === "tool-call" || part.type === "tool-result" ? part.toolName : approvalIdToToolName.get(part.approvalId);
+        return toolCall.tools != null && partToolName != null && !toolCall.tools.includes(partToolName);
+      });
+      if (message.role === "assistant") {
+        for (let reasoningIndex = 0; reasoningIndex < message.content.length; reasoningIndex++) {
+          const part = message.content[reasoningIndex];
+          if (part.type !== "reasoning") {
+            continue;
+          }
+          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) || (part.type === "tool-approval-request" || part.type === "tool-approval-response") && keptApprovalIds.has(part.approvalId)) {
-            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++;
           }
-          const partToolName = part.type === "tool-call" || part.type === "tool-result" ? part.toolName : approvalIdToToolName.get(part.approvalId);
-          return toolCall.tools != null && partToolName != null && !toolCall.tools.includes(partToolName);
-        })
+          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])
       };
     });
   }