ai 6.0.214 → 6.0.216

package/dist/index.mjs

@@ -1171,7 +1171,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.214" : "0.0.0-test";
+var VERSION = true ? "6.0.216" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -11693,29 +11693,44 @@ function pruneMessages({
         }
       }
     }
+    const toolCallIdToToolName = /* @__PURE__ */ new Map();
+    for (const message of messages) {
+      if ((message.role === "assistant" || message.role === "tool") && typeof message.content !== "string") {
+        for (const part of message.content) {
+          if (part.type === "tool-call" || part.type === "tool-result") {
+            toolCallIdToToolName.set(part.toolCallId, part.toolName);
+          }
+        }
+      }
+    }
+    const approvalIdToToolName = /* @__PURE__ */ new Map();
+    for (const message of messages) {
+      if ((message.role === "assistant" || message.role === "tool") && typeof message.content !== "string") {
+        for (const part of message.content) {
+          if (part.type === "tool-approval-request") {
+            const toolName = toolCallIdToToolName.get(part.toolCallId);
+            if (toolName != null) {
+              approvalIdToToolName.set(part.approvalId, toolName);
+            }
+          }
+        }
+      }
+    }
     messages = messages.map((message, messageIndex) => {
       if (message.role !== "assistant" && message.role !== "tool" || typeof message.content === "string" || keepLastMessagesCount && messageIndex >= messages.length - keepLastMessagesCount) {
         return message;
       }
-      const toolCallIdToToolName = {};
-      const approvalIdToToolName = {};
       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;
           }
-          if (part.type === "tool-call") {
-            toolCallIdToToolName[part.toolCallId] = part.toolName;
-          } else if (part.type === "tool-approval-request") {
-            approvalIdToToolName[part.approvalId] = toolCallIdToToolName[part.toolCallId];
-          }
           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;
           }
-          return toolCall.tools != null && !toolCall.tools.includes(
-            part.type === "tool-call" || part.type === "tool-result" ? part.toolName : approvalIdToToolName[part.approvalId]
-          );
+          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);
         })
       };
     });