ai 6.0.0-beta.43 → 6.0.0-beta.45

package/dist/index.mjs

@@ -720,7 +720,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.43" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.45" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -3044,7 +3044,12 @@ function writeToServerResponse({
         const { done, value } = await reader.read();
         if (done)
           break;
-        response.write(value);
+        const canContinue = response.write(value);
+        if (!canContinue) {
+          await new Promise((resolve3) => {
+            response.once("drain", resolve3);
+          });
+        }
       }
     } catch (error) {
       throw error;
@@ -8401,6 +8406,85 @@ var object = ({
   };
 };
 
+// src/generate-text/prune-messages.ts
+function pruneMessages({
+  messages,
+  reasoning = "none",
+  toolCalls = [],
+  emptyMessages = "remove"
+}) {
+  if (reasoning === "all" || reasoning === "before-last-message") {
+    messages = messages.map((message, messageIndex) => {
+      if (message.role !== "assistant" || typeof message.content === "string" || reasoning === "before-last-message" && messageIndex === messages.length - 1) {
+        return message;
+      }
+      return {
+        ...message,
+        content: message.content.filter((part) => part.type !== "reasoning")
+      };
+    });
+  }
+  if (toolCalls === "none") {
+    toolCalls = [];
+  } else if (toolCalls === "all") {
+    toolCalls = [{ type: "all" }];
+  } else if (toolCalls === "before-last-message") {
+    toolCalls = [{ type: "before-last-message" }];
+  } else if (typeof toolCalls === "string") {
+    toolCalls = [{ type: toolCalls }];
+  }
+  for (const toolCall of toolCalls) {
+    const keepLastMessagesCount = toolCall.type === "all" ? void 0 : toolCall.type === "before-last-message" ? 1 : Number(
+      toolCall.type.slice("before-last-".length).slice(0, -"-messages".length)
+    );
+    const keptToolCallIds = /* @__PURE__ */ new Set();
+    const keptApprovalIds = /* @__PURE__ */ new Set();
+    if (keepLastMessagesCount != null) {
+      for (const message of messages.slice(0, -keepLastMessagesCount)) {
+        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") {
+              keptToolCallIds.add(part.toolCallId);
+            } else if (part.type === "tool-approval-request" || part.type === "tool-approval-response") {
+              keptApprovalIds.add(part.approvalId);
+            }
+          }
+        }
+      }
+    }
+    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]
+          );
+        })
+      };
+    });
+  }
+  if (emptyMessages === "remove") {
+    messages = messages.filter((message) => message.content.length > 0);
+  }
+  return messages;
+}
+
 // src/generate-text/smooth-stream.ts
 import { delay as originalDelay } from "@ai-sdk/provider-utils";
 import { InvalidArgumentError as InvalidArgumentError2 } from "@ai-sdk/provider";
@@ -10971,6 +11055,7 @@ export {
   parsePartialJson,
   pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse,
+  pruneMessages,
   readUIMessageStream,
   safeValidateUIMessages,
   simulateReadableStream,