ai 3.3.6 → 3.3.8

package/dist/index.mjs

@@ -22,7 +22,7 @@ import { getErrorMessage, isAbortError } from "@ai-sdk/provider-utils";
 
 // util/delay.ts
 async function delay(delayInMs) {
-  return new Promise((resolve) => setTimeout(resolve, delayInMs));
+  return delayInMs === void 0 ? Promise.resolve() : new Promise((resolve) => setTimeout(resolve, delayInMs));
 }
 
 // util/retry-error.ts
@@ -2776,6 +2776,7 @@ async function streamText({
   headers,
   experimental_telemetry: telemetry,
   experimental_toolCallStreaming: toolCallStreaming = false,
+  onChunk,
   onFinish,
   ...settings
 }) {
@@ -2867,6 +2868,7 @@ async function streamText({
         }),
         warnings,
         rawResponse,
+        onChunk,
         onFinish,
         rootSpan,
         doStreamSpan,
@@ -2880,6 +2882,7 @@ var DefaultStreamTextResult = class {
     stream,
     warnings,
     rawResponse,
+    onChunk,
     onFinish,
     rootSpan,
     doStreamSpan,
@@ -2887,7 +2890,6 @@ var DefaultStreamTextResult = class {
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
-    this.onFinish = onFinish;
     const { resolve: resolveUsage, promise: usagePromise } = createResolvablePromise();
     this.usage = usagePromise;
     const { resolve: resolveFinishReason, promise: finishReasonPromise } = createResolvablePromise();
@@ -2904,25 +2906,30 @@ var DefaultStreamTextResult = class {
     const toolCalls = [];
     const toolResults = [];
     let firstChunk = true;
-    const self = this;
     this.originalStream = stream.pipeThrough(
       new TransformStream({
         async transform(chunk, controller) {
-          controller.enqueue(chunk);
           if (firstChunk) {
             firstChunk = false;
             doStreamSpan.addEvent("ai.stream.firstChunk");
           }
+          if (chunk.type === "text-delta" && chunk.textDelta.length === 0) {
+            return;
+          }
+          controller.enqueue(chunk);
           const chunkType = chunk.type;
           switch (chunkType) {
             case "text-delta":
               text += chunk.textDelta;
+              await (onChunk == null ? void 0 : onChunk({ chunk }));
               break;
             case "tool-call":
               toolCalls.push(chunk);
+              await (onChunk == null ? void 0 : onChunk({ chunk }));
               break;
             case "tool-result":
               toolResults.push(chunk);
+              await (onChunk == null ? void 0 : onChunk({ chunk }));
               break;
             case "finish":
               usage = chunk.usage;
@@ -2933,7 +2940,10 @@ var DefaultStreamTextResult = class {
               resolveToolCalls(toolCalls);
               break;
             case "tool-call-streaming-start":
-            case "tool-call-delta":
+            case "tool-call-delta": {
+              await (onChunk == null ? void 0 : onChunk({ chunk }));
+              break;
+            }
             case "error":
               break;
             default: {
@@ -2944,7 +2954,6 @@ var DefaultStreamTextResult = class {
         },
         // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
         async flush(controller) {
-          var _a12;
           try {
             const finalUsage = usage != null ? usage : {
               promptTokens: NaN,
@@ -2983,7 +2992,7 @@ var DefaultStreamTextResult = class {
               })
             );
             resolveToolResults(toolResults);
-            await ((_a12 = self.onFinish) == null ? void 0 : _a12.call(self, {
+            await (onFinish == null ? void 0 : onFinish({
               finishReason: finalFinishReason,
               usage: finalUsage,
               text,
@@ -3022,9 +3031,7 @@ var DefaultStreamTextResult = class {
     return createAsyncIterableStream(this.teeStream(), {
       transform(chunk, controller) {
         if (chunk.type === "text-delta") {
-          if (chunk.textDelta.length > 0) {
-            controller.enqueue(chunk.textDelta);
-          }
+          controller.enqueue(chunk.textDelta);
         } else if (chunk.type === "error") {
           controller.error(chunk.error);
         }
@@ -3034,13 +3041,7 @@ var DefaultStreamTextResult = class {
   get fullStream() {
     return createAsyncIterableStream(this.teeStream(), {
       transform(chunk, controller) {
-        if (chunk.type === "text-delta") {
-          if (chunk.textDelta.length > 0) {
-            controller.enqueue(chunk);
-          }
-        } else {
-          controller.enqueue(chunk);
-        }
+        controller.enqueue(chunk);
       }
     });
   }
@@ -3695,7 +3696,11 @@ function readableFromAsyncIterable(iterable) {
 
 // streams/stream-data.ts
 import { formatStreamPart as formatStreamPart2 } from "@ai-sdk/ui-utils";
-var STREAM_DATA_WARNING_TIME_MS = 15 * 1e3;
+
+// util/constants.ts
+var HANGING_STREAM_WARNING_TIME_MS = 15 * 1e3;
+
+// streams/stream-data.ts
 var StreamData2 = class {
   constructor() {
     this.encoder = new TextEncoder();
@@ -3711,7 +3716,7 @@ var StreamData2 = class {
             console.warn(
               "The data stream is hanging. Did you forget to close it with `data.close()`?"
             );
-          }, STREAM_DATA_WARNING_TIME_MS);
+          }, HANGING_STREAM_WARNING_TIME_MS);
         }
       },
       pull: (controller) => {