ai 4.1.42 → 4.1.44

package/dist/index.mjs

@@ -918,6 +918,10 @@ var DefaultGeneratedImage = class {
 };
 
 // core/generate-object/generate-object.ts
+import {
+  JSONParseError,
+  TypeValidationError as TypeValidationError2
+} from "@ai-sdk/provider";
 import { createIdGenerator, safeParseJSON } from "@ai-sdk/provider-utils";
 
 // errors/no-object-generated-error.ts
@@ -2370,6 +2374,7 @@ async function generateObject({
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
+  experimental_repairText: repairText,
   experimental_telemetry: telemetry,
   experimental_providerMetadata,
   providerOptions = experimental_providerMetadata,
@@ -2670,32 +2675,52 @@ async function generateObject({
           throw new Error(`Unsupported mode: ${_exhaustiveCheck}`);
         }
       }
-      const parseResult = safeParseJSON({ text: result });
-      if (!parseResult.success) {
-        throw new NoObjectGeneratedError({
-          message: "No object generated: could not parse the response.",
-          cause: parseResult.error,
-          text: result,
-          response,
-          usage: calculateLanguageModelUsage(usage)
-        });
+      function processResult(result2) {
+        const parseResult = safeParseJSON({ text: result2 });
+        if (!parseResult.success) {
+          throw new NoObjectGeneratedError({
+            message: "No object generated: could not parse the response.",
+            cause: parseResult.error,
+            text: result2,
+            response,
+            usage: calculateLanguageModelUsage(usage)
+          });
+        }
+        const validationResult = outputStrategy.validateFinalResult(
+          parseResult.value,
+          {
+            text: result2,
+            response,
+            usage: calculateLanguageModelUsage(usage)
+          }
+        );
+        if (!validationResult.success) {
+          throw new NoObjectGeneratedError({
+            message: "No object generated: response did not match schema.",
+            cause: validationResult.error,
+            text: result2,
+            response,
+            usage: calculateLanguageModelUsage(usage)
+          });
+        }
+        return validationResult.value;
       }
-      const validationResult = outputStrategy.validateFinalResult(
-        parseResult.value,
-        {
-          text: result,
-          response,
-          usage: calculateLanguageModelUsage(usage)
+      let object2;
+      try {
+        object2 = processResult(result);
+      } catch (error) {
+        if (repairText != null && NoObjectGeneratedError.isInstance(error) && (JSONParseError.isInstance(error.cause) || TypeValidationError2.isInstance(error.cause))) {
+          const repairedText = await repairText({
+            text: result,
+            error: error.cause
+          });
+          if (repairedText === null) {
+            throw error;
+          }
+          object2 = processResult(repairedText);
+        } else {
+          throw error;
         }
-      );
-      if (!validationResult.success) {
-        throw new NoObjectGeneratedError({
-          message: "No object generated: response did not match schema.",
-          cause: validationResult.error,
-          text: result,
-          response,
-          usage: calculateLanguageModelUsage(usage)
-        });
       }
       span.setAttributes(
         selectTelemetryAttributes({
@@ -2703,7 +2728,7 @@ async function generateObject({
           attributes: {
             "ai.response.finishReason": finishReason,
             "ai.response.object": {
-              output: () => JSON.stringify(validationResult.value)
+              output: () => JSON.stringify(object2)
             },
             "ai.usage.promptTokens": usage.promptTokens,
             "ai.usage.completionTokens": usage.completionTokens
@@ -2711,7 +2736,7 @@ async function generateObject({
         })
       );
       return new DefaultGenerateObjectResult({
-        object: validationResult.value,
+        object: object2,
         finishReason,
         usage: calculateLanguageModelUsage(usage),
         warnings,
@@ -4216,11 +4241,11 @@ import {
   EmptyResponseBodyError,
   InvalidPromptError as InvalidPromptError2,
   InvalidResponseDataError,
-  JSONParseError,
+  JSONParseError as JSONParseError2,
   LoadAPIKeyError,
   NoContentGeneratedError,
   NoSuchModelError,
-  TypeValidationError as TypeValidationError2,
+  TypeValidationError as TypeValidationError3,
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 
@@ -5768,17 +5793,19 @@ function getPotentialStartIndex(text2, searchedText) {
 // core/middleware/extract-reasoning-middleware.ts
 function extractReasoningMiddleware({
   tagName,
-  separator = "\n"
+  separator = "\n",
+  startWithReasoning = false
 }) {
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
     middlewareVersion: "v1",
     wrapGenerate: async ({ doGenerate }) => {
-      const { text: text2, ...rest } = await doGenerate();
-      if (text2 == null) {
-        return { text: text2, ...rest };
+      const { text: rawText, ...rest } = await doGenerate();
+      if (rawText == null) {
+        return { text: rawText, ...rest };
       }
+      const text2 = startWithReasoning ? openingTag + rawText : rawText;
       const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
       const matches = Array.from(text2.matchAll(regexp));
       if (!matches.length) {
@@ -5801,7 +5828,7 @@ function extractReasoningMiddleware({
       let isFirstReasoning = true;
       let isFirstText = true;
       let afterSwitch = false;
-      let isReasoning = false;
+      let isReasoning = startWithReasoning;
       let buffer = "";
       return {
         stream: stream.pipeThrough(
@@ -6576,7 +6603,7 @@ export {
   InvalidPromptError2 as InvalidPromptError,
   InvalidResponseDataError,
   InvalidToolArgumentsError,
-  JSONParseError,
+  JSONParseError2 as JSONParseError,
   langchain_adapter_exports as LangChainAdapter,
   llamaindex_adapter_exports as LlamaIndexAdapter,
   LoadAPIKeyError,
@@ -6593,7 +6620,7 @@ export {
   StreamData,
   ToolCallRepairError,
   ToolExecutionError,
-  TypeValidationError2 as TypeValidationError,
+  TypeValidationError3 as TypeValidationError,
   UnsupportedFunctionalityError2 as UnsupportedFunctionalityError,
   appendClientMessage,
   appendResponseMessages,