ai 5.0.212 → 5.0.213

package/dist/index.mjs

@@ -689,7 +689,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.212" : "0.0.0-test";
+var VERSION = true ? "5.0.213" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -4245,11 +4245,35 @@ function runToolsTransformation({
   experimental_context
 }) {
   let toolResultsStreamController = null;
+  let toolResultsStreamClosed = false;
   const toolResultsStream = new ReadableStream({
     start(controller) {
       toolResultsStreamController = controller;
+    },
+    cancel() {
+      toolResultsStreamClosed = true;
     }
   });
+  function enqueueToolResult(chunk) {
+    if (toolResultsStreamClosed) {
+      return;
+    }
+    try {
+      toolResultsStreamController.enqueue(chunk);
+    } catch (e) {
+      toolResultsStreamClosed = true;
+    }
+  }
+  function closeToolResultsStream() {
+    if (toolResultsStreamClosed) {
+      return;
+    }
+    toolResultsStreamClosed = true;
+    try {
+      toolResultsStreamController.close();
+    } catch (e) {
+    }
+  }
   const outstandingToolResults = /* @__PURE__ */ new Set();
   const toolInputs = /* @__PURE__ */ new Map();
   let canClose = false;
@@ -4257,9 +4281,9 @@ function runToolsTransformation({
   function attemptClose() {
     if (canClose && outstandingToolResults.size === 0) {
       if (finishChunk != null) {
-        toolResultsStreamController.enqueue(finishChunk);
+        enqueueToolResult(finishChunk);
       }
-      toolResultsStreamController.close();
+      closeToolResultsStream();
     }
   }
   const forwardStream = new TransformStream({
@@ -4313,7 +4337,7 @@ function runToolsTransformation({
             });
             controller.enqueue(toolCall);
             if (toolCall.invalid) {
-              toolResultsStreamController.enqueue({
+              enqueueToolResult({
                 type: "tool-error",
                 toolCallId: toolCall.toolCallId,
                 toolName: toolCall.toolName,
@@ -4368,7 +4392,7 @@ function runToolsTransformation({
                       }
                     });
                     for await (const part of stream) {
-                      toolResultsStreamController.enqueue({
+                      enqueueToolResult({
                         ...toolCall,
                         type: "tool-result",
                         output: part.output,
@@ -4382,7 +4406,7 @@ function runToolsTransformation({
                     }
                   } catch (error) {
                     recordErrorOnSpan(span, error);
-                    toolResultsStreamController.enqueue({
+                    enqueueToolResult({
                       ...toolCall,
                       type: "tool-error",
                       error
@@ -4410,14 +4434,14 @@ function runToolsTransformation({
               });
             }
           } catch (error) {
-            toolResultsStreamController.enqueue({ type: "error", error });
+            enqueueToolResult({ type: "error", error });
           }
           break;
         }
         case "tool-result": {
           const toolName = chunk.toolName;
           if (chunk.isError) {
-            toolResultsStreamController.enqueue({
+            enqueueToolResult({
               type: "tool-error",
               toolCallId: chunk.toolCallId,
               toolName,
@@ -9790,7 +9814,19 @@ var uiMessagesSchema = lazyValidator2(
               })
             })
           ])
-        ).nonempty("Message must contain at least one part")
+        )
+      }).superRefine((message, context) => {
+        if (message.role !== "assistant" && message.parts.length === 0) {
+          context.addIssue({
+            origin: "array",
+            code: "too_small",
+            minimum: 1,
+            inclusive: true,
+            input: message.parts,
+            path: ["parts"],
+            message: "Message must contain at least one part"
+          });
+        }
       })
     ).nonempty("Messages array must not be empty")
   )