ai 4.0.19 → 4.0.20

package/dist/index.mjs

@@ -32,15 +32,19 @@ function createDataStream({
       controller = controllerArg;
     }
   });
+  function safeEnqueue(data) {
+    try {
+      controller.enqueue(data);
+    } catch (error) {
+    }
+  }
   try {
     const result = execute({
       writeData(data) {
-        controller.enqueue(formatDataStreamPart("data", [data]));
+        safeEnqueue(formatDataStreamPart("data", [data]));
       },
       writeMessageAnnotation(annotation) {
-        controller.enqueue(
-          formatDataStreamPart("message_annotations", [annotation])
-        );
+        safeEnqueue(formatDataStreamPart("message_annotations", [annotation]));
       },
       merge(streamArg) {
         ongoingStreamPromises.push(
@@ -50,10 +54,10 @@ function createDataStream({
               const { done, value } = await reader.read();
               if (done)
                 break;
-              controller.enqueue(value);
+              safeEnqueue(value);
             }
           })().catch((error) => {
-            controller.enqueue(formatDataStreamPart("error", onError(error)));
+            safeEnqueue(formatDataStreamPart("error", onError(error)));
           })
         );
       },
@@ -62,12 +66,12 @@ function createDataStream({
     if (result) {
       ongoingStreamPromises.push(
         result.catch((error) => {
-          controller.enqueue(formatDataStreamPart("error", onError(error)));
+          safeEnqueue(formatDataStreamPart("error", onError(error)));
         })
       );
     }
   } catch (error) {
-    controller.enqueue(formatDataStreamPart("error", onError(error)));
+    safeEnqueue(formatDataStreamPart("error", onError(error)));
   }
   const waitForStreams = new Promise(async (resolve) => {
     while (ongoingStreamPromises.length > 0) {
@@ -76,7 +80,10 @@ function createDataStream({
     resolve();
   });
   waitForStreams.finally(() => {
-    controller.close();
+    try {
+      controller.close();
+    } catch (error) {
+    }
   });
   return stream;
 }
@@ -4127,25 +4134,6 @@ function runToolsTransformation({
           break;
         }
         case "tool-call": {
-          const toolName = chunk.toolName;
-          if (tools == null) {
-            toolResultsStreamController.enqueue({
-              type: "error",
-              error: new NoSuchToolError({ toolName: chunk.toolName })
-            });
-            break;
-          }
-          const tool2 = tools[toolName];
-          if (tool2 == null) {
-            toolResultsStreamController.enqueue({
-              type: "error",
-              error: new NoSuchToolError({
-                toolName: chunk.toolName,
-                availableTools: Object.keys(tools)
-              })
-            });
-            break;
-          }
           try {
             const toolCall = await parseToolCall({
               toolCall: chunk,
@@ -4155,6 +4143,7 @@ function runToolsTransformation({
               messages
             });
             controller.enqueue(toolCall);
+            const tool2 = tools[toolCall.toolName];
             if (tool2.execute != null) {
               const toolExecutionId = generateId();
               outstandingToolResults.add(toolExecutionId);