ai 6.0.0-beta.69 → 6.0.0-beta.70

package/dist/index.mjs

@@ -775,7 +775,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.69" : "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 }) => {
@@ -9061,6 +9061,7 @@ var DefaultSpeechResult = class {
 var output_exports = {};
 __export(output_exports, {
   array: () => array,
+  choice: () => choice,
   object: () => object,
   text: () => text
 });
@@ -9149,7 +9150,7 @@ var array = ({
   const elementSchema = asSchema4(inputElementSchema);
   return {
     type: "object",
-    // returns a JSON schema that describes an array of elements:
+    // JSON schema that describes an array of elements:
     responseFormat: resolve(elementSchema.jsonSchema).then((jsonSchema3) => {
       const { $schema, ...itemSchema } = jsonSchema3;
       return {
@@ -9243,6 +9244,82 @@ var array = ({
     }
   };
 };
+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 safeParseJSON4({ 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" || !("result" in outerValue) || typeof outerValue.result !== "string" || !choiceOptions.includes(outerValue.result)) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: response did not match schema.",
+          cause: new TypeValidationError5({
+            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 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}`);
+        }
+      }
+    }
+  };
+};
 
 // src/generate-text/prune-messages.ts
 function pruneMessages({