ai 4.0.13 → 4.0.15

package/dist/index.mjs

@@ -795,6 +795,45 @@ var DefaultEmbedManyResult = class {
   }
 };
 
+// core/generate-image/generate-image.ts
+import { convertBase64ToUint8Array } from "@ai-sdk/provider-utils";
+async function generateImage({
+  model,
+  prompt,
+  n,
+  size,
+  providerOptions,
+  maxRetries: maxRetriesArg,
+  abortSignal,
+  headers
+}) {
+  const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { images } = await retry(
+    () => model.doGenerate({
+      prompt,
+      n: n != null ? n : 1,
+      abortSignal,
+      headers,
+      size,
+      providerOptions: providerOptions != null ? providerOptions : {}
+    })
+  );
+  return new DefaultGenerateImageResult({ base64Images: images });
+}
+var DefaultGenerateImageResult = class {
+  constructor(options) {
+    this.images = options.base64Images.map((base64) => ({
+      base64,
+      get uint8Array() {
+        return convertBase64ToUint8Array(this.base64);
+      }
+    }));
+  }
+  get image() {
+    return this.images[0];
+  }
+};
+
 // core/generate-object/generate-object.ts
 import { createIdGenerator, safeParseJSON } from "@ai-sdk/provider-utils";
 
@@ -870,7 +909,7 @@ function detectImageMimeType(image) {
 
 // core/prompt/data-content.ts
 import {
-  convertBase64ToUint8Array,
+  convertBase64ToUint8Array as convertBase64ToUint8Array2,
   convertUint8ArrayToBase64
 } from "@ai-sdk/provider-utils";
 
@@ -926,7 +965,7 @@ function convertDataContentToUint8Array(content) {
   }
   if (typeof content === "string") {
     try {
-      return convertBase64ToUint8Array(content);
+      return convertBase64ToUint8Array2(content);
     } catch (error) {
       throw new InvalidDataContentError({
         message: "Invalid data content. Content string is not a base64-encoded media.",
@@ -4165,6 +4204,7 @@ function streamText({
   experimental_toolCallStreaming: toolCallStreaming = false,
   experimental_activeTools: activeTools,
   experimental_repairToolCall: repairToolCall,
+  experimental_transform: transform,
   onChunk,
   onFinish,
   onStepFinish,
@@ -4188,6 +4228,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4215,6 +4256,7 @@ var DefaultStreamTextResult = class {
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4237,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",
@@ -4245,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
     });
@@ -4410,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) {
@@ -4638,7 +4683,7 @@ var DefaultStreamTextResult = class {
                         ...stepResponse
                       }
                     });
-                    self.stitchableStream.close();
+                    self.closeStream();
                     rootSpan.setAttributes(
                       selectTelemetryAttributes({
                         telemetry,
@@ -4712,7 +4757,7 @@ var DefaultStreamTextResult = class {
         });
       }
     }).catch((error) => {
-      self.stitchableStream.addStream(
+      self.addStream(
         new ReadableStream({
           start(controller) {
             controller.enqueue({ type: "error", error });
@@ -4720,7 +4765,7 @@ var DefaultStreamTextResult = class {
           }
         })
       );
-      self.stitchableStream.close();
+      self.closeStream();
     });
   }
   get warnings() {
@@ -4762,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() {
@@ -4959,6 +5004,40 @@ var DefaultStreamTextResult = class {
   }
 };
 
+// core/generate-text/smooth-stream.ts
+function smoothStream({
+  delayInMs = 40,
+  _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,
@@ -5511,6 +5590,7 @@ export {
   embedMany,
   experimental_createProviderRegistry,
   experimental_customProvider,
+  generateImage as experimental_generateImage,
   experimental_wrapLanguageModel,
   formatAssistantStreamPart2 as formatAssistantStreamPart,
   formatDataStreamPart6 as formatDataStreamPart,
@@ -5523,6 +5603,7 @@ export {
   pipeDataStreamToResponse,
   processDataStream,
   processTextStream,
+  smoothStream,
   streamObject,
   streamText,
   tool