ai 4.0.13 → 4.0.15

package/dist/index.js

@@ -56,10 +56,11 @@ __export(streams_exports, {
   embedMany: () => embedMany,
   experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_customProvider: () => experimental_customProvider,
+  experimental_generateImage: () => generateImage,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
   formatAssistantStreamPart: () => import_ui_utils14.formatAssistantStreamPart,
   formatDataStreamPart: () => import_ui_utils14.formatDataStreamPart,
-  generateId: () => import_provider_utils12.generateId,
+  generateId: () => import_provider_utils13.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
   jsonSchema: () => import_ui_utils9.jsonSchema,
@@ -68,13 +69,14 @@ __export(streams_exports, {
   pipeDataStreamToResponse: () => pipeDataStreamToResponse,
   processDataStream: () => import_ui_utils14.processDataStream,
   processTextStream: () => import_ui_utils14.processTextStream,
+  smoothStream: () => smoothStream,
   streamObject: () => streamObject,
   streamText: () => streamText,
   tool: () => tool
 });
 module.exports = __toCommonJS(streams_exports);
 var import_ui_utils14 = require("@ai-sdk/ui-utils");
-var import_provider_utils12 = require("@ai-sdk/provider-utils");
+var import_provider_utils13 = require("@ai-sdk/provider-utils");
 
 // core/index.ts
 var import_ui_utils9 = require("@ai-sdk/ui-utils");
@@ -856,8 +858,47 @@ var DefaultEmbedManyResult = class {
   }
 };
 
+// core/generate-image/generate-image.ts
+var import_provider_utils2 = require("@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 (0, import_provider_utils2.convertBase64ToUint8Array)(this.base64);
+      }
+    }));
+  }
+  get image() {
+    return this.images[0];
+  }
+};
+
 // core/generate-object/generate-object.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
 
 // util/download-error.ts
 var import_provider4 = require("@ai-sdk/provider");
@@ -930,7 +971,7 @@ function detectImageMimeType(image) {
 }
 
 // core/prompt/data-content.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // core/prompt/invalid-data-content-error.ts
 var import_provider5 = require("@ai-sdk/provider");
@@ -974,9 +1015,9 @@ function convertDataContentToBase64String(content) {
     return content;
   }
   if (content instanceof ArrayBuffer) {
-    return (0, import_provider_utils2.convertUint8ArrayToBase64)(new Uint8Array(content));
+    return (0, import_provider_utils3.convertUint8ArrayToBase64)(new Uint8Array(content));
   }
-  return (0, import_provider_utils2.convertUint8ArrayToBase64)(content);
+  return (0, import_provider_utils3.convertUint8ArrayToBase64)(content);
 }
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
@@ -984,7 +1025,7 @@ function convertDataContentToUint8Array(content) {
   }
   if (typeof content === "string") {
     try {
-      return (0, import_provider_utils2.convertBase64ToUint8Array)(content);
+      return (0, import_provider_utils3.convertBase64ToUint8Array)(content);
     } catch (error) {
       throw new InvalidDataContentError({
         message: "Invalid data content. Content string is not a base64-encoded media.",
@@ -1323,7 +1364,7 @@ function prepareCallSettings({
 
 // core/prompt/standardize-prompt.ts
 var import_provider8 = require("@ai-sdk/provider");
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
 var import_zod7 = require("zod");
 
 // core/prompt/message.ts
@@ -1706,7 +1747,7 @@ function standardizePrompt({
     const messages = promptType === "ui-messages" ? convertToCoreMessages(prompt.messages, {
       tools
     }) : prompt.messages;
-    const validationResult = (0, import_provider_utils3.safeValidateTypes)({
+    const validationResult = (0, import_provider_utils4.safeValidateTypes)({
       value: messages,
       schema: import_zod7.z.array(coreMessageSchema)
     });
@@ -1778,7 +1819,7 @@ _a7 = symbol7;
 
 // core/generate-object/output-strategy.ts
 var import_provider10 = require("@ai-sdk/provider");
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
 var import_ui_utils2 = require("@ai-sdk/ui-utils");
 
 // core/util/async-iterable-stream.ts
@@ -1828,7 +1869,7 @@ var objectOutputStrategy = (schema) => ({
     };
   },
   validateFinalResult(value) {
-    return (0, import_provider_utils4.safeValidateTypes)({ value, schema });
+    return (0, import_provider_utils5.safeValidateTypes)({ value, schema });
   },
   createElementStream() {
     throw new import_provider10.UnsupportedFunctionalityError({
@@ -1867,7 +1908,7 @@ var arrayOutputStrategy = (schema) => {
       const resultArray = [];
       for (let i = 0; i < inputArray.length; i++) {
         const element = inputArray[i];
-        const result = (0, import_provider_utils4.safeValidateTypes)({ value: element, schema });
+        const result = (0, import_provider_utils5.safeValidateTypes)({ value: element, schema });
         if (i === inputArray.length - 1 && !isFinalDelta) {
           continue;
         }
@@ -1908,7 +1949,7 @@ var arrayOutputStrategy = (schema) => {
       }
       const inputArray = value.elements;
       for (const element of inputArray) {
-        const result = (0, import_provider_utils4.safeValidateTypes)({ value: element, schema });
+        const result = (0, import_provider_utils5.safeValidateTypes)({ value: element, schema });
         if (!result.success) {
           return result;
         }
@@ -2137,7 +2178,7 @@ function validateObjectGenerationInput({
 }
 
 // core/generate-object/generate-object.ts
-var originalGenerateId = (0, import_provider_utils5.createIdGenerator)({ prefix: "aiobj", size: 24 });
+var originalGenerateId = (0, import_provider_utils6.createIdGenerator)({ prefix: "aiobj", size: 24 });
 async function generateObject({
   model,
   enum: enumValues,
@@ -2442,7 +2483,7 @@ async function generateObject({
           throw new Error(`Unsupported mode: ${_exhaustiveCheck}`);
         }
       }
-      const parseResult = (0, import_provider_utils5.safeParseJSON)({ text: result });
+      const parseResult = (0, import_provider_utils6.safeParseJSON)({ text: result });
       if (!parseResult.success) {
         throw parseResult.error;
       }
@@ -2504,7 +2545,7 @@ var DefaultGenerateObjectResult = class {
 };
 
 // core/generate-object/stream-object.ts
-var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_provider_utils7 = require("@ai-sdk/provider-utils");
 var import_ui_utils3 = require("@ai-sdk/ui-utils");
 
 // util/delayed-promise.ts
@@ -2634,7 +2675,7 @@ function now() {
 }
 
 // core/generate-object/stream-object.ts
-var originalGenerateId2 = (0, import_provider_utils6.createIdGenerator)({ prefix: "aiobj", size: 24 });
+var originalGenerateId2 = (0, import_provider_utils7.createIdGenerator)({ prefix: "aiobj", size: 24 });
 function streamObject({
   model,
   schema: inputSchema,
@@ -3173,7 +3214,7 @@ var DefaultStreamObjectResult = class {
 };
 
 // core/generate-text/generate-text.ts
-var import_provider_utils8 = require("@ai-sdk/provider-utils");
+var import_provider_utils9 = require("@ai-sdk/provider-utils");
 
 // errors/index.ts
 var import_provider15 = require("@ai-sdk/provider");
@@ -3339,7 +3380,7 @@ function removeTextAfterLastWhitespace(text2) {
 }
 
 // core/generate-text/parse-tool-call.ts
-var import_provider_utils7 = require("@ai-sdk/provider-utils");
+var import_provider_utils8 = require("@ai-sdk/provider-utils");
 var import_ui_utils5 = require("@ai-sdk/ui-utils");
 async function parseToolCall({
   toolCall,
@@ -3392,7 +3433,7 @@ async function doParseToolCall({
     });
   }
   const schema = (0, import_ui_utils5.asSchema)(tool2.parameters);
-  const parseResult = toolCall.args.trim() === "" ? (0, import_provider_utils7.safeValidateTypes)({ value: {}, schema }) : (0, import_provider_utils7.safeParseJSON)({ text: toolCall.args, schema });
+  const parseResult = toolCall.args.trim() === "" ? (0, import_provider_utils8.safeValidateTypes)({ value: {}, schema }) : (0, import_provider_utils8.safeParseJSON)({ text: toolCall.args, schema });
   if (parseResult.success === false) {
     throw new InvalidToolArgumentsError({
       toolName,
@@ -3446,7 +3487,7 @@ function toResponseMessages({
 }
 
 // core/generate-text/generate-text.ts
-var originalGenerateId3 = (0, import_provider_utils8.createIdGenerator)({ prefix: "aitxt", size: 24 });
+var originalGenerateId3 = (0, import_provider_utils9.createIdGenerator)({ prefix: "aitxt", size: 24 });
 async function generateText({
   model,
   tools,
@@ -3851,7 +3892,7 @@ __export(output_exports, {
   object: () => object,
   text: () => text
 });
-var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
 var import_ui_utils6 = require("@ai-sdk/ui-utils");
 var text = () => ({
   type: "text",
@@ -3880,13 +3921,13 @@ var object = ({
       });
     },
     parseOutput({ text: text2 }) {
-      return (0, import_provider_utils9.parseJSON)({ text: text2, schema });
+      return (0, import_provider_utils10.parseJSON)({ text: text2, schema });
     }
   };
 };
 
 // core/generate-text/stream-text.ts
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
 var import_ui_utils8 = require("@ai-sdk/ui-utils");
 
 // core/util/merge-streams.ts
@@ -4185,7 +4226,7 @@ function runToolsTransformation({
 }
 
 // core/generate-text/stream-text.ts
-var originalGenerateId4 = (0, import_provider_utils10.createIdGenerator)({ prefix: "aitxt", size: 24 });
+var originalGenerateId4 = (0, import_provider_utils11.createIdGenerator)({ prefix: "aitxt", size: 24 });
 function streamText({
   model,
   tools,
@@ -4203,6 +4244,7 @@ function streamText({
   experimental_toolCallStreaming: toolCallStreaming = false,
   experimental_activeTools: activeTools,
   experimental_repairToolCall: repairToolCall,
+  experimental_transform: transform,
   onChunk,
   onFinish,
   onStepFinish,
@@ -4226,6 +4268,7 @@ function streamText({
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4253,6 +4296,7 @@ var DefaultStreamTextResult = class {
     tools,
     toolChoice,
     toolCallStreaming,
+    transform,
     activeTools,
     repairToolCall,
     maxSteps,
@@ -4275,7 +4319,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",
@@ -4283,6 +4326,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
     });
@@ -4448,7 +4495,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) {
@@ -4676,7 +4723,7 @@ var DefaultStreamTextResult = class {
                         ...stepResponse
                       }
                     });
-                    self.stitchableStream.close();
+                    self.closeStream();
                     rootSpan.setAttributes(
                       selectTelemetryAttributes({
                         telemetry,
@@ -4750,7 +4797,7 @@ var DefaultStreamTextResult = class {
         });
       }
     }).catch((error) => {
-      self.stitchableStream.addStream(
+      self.addStream(
         new ReadableStream({
           start(controller) {
             controller.enqueue({ type: "error", error });
@@ -4758,7 +4805,7 @@ var DefaultStreamTextResult = class {
           }
         })
       );
-      self.stitchableStream.close();
+      self.closeStream();
     });
   }
   get warnings() {
@@ -4800,8 +4847,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() {
@@ -4997,6 +5044,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,
@@ -5388,11 +5469,11 @@ __export(llamaindex_adapter_exports, {
   toDataStream: () => toDataStream2,
   toDataStreamResponse: () => toDataStreamResponse2
 });
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_provider_utils12 = require("@ai-sdk/provider-utils");
 var import_ui_utils12 = require("@ai-sdk/ui-utils");
 function toDataStreamInternal2(stream, callbacks) {
   const trimStart = trimStartOfStream();
-  return (0, import_provider_utils11.convertAsyncIteratorToReadableStream)(stream[Symbol.asyncIterator]()).pipeThrough(
+  return (0, import_provider_utils12.convertAsyncIteratorToReadableStream)(stream[Symbol.asyncIterator]()).pipeThrough(
     new TransformStream({
       async transform(message, controller) {
         controller.enqueue(trimStart(message.delta));
@@ -5548,6 +5629,7 @@ var StreamData = class {
   embedMany,
   experimental_createProviderRegistry,
   experimental_customProvider,
+  experimental_generateImage,
   experimental_wrapLanguageModel,
   formatAssistantStreamPart,
   formatDataStreamPart,
@@ -5560,6 +5642,7 @@ var StreamData = class {
   pipeDataStreamToResponse,
   processDataStream,
   processTextStream,
+  smoothStream,
   streamObject,
   streamText,
   tool