ai 6.0.0-beta.68 → 6.0.0-beta.70

package/dist/index.js

@@ -876,7 +876,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.68" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.70" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -7291,13 +7291,13 @@ var DefaultEmbedResult = class {
 var import_provider_utils21 = require("@ai-sdk/provider-utils");
 
 // src/util/split-array.ts
-function splitArray(array, chunkSize) {
+function splitArray(array2, chunkSize) {
   if (chunkSize <= 0) {
     throw new Error("chunkSize must be greater than 0");
   }
   const result = [];
-  for (let i = 0; i < array.length; i += chunkSize) {
-    result.push(array.slice(i, i + chunkSize));
+  for (let i = 0; i < array2.length; i += chunkSize) {
+    result.push(array2.slice(i, i + chunkSize));
   }
   return result;
 }
@@ -7808,9 +7808,9 @@ var arrayOutputStrategy = (schema) => {
             transform(chunk, controller) {
               switch (chunk.type) {
                 case "object": {
-                  const array = chunk.object;
-                  for (; publishedElements < array.length; publishedElements++) {
-                    controller.enqueue(array[publishedElements]);
+                  const array2 = chunk.object;
+                  for (; publishedElements < array2.length; publishedElements++) {
+                    controller.enqueue(array2[publishedElements]);
                   }
                   break;
                 }
@@ -9119,18 +9119,21 @@ var DefaultSpeechResult = class {
 // src/generate-text/output.ts
 var output_exports = {};
 __export(output_exports, {
+  array: () => array,
+  choice: () => choice,
   object: () => object,
   text: () => text
 });
+var import_provider28 = require("@ai-sdk/provider");
 var import_provider_utils29 = require("@ai-sdk/provider-utils");
 var text = () => ({
   type: "text",
   responseFormat: Promise.resolve({ type: "text" }),
-  async parsePartial({ text: text2 }) {
-    return { partial: text2 };
-  },
   async parseOutput({ text: text2 }) {
     return text2;
+  },
+  async parsePartial({ text: text2 }) {
+    return { partial: text2 };
   }
 });
 var object = ({
@@ -9143,24 +9146,174 @@ var object = ({
       type: "json",
       schema: jsonSchema3
     })),
+    async parseOutput({ text: text2 }, context) {
+      const parseResult = await (0, import_provider_utils29.safeParseJSON)({ text: text2 });
+      if (!parseResult.success) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: could not parse the response.",
+          cause: parseResult.error,
+          text: text2,
+          response: context.response,
+          usage: context.usage,
+          finishReason: context.finishReason
+        });
+      }
+      const validationResult = await (0, import_provider_utils29.safeValidateTypes)({
+        value: parseResult.value,
+        schema
+      });
+      if (!validationResult.success) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: response did not match schema.",
+          cause: validationResult.error,
+          text: text2,
+          response: context.response,
+          usage: context.usage,
+          finishReason: context.finishReason
+        });
+      }
+      return validationResult.value;
+    },
     async parsePartial({ text: text2 }) {
       const result = await parsePartialJson(text2);
       switch (result.state) {
         case "failed-parse":
-        case "undefined-input":
+        case "undefined-input": {
           return void 0;
+        }
         case "repaired-parse":
-        case "successful-parse":
+        case "successful-parse": {
           return {
             // Note: currently no validation of partial results:
             partial: result.value
           };
+        }
         default: {
           const _exhaustiveCheck = result.state;
           throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
         }
       }
+    }
+  };
+};
+var array = ({
+  element: inputElementSchema
+}) => {
+  const elementSchema = (0, import_provider_utils29.asSchema)(inputElementSchema);
+  return {
+    type: "object",
+    // JSON schema that describes an array of elements:
+    responseFormat: (0, import_provider_utils29.resolve)(elementSchema.jsonSchema).then((jsonSchema3) => {
+      const { $schema, ...itemSchema } = jsonSchema3;
+      return {
+        type: "json",
+        schema: {
+          $schema: "http://json-schema.org/draft-07/schema#",
+          type: "object",
+          properties: {
+            elements: { type: "array", items: itemSchema }
+          },
+          required: ["elements"],
+          additionalProperties: false
+        }
+      };
+    }),
+    async parseOutput({ text: text2 }, context) {
+      const parseResult = await (0, import_provider_utils29.safeParseJSON)({ text: text2 });
+      if (!parseResult.success) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: could not parse the response.",
+          cause: parseResult.error,
+          text: text2,
+          response: context.response,
+          usage: context.usage,
+          finishReason: context.finishReason
+        });
+      }
+      const outerValue = parseResult.value;
+      if (outerValue == null || typeof outerValue !== "object" || !("elements" in outerValue) || !Array.isArray(outerValue.elements)) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: response did not match schema.",
+          cause: new import_provider28.TypeValidationError({
+            value: outerValue,
+            cause: "response must be an object with an elements array"
+          }),
+          text: text2,
+          response: context.response,
+          usage: context.usage,
+          finishReason: context.finishReason
+        });
+      }
+      for (const element of outerValue.elements) {
+        const validationResult = await (0, import_provider_utils29.safeValidateTypes)({
+          value: element,
+          schema: elementSchema
+        });
+        if (!validationResult.success) {
+          throw new NoObjectGeneratedError({
+            message: "No object generated: response did not match schema.",
+            cause: validationResult.error,
+            text: text2,
+            response: context.response,
+            usage: context.usage,
+            finishReason: context.finishReason
+          });
+        }
+      }
+      return outerValue.elements;
     },
+    async parsePartial({ text: text2 }) {
+      const result = await parsePartialJson(text2);
+      switch (result.state) {
+        case "failed-parse":
+        case "undefined-input": {
+          return void 0;
+        }
+        case "repaired-parse":
+        case "successful-parse": {
+          const outerValue = result.value;
+          if (outerValue == null || typeof outerValue !== "object" || !("elements" in outerValue) || !Array.isArray(outerValue.elements)) {
+            return void 0;
+          }
+          const rawElements = result.state === "repaired-parse" && outerValue.elements.length > 0 ? outerValue.elements.slice(0, -1) : outerValue.elements;
+          const parsedElements = [];
+          for (const rawElement of rawElements) {
+            const validationResult = await (0, import_provider_utils29.safeValidateTypes)({
+              value: rawElement,
+              schema: elementSchema
+            });
+            if (validationResult.success) {
+              parsedElements.push(validationResult.value);
+            }
+          }
+          return { partial: parsedElements };
+        }
+        default: {
+          const _exhaustiveCheck = result.state;
+          throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
+        }
+      }
+    }
+  };
+};
+var choice = ({
+  options: choiceOptions
+}) => {
+  return {
+    type: "object",
+    // JSON schema that describes an enumeration:
+    responseFormat: Promise.resolve({
+      type: "json",
+      schema: {
+        $schema: "http://json-schema.org/draft-07/schema#",
+        type: "object",
+        properties: {
+          result: { type: "string", enum: choiceOptions }
+        },
+        required: ["result"],
+        additionalProperties: false
+      }
+    }),
     async parseOutput({ text: text2 }, context) {
       const parseResult = await (0, import_provider_utils29.safeParseJSON)({ text: text2 });
       if (!parseResult.success) {
@@ -9173,21 +9326,49 @@ var object = ({
           finishReason: context.finishReason
         });
       }
-      const validationResult = await (0, import_provider_utils29.safeValidateTypes)({
-        value: parseResult.value,
-        schema
-      });
-      if (!validationResult.success) {
+      const outerValue = parseResult.value;
+      if (outerValue == null || typeof outerValue !== "object" || !("result" in outerValue) || typeof outerValue.result !== "string" || !choiceOptions.includes(outerValue.result)) {
         throw new NoObjectGeneratedError({
           message: "No object generated: response did not match schema.",
-          cause: validationResult.error,
+          cause: new import_provider28.TypeValidationError({
+            value: outerValue,
+            cause: "response must be an object that contains a choice value."
+          }),
           text: text2,
           response: context.response,
           usage: context.usage,
           finishReason: context.finishReason
         });
       }
-      return validationResult.value;
+      return outerValue.result;
+    },
+    async parsePartial({ text: text2 }) {
+      const result = await parsePartialJson(text2);
+      switch (result.state) {
+        case "failed-parse":
+        case "undefined-input": {
+          return void 0;
+        }
+        case "repaired-parse":
+        case "successful-parse": {
+          const outerValue = result.value;
+          if (outerValue == null || typeof outerValue !== "object" || !("result" in outerValue) || typeof outerValue.result !== "string") {
+            return void 0;
+          }
+          const potentialMatches = choiceOptions.filter(
+            (choiceOption) => choiceOption.startsWith(outerValue.result)
+          );
+          if (result.state === "successful-parse") {
+            return potentialMatches.includes(outerValue.result) ? { partial: outerValue.result } : void 0;
+          } else {
+            return potentialMatches.length === 1 ? { partial: potentialMatches[0] } : void 0;
+          }
+        }
+        default: {
+          const _exhaustiveCheck = result.state;
+          throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
+        }
+      }
     }
   };
 };
@@ -9273,7 +9454,7 @@ function pruneMessages({
 
 // src/generate-text/smooth-stream.ts
 var import_provider_utils30 = require("@ai-sdk/provider-utils");
-var import_provider28 = require("@ai-sdk/provider");
+var import_provider29 = require("@ai-sdk/provider");
 var CHUNKING_REGEXPS = {
   word: /\S+\s+/m,
   line: /\n+/m
@@ -9303,7 +9484,7 @@ function smoothStream({
   } else {
     const chunkingRegex = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking;
     if (chunkingRegex == null) {
-      throw new import_provider28.InvalidArgumentError({
+      throw new import_provider29.InvalidArgumentError({
         argument: "chunking",
         message: `Chunking must be "word" or "line" or a RegExp. Received: ${chunking}`
       });
@@ -9675,7 +9856,7 @@ function wrapProvider({
 }
 
 // src/registry/custom-provider.ts
-var import_provider29 = require("@ai-sdk/provider");
+var import_provider30 = require("@ai-sdk/provider");
 function customProvider({
   languageModels,
   textEmbeddingModels,
@@ -9692,7 +9873,7 @@ function customProvider({
       if (fallbackProvider) {
         return fallbackProvider.languageModel(modelId);
       }
-      throw new import_provider29.NoSuchModelError({ modelId, modelType: "languageModel" });
+      throw new import_provider30.NoSuchModelError({ modelId, modelType: "languageModel" });
     },
     textEmbeddingModel(modelId) {
       if (textEmbeddingModels != null && modelId in textEmbeddingModels) {
@@ -9701,7 +9882,7 @@ function customProvider({
       if (fallbackProvider) {
         return fallbackProvider.textEmbeddingModel(modelId);
       }
-      throw new import_provider29.NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+      throw new import_provider30.NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
     },
     imageModel(modelId) {
       if (imageModels != null && modelId in imageModels) {
@@ -9710,7 +9891,7 @@ function customProvider({
       if (fallbackProvider == null ? void 0 : fallbackProvider.imageModel) {
         return fallbackProvider.imageModel(modelId);
       }
-      throw new import_provider29.NoSuchModelError({ modelId, modelType: "imageModel" });
+      throw new import_provider30.NoSuchModelError({ modelId, modelType: "imageModel" });
     },
     transcriptionModel(modelId) {
       if (transcriptionModels != null && modelId in transcriptionModels) {
@@ -9719,7 +9900,7 @@ function customProvider({
       if (fallbackProvider == null ? void 0 : fallbackProvider.transcriptionModel) {
         return fallbackProvider.transcriptionModel(modelId);
       }
-      throw new import_provider29.NoSuchModelError({ modelId, modelType: "transcriptionModel" });
+      throw new import_provider30.NoSuchModelError({ modelId, modelType: "transcriptionModel" });
     },
     speechModel(modelId) {
       if (speechModels != null && modelId in speechModels) {
@@ -9728,19 +9909,19 @@ function customProvider({
       if (fallbackProvider == null ? void 0 : fallbackProvider.speechModel) {
         return fallbackProvider.speechModel(modelId);
       }
-      throw new import_provider29.NoSuchModelError({ modelId, modelType: "speechModel" });
+      throw new import_provider30.NoSuchModelError({ modelId, modelType: "speechModel" });
     }
   };
 }
 var experimental_customProvider = customProvider;
 
 // src/registry/no-such-provider-error.ts
-var import_provider30 = require("@ai-sdk/provider");
+var import_provider31 = require("@ai-sdk/provider");
 var name16 = "AI_NoSuchProviderError";
 var marker16 = `vercel.ai.error.${name16}`;
 var symbol16 = Symbol.for(marker16);
 var _a16;
-var NoSuchProviderError = class extends import_provider30.NoSuchModelError {
+var NoSuchProviderError = class extends import_provider31.NoSuchModelError {
   constructor({
     modelId,
     modelType,
@@ -9754,13 +9935,13 @@ var NoSuchProviderError = class extends import_provider30.NoSuchModelError {
     this.availableProviders = availableProviders;
   }
   static isInstance(error) {
-    return import_provider30.AISDKError.hasMarker(error, marker16);
+    return import_provider31.AISDKError.hasMarker(error, marker16);
   }
 };
 _a16 = symbol16;
 
 // src/registry/provider-registry.ts
-var import_provider31 = require("@ai-sdk/provider");
+var import_provider32 = require("@ai-sdk/provider");
 function createProviderRegistry(providers, {
   separator = ":",
   languageModelMiddleware
@@ -9805,7 +9986,7 @@ var DefaultProviderRegistry = class {
   splitId(id, modelType) {
     const index = id.indexOf(this.separator);
     if (index === -1) {
-      throw new import_provider31.NoSuchModelError({
+      throw new import_provider32.NoSuchModelError({
         modelId: id,
         modelType,
         message: `Invalid ${modelType} id for registry: ${id} (must be in the format "providerId${this.separator}modelId")`
@@ -9821,7 +10002,7 @@ var DefaultProviderRegistry = class {
       modelId
     );
     if (model == null) {
-      throw new import_provider31.NoSuchModelError({ modelId: id, modelType: "languageModel" });
+      throw new import_provider32.NoSuchModelError({ modelId: id, modelType: "languageModel" });
     }
     if (this.languageModelMiddleware != null) {
       model = wrapLanguageModel({
@@ -9837,7 +10018,7 @@ var DefaultProviderRegistry = class {
     const provider = this.getProvider(providerId, "textEmbeddingModel");
     const model = (_a17 = provider.textEmbeddingModel) == null ? void 0 : _a17.call(provider, modelId);
     if (model == null) {
-      throw new import_provider31.NoSuchModelError({
+      throw new import_provider32.NoSuchModelError({
         modelId: id,
         modelType: "textEmbeddingModel"
       });
@@ -9850,7 +10031,7 @@ var DefaultProviderRegistry = class {
     const provider = this.getProvider(providerId, "imageModel");
     const model = (_a17 = provider.imageModel) == null ? void 0 : _a17.call(provider, modelId);
     if (model == null) {
-      throw new import_provider31.NoSuchModelError({ modelId: id, modelType: "imageModel" });
+      throw new import_provider32.NoSuchModelError({ modelId: id, modelType: "imageModel" });
     }
     return model;
   }
@@ -9860,7 +10041,7 @@ var DefaultProviderRegistry = class {
     const provider = this.getProvider(providerId, "transcriptionModel");
     const model = (_a17 = provider.transcriptionModel) == null ? void 0 : _a17.call(provider, modelId);
     if (model == null) {
-      throw new import_provider31.NoSuchModelError({
+      throw new import_provider32.NoSuchModelError({
         modelId: id,
         modelType: "transcriptionModel"
       });
@@ -9873,7 +10054,7 @@ var DefaultProviderRegistry = class {
     const provider = this.getProvider(providerId, "speechModel");
     const model = (_a17 = provider.speechModel) == null ? void 0 : _a17.call(provider, modelId);
     if (model == null) {
-      throw new import_provider31.NoSuchModelError({ modelId: id, modelType: "speechModel" });
+      throw new import_provider32.NoSuchModelError({ modelId: id, modelType: "speechModel" });
     }
     return model;
   }
@@ -10460,8 +10641,8 @@ var DefaultMCPClient = class {
 var import_provider_utils33 = require("@ai-sdk/provider-utils");
 
 // src/error/no-transcript-generated-error.ts
-var import_provider32 = require("@ai-sdk/provider");
-var NoTranscriptGeneratedError = class extends import_provider32.AISDKError {
+var import_provider33 = require("@ai-sdk/provider");
+var NoTranscriptGeneratedError = class extends import_provider33.AISDKError {
   constructor(options) {
     super({
       name: "AI_NoTranscriptGeneratedError",