ai 4.0.26 → 4.0.27

package/dist/index.mjs

@@ -225,7 +225,7 @@ import { getErrorMessage, isAbortError } from "@ai-sdk/provider-utils";
 
 // util/delay.ts
 async function delay(delayInMs) {
-  return delayInMs === void 0 ? Promise.resolve() : new Promise((resolve) => setTimeout(resolve, delayInMs));
+  return delayInMs == null ? Promise.resolve() : new Promise((resolve) => setTimeout(resolve, delayInMs));
 }
 
 // util/retry-error.ts
@@ -5309,38 +5309,50 @@ var DefaultStreamTextResult = class {
 };
 
 // core/generate-text/smooth-stream.ts
+import { InvalidArgumentError as InvalidArgumentError2 } from "@ai-sdk/provider";
+var CHUNKING_REGEXPS = {
+  word: /\s*\S+\s+/m,
+  line: /[^\n]*\n/m
+};
 function smoothStream({
   delayInMs = 10,
   chunking = "word",
   _internal: { delay: delay2 = delay } = {}
 } = {}) {
-  let buffer = "";
-  return () => new TransformStream({
-    async transform(chunk, controller) {
-      if (chunk.type === "step-finish") {
-        if (buffer.length > 0) {
-          controller.enqueue({ type: "text-delta", textDelta: buffer });
-          buffer = "";
+  const chunkingRegexp = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking;
+  if (chunkingRegexp == null) {
+    throw new InvalidArgumentError2({
+      argument: "chunking",
+      message: `Chunking must be "word" or "line" or a RegExp. Received: ${chunking}`
+    });
+  }
+  return () => {
+    let buffer = "";
+    return new TransformStream({
+      async transform(chunk, controller) {
+        if (chunk.type === "step-finish") {
+          if (buffer.length > 0) {
+            controller.enqueue({ type: "text-delta", textDelta: buffer });
+            buffer = "";
+          }
+          controller.enqueue(chunk);
+          return;
         }
-        controller.enqueue(chunk);
-        return;
-      }
-      if (chunk.type !== "text-delta") {
-        controller.enqueue(chunk);
-        return;
-      }
-      buffer += chunk.textDelta;
-      const regexp = chunking === "line" ? /[^\n]*\n/m : /\s*\S+\s+/m;
-      while (regexp.test(buffer)) {
-        const chunk2 = buffer.match(regexp)[0];
-        controller.enqueue({ type: "text-delta", textDelta: chunk2 });
-        buffer = buffer.slice(chunk2.length);
-        if (delayInMs > 0) {
+        if (chunk.type !== "text-delta") {
+          controller.enqueue(chunk);
+          return;
+        }
+        buffer += chunk.textDelta;
+        let match;
+        while ((match = chunkingRegexp.exec(buffer)) != null) {
+          const chunk2 = match[0];
+          controller.enqueue({ type: "text-delta", textDelta: chunk2 });
+          buffer = buffer.slice(chunk2.length);
           await delay2(delayInMs);
         }
       }
-    }
-  });
+    });
+  };
 }
 
 // core/middleware/wrap-language-model.ts
@@ -5524,6 +5536,28 @@ function magnitude(vector) {
   return Math.sqrt(dotProduct(vector, vector));
 }
 
+// core/util/simulate-readable-stream.ts
+function simulateReadableStream({
+  chunks,
+  initialDelayInMs = 0,
+  chunkDelayInMs = 0,
+  _internal
+}) {
+  var _a14;
+  const delay2 = (_a14 = _internal == null ? void 0 : _internal.delay) != null ? _a14 : delay;
+  let index = 0;
+  return new ReadableStream({
+    async pull(controller) {
+      if (index < chunks.length) {
+        await delay2(index === 0 ? initialDelayInMs : chunkDelayInMs);
+        controller.enqueue(chunks[index++]);
+      } else {
+        controller.close();
+      }
+    }
+  });
+}
+
 // streams/assistant-response.ts
 import {
   formatAssistantStreamPart
@@ -5909,6 +5943,7 @@ export {
   pipeDataStreamToResponse,
   processDataStream,
   processTextStream,
+  simulateReadableStream,
   smoothStream,
   streamObject,
   streamText,