ai 6.0.0-beta.81 → 6.0.0-beta.83

package/dist/index.mjs

@@ -775,7 +775,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.81" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.83" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -2969,7 +2969,7 @@ async function generateText({
         }));
         let resolvedOutput;
         if (lastStep.finishReason === "stop") {
-          resolvedOutput = await (output == null ? void 0 : output.parseOutput(
+          resolvedOutput = await (output == null ? void 0 : output.parseCompleteOutput(
             { text: lastStep.text },
             {
               response: lastStep.response,
@@ -5003,7 +5003,7 @@ function createOutputTransformStream(output) {
       }
       text2 += chunk.text;
       textChunk += chunk.text;
-      const result = await output.parsePartial({ text: text2 });
+      const result = await output.parsePartialOutput({ text: text2 });
       if (result != null) {
         const currentJson = JSON.stringify(result.partial);
         if (currentJson !== lastPublishedJson) {
@@ -9121,6 +9121,7 @@ var output_exports = {};
 __export(output_exports, {
   array: () => array,
   choice: () => choice,
+  json: () => json,
   object: () => object,
   text: () => text
 });
@@ -9134,12 +9135,11 @@ import {
   safeValidateTypes as safeValidateTypes4
 } from "@ai-sdk/provider-utils";
 var text = () => ({
-  type: "text",
   responseFormat: Promise.resolve({ type: "text" }),
-  async parseOutput({ text: text2 }) {
+  async parseCompleteOutput({ text: text2 }) {
     return text2;
   },
-  async parsePartial({ text: text2 }) {
+  async parsePartialOutput({ text: text2 }) {
     return { partial: text2 };
   }
 });
@@ -9148,12 +9148,11 @@ var object = ({
 }) => {
   const schema = asSchema4(inputSchema);
   return {
-    type: "object",
     responseFormat: resolve(schema.jsonSchema).then((jsonSchema2) => ({
       type: "json",
       schema: jsonSchema2
     })),
-    async parseOutput({ text: text2 }, context) {
+    async parseCompleteOutput({ text: text2 }, context) {
       const parseResult = await safeParseJSON4({ text: text2 });
       if (!parseResult.success) {
         throw new NoObjectGeneratedError({
@@ -9181,7 +9180,7 @@ var object = ({
       }
       return validationResult.value;
     },
-    async parsePartial({ text: text2 }) {
+    async parsePartialOutput({ text: text2 }) {
       const result = await parsePartialJson(text2);
       switch (result.state) {
         case "failed-parse":
@@ -9195,10 +9194,6 @@ var object = ({
             partial: result.value
           };
         }
-        default: {
-          const _exhaustiveCheck = result.state;
-          throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
-        }
       }
     }
   };
@@ -9208,7 +9203,6 @@ var array = ({
 }) => {
   const elementSchema = asSchema4(inputElementSchema);
   return {
-    type: "object",
     // JSON schema that describes an array of elements:
     responseFormat: resolve(elementSchema.jsonSchema).then((jsonSchema2) => {
       const { $schema, ...itemSchema } = jsonSchema2;
@@ -9225,7 +9219,7 @@ var array = ({
         }
       };
     }),
-    async parseOutput({ text: text2 }, context) {
+    async parseCompleteOutput({ text: text2 }, context) {
       const parseResult = await safeParseJSON4({ text: text2 });
       if (!parseResult.success) {
         throw new NoObjectGeneratedError({
@@ -9269,7 +9263,7 @@ var array = ({
       }
       return outerValue.elements;
     },
-    async parsePartial({ text: text2 }) {
+    async parsePartialOutput({ text: text2 }) {
       const result = await parsePartialJson(text2);
       switch (result.state) {
         case "failed-parse":
@@ -9295,10 +9289,6 @@ var array = ({
           }
           return { partial: parsedElements };
         }
-        default: {
-          const _exhaustiveCheck = result.state;
-          throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
-        }
       }
     }
   };
@@ -9307,7 +9297,6 @@ var choice = ({
   options: choiceOptions
 }) => {
   return {
-    type: "object",
     // JSON schema that describes an enumeration:
     responseFormat: Promise.resolve({
       type: "json",
@@ -9321,7 +9310,7 @@ var choice = ({
         additionalProperties: false
       }
     }),
-    async parseOutput({ text: text2 }, context) {
+    async parseCompleteOutput({ text: text2 }, context) {
       const parseResult = await safeParseJSON4({ text: text2 });
       if (!parseResult.success) {
         throw new NoObjectGeneratedError({
@@ -9349,7 +9338,7 @@ var choice = ({
       }
       return outerValue.result;
     },
-    async parsePartial({ text: text2 }) {
+    async parsePartialOutput({ text: text2 }) {
       const result = await parsePartialJson(text2);
       switch (result.state) {
         case "failed-parse":
@@ -9371,9 +9360,39 @@ var choice = ({
             return potentialMatches.length === 1 ? { partial: potentialMatches[0] } : void 0;
           }
         }
-        default: {
-          const _exhaustiveCheck = result.state;
-          throw new Error(`Unsupported parse state: ${_exhaustiveCheck}`);
+      }
+    }
+  };
+};
+var json = () => {
+  return {
+    responseFormat: Promise.resolve({
+      type: "json"
+    }),
+    async parseCompleteOutput({ 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
+        });
+      }
+      return parseResult.value;
+    },
+    async parsePartialOutput({ 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": {
+          return result.value === void 0 ? void 0 : { partial: result.value };
         }
       }
     }
@@ -9534,12 +9553,27 @@ function smoothStream({
   };
 }
 
+// src/middleware/default-embedding-settings-middleware.ts
+function defaultEmbeddingSettingsMiddleware({
+  settings
+}) {
+  return {
+    specificationVersion: "v3",
+    transformParams: async ({ params }) => {
+      return mergeObjects(
+        settings,
+        params
+      );
+    }
+  };
+}
+
 // src/middleware/default-settings-middleware.ts
 function defaultSettingsMiddleware({
   settings
 }) {
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     transformParams: async ({ params }) => {
       return mergeObjects(settings, params);
     }
@@ -9573,7 +9607,7 @@ function extractReasoningMiddleware({
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     wrapGenerate: async ({ doGenerate }) => {
       const { content, ...rest } = await doGenerate();
       const transformedContent = [];
@@ -9719,7 +9753,7 @@ function extractReasoningMiddleware({
 // src/middleware/simulate-streaming-middleware.ts
 function simulateStreamingMiddleware() {
   return {
-    middlewareVersion: "v3",
+    specificationVersion: "v3",
     wrapStream: async ({ doGenerate }) => {
       const result = await doGenerate();
       let id = 0;
@@ -9840,6 +9874,54 @@ var doWrap = ({
   };
 };
 
+// src/middleware/wrap-embedding-model.ts
+var wrapEmbeddingModel = ({
+  model,
+  middleware: middlewareArg,
+  modelId,
+  providerId
+}) => {
+  return asArray(middlewareArg).reverse().reduce((wrappedModel, middleware) => {
+    return doWrap2({ model: wrappedModel, middleware, modelId, providerId });
+  }, model);
+};
+var doWrap2 = ({
+  model,
+  middleware: {
+    transformParams,
+    wrapEmbed,
+    overrideProvider,
+    overrideModelId,
+    overrideMaxEmbeddingsPerCall,
+    overrideSupportsParallelCalls
+  },
+  modelId,
+  providerId
+}) => {
+  var _a16, _b, _c, _d;
+  async function doTransform({
+    params
+  }) {
+    return transformParams ? await transformParams({ params, model }) : params;
+  }
+  return {
+    specificationVersion: "v3",
+    provider: (_a16 = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _a16 : model.provider,
+    modelId: (_b = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _b : model.modelId,
+    maxEmbeddingsPerCall: (_c = overrideMaxEmbeddingsPerCall == null ? void 0 : overrideMaxEmbeddingsPerCall({ model })) != null ? _c : model.maxEmbeddingsPerCall,
+    supportsParallelCalls: (_d = overrideSupportsParallelCalls == null ? void 0 : overrideSupportsParallelCalls({ model })) != null ? _d : model.supportsParallelCalls,
+    async doEmbed(params) {
+      const transformedParams = await doTransform({ params });
+      const doEmbed = async () => model.doEmbed(transformedParams);
+      return wrapEmbed ? wrapEmbed({
+        doEmbed,
+        params: transformedParams,
+        model
+      }) : doEmbed();
+    }
+  };
+};
+
 // src/model/as-image-model-v3.ts
 function asImageModelV3(model) {
   if (model.specificationVersion === "v3") {
@@ -11158,6 +11240,7 @@ export {
   createUIMessageStream,
   createUIMessageStreamResponse,
   customProvider,
+  defaultEmbeddingSettingsMiddleware,
   defaultSettingsMiddleware,
   dynamicTool,
   embed,
@@ -11208,6 +11291,7 @@ export {
   uiMessageChunkSchema,
   userModelMessageSchema,
   validateUIMessages,
+  wrapEmbeddingModel,
   wrapLanguageModel,
   wrapProvider,
   zodSchema3 as zodSchema