ai 4.0.14 → 4.0.16

package/dist/index.mjs

@@ -4204,6 +4204,7 @@ function streamText({
   experimental_toolCallStreaming: toolCallStreaming = false,
   experimental_activeTools: activeTools,
   experimental_repairToolCall: repairToolCall,
+  experimental_transform: transform,
   onChunk,
   onFinish,
   onStepFinish,
@@ -4227,6 +4228,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4254,6 +4256,7 @@ var DefaultStreamTextResult = class {
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4276,7 +4279,6 @@ var DefaultStreamTextResult = class {
     this.requestPromise = new DelayedPromise();
     this.responsePromise = new DelayedPromise();
     this.stepsPromise = new DelayedPromise();
-    this.stitchableStream = createStitchableStream();
     if (maxSteps < 1) {
       throw new InvalidArgumentError({
         parameter: "maxSteps",
@@ -4284,6 +4286,10 @@ var DefaultStreamTextResult = class {
         message: "maxSteps must be at least 1"
       });
     }
+    const stitchableStream = createStitchableStream();
+    this.addStream = stitchableStream.addStream;
+    this.closeStream = stitchableStream.close;
+    this.baseStream = transform ? stitchableStream.stream.pipeThrough(transform) : stitchableStream.stream;
     const { maxRetries, retry } = prepareRetries({
       maxRetries: maxRetriesArg
     });
@@ -4449,7 +4455,7 @@ var DefaultStreamTextResult = class {
             hasWhitespaceSuffix = chunk.textDelta.trimEnd() !== chunk.textDelta;
             await (onChunk == null ? void 0 : onChunk({ chunk }));
           }
-          self.stitchableStream.addStream(
+          self.addStream(
             transformedStream.pipeThrough(
               new TransformStream({
                 async transform(chunk, controller) {
@@ -4677,7 +4683,7 @@ var DefaultStreamTextResult = class {
                         ...stepResponse
                       }
                     });
-                    self.stitchableStream.close();
+                    self.closeStream();
                     rootSpan.setAttributes(
                       selectTelemetryAttributes({
                         telemetry,
@@ -4751,7 +4757,7 @@ var DefaultStreamTextResult = class {
         });
       }
     }).catch((error) => {
-      self.stitchableStream.addStream(
+      self.addStream(
         new ReadableStream({
           start(controller) {
             controller.enqueue({ type: "error", error });
@@ -4759,7 +4765,7 @@ var DefaultStreamTextResult = class {
           }
         })
       );
-      self.stitchableStream.close();
+      self.closeStream();
     });
   }
   get warnings() {
@@ -4801,8 +4807,8 @@ var DefaultStreamTextResult = class {
   However, the LLM results are expected to be small enough to not cause issues.
      */
   teeStream() {
-    const [stream1, stream2] = this.stitchableStream.stream.tee();
-    this.stitchableStream.stream = stream2;
+    const [stream1, stream2] = this.baseStream.tee();
+    this.baseStream = stream2;
     return stream1;
   }
   get textStream() {
@@ -4998,6 +5004,40 @@ var DefaultStreamTextResult = class {
   }
 };
 
+// core/generate-text/smooth-stream.ts
+function smoothStream({
+  delayInMs = 10,
+  _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 = "";
+        }
+        controller.enqueue(chunk);
+        return;
+      }
+      if (chunk.type !== "text-delta") {
+        controller.enqueue(chunk);
+        return;
+      }
+      buffer += chunk.textDelta;
+      while (buffer.match(/\s/)) {
+        const whitespaceIndex = buffer.search(/\s/);
+        const word = buffer.slice(0, whitespaceIndex + 1);
+        controller.enqueue({ type: "text-delta", textDelta: word });
+        buffer = buffer.slice(whitespaceIndex + 1);
+        if (delayInMs > 0) {
+          await delay2(delayInMs);
+        }
+      }
+    }
+  });
+}
+
 // core/middleware/wrap-language-model.ts
 var experimental_wrapLanguageModel = ({
   model,
@@ -5563,6 +5603,7 @@ export {
   pipeDataStreamToResponse,
   processDataStream,
   processTextStream,
+  smoothStream,
   streamObject,
   streamText,
   tool