ai 7.0.4 → 7.0.5

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # ai
 
+## 7.0.5
+
+### Patch Changes
+
+- a2750db: fix(ai): prune orphaned tool-approval responses in `pruneMessages`
+
+  When pruning a specific tool by name (`toolCalls: [{ type, tools: [...] }]`), `pruneMessages` left the tool's `tool-approval-response` in place while removing its `tool-approval-request` and `tool-call`. The tool name of an approval response was resolved per-message, but approval responses live in a separate `tool` message from their approval request, so the name could never be resolved and the response was always kept. Tool name resolution is now done across all messages, so approval requests and responses are pruned together.
+
 ## 7.0.4
 
 ### Patch Changes

package/dist/index.js

@@ -1089,7 +1089,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.4" : "0.0.0-test";
+var VERSION = true ? "7.0.5" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -13262,29 +13262,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);
         })
       };
     });