ai 4.2.11 → 4.3.1

package/dist/index.mjs

@@ -1037,13 +1037,15 @@ var NoObjectGeneratedError = class extends AISDKError4 {
     cause,
     text: text2,
     response,
-    usage
+    usage,
+    finishReason
   }) {
     super({ name: name4, message, cause });
     this[_a4] = true;
     this.text = text2;
     this.response = response;
     this.usage = usage;
+    this.finishReason = finishReason;
   }
   static isInstance(error) {
     return AISDKError4.hasMarker(error, marker4);
@@ -2213,7 +2215,8 @@ var noSchemaOutputStrategy = {
         message: "No object generated: response did not match schema.",
         text: context.text,
         response: context.response,
-        usage: context.usage
+        usage: context.usage,
+        finishReason: context.finishReason
       })
     } : { success: true, value };
   },
@@ -2710,7 +2713,8 @@ async function generateObject({
                   throw new NoObjectGeneratedError({
                     message: "No object generated: the model did not return a response.",
                     response: responseData,
-                    usage: calculateLanguageModelUsage(result2.usage)
+                    usage: calculateLanguageModelUsage(result2.usage),
+                    finishReason: result2.finishReason
                   });
                 }
                 span2.setAttributes(
@@ -2819,7 +2823,8 @@ async function generateObject({
                   throw new NoObjectGeneratedError({
                     message: "No object generated: the tool was not called.",
                     response: responseData,
-                    usage: calculateLanguageModelUsage(result2.usage)
+                    usage: calculateLanguageModelUsage(result2.usage),
+                    finishReason: result2.finishReason
                   });
                 }
                 span2.setAttributes(
@@ -2875,7 +2880,8 @@ async function generateObject({
             cause: parseResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage(usage)
+            usage: calculateLanguageModelUsage(usage),
+            finishReason
           });
         }
         const validationResult = outputStrategy.validateFinalResult(
@@ -2892,7 +2898,8 @@ async function generateObject({
             cause: validationResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage(usage)
+            usage: calculateLanguageModelUsage(usage),
+            finishReason
           });
         }
         return validationResult.value;
@@ -3500,7 +3507,8 @@ var DefaultStreamObjectResult = class {
                       cause: validationResult.error,
                       text: accumulatedText,
                       response,
-                      usage
+                      usage,
+                      finishReason
                     });
                     self.objectPromise.reject(error);
                   }
@@ -4328,7 +4336,11 @@ async function generateText({
           }
           return output.parseOutput(
             { text: text2 },
-            { response: currentModelResponse.response, usage }
+            {
+              response: currentModelResponse.response,
+              usage,
+              finishReason: currentModelResponse.finishReason
+            }
           );
         },
         toolCalls: currentToolCalls,
@@ -4588,7 +4600,8 @@ var object = ({
           cause: parseResult.error,
           text: text2,
           response: context.response,
-          usage: context.usage
+          usage: context.usage,
+          finishReason: context.finishReason
         });
       }
       const validationResult = safeValidateTypes4({
@@ -4601,7 +4614,8 @@ var object = ({
           cause: validationResult.error,
           text: text2,
           response: context.response,
-          usage: context.usage
+          usage: context.usage,
+          finishReason: context.finishReason
         });
       }
       return validationResult.value;
@@ -4610,23 +4624,49 @@ var object = ({
 };
 
 // core/generate-text/smooth-stream.ts
-import { InvalidArgumentError as InvalidArgumentError2 } from "@ai-sdk/provider";
 import { delay as originalDelay } from "@ai-sdk/provider-utils";
+import { InvalidArgumentError as InvalidArgumentError2 } from "@ai-sdk/provider";
 var CHUNKING_REGEXPS = {
-  word: /\s*\S+\s+/m,
-  line: /[^\n]*\n/m
+  word: /\S+\s+/m,
+  line: /\n+/m
 };
 function smoothStream({
   delayInMs = 10,
   chunking = "word",
   _internal: { delay: delay2 = originalDelay } = {}
 } = {}) {
-  const chunkingRegexp = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking;
-  if (chunkingRegexp == null) {
-    throw new InvalidArgumentError2({
-      argument: "chunking",
-      message: `Chunking must be "word" or "line" or a RegExp. Received: ${chunking}`
-    });
+  let detectChunk;
+  if (typeof chunking === "function") {
+    detectChunk = (buffer) => {
+      const match = chunking(buffer);
+      if (match == null) {
+        return null;
+      }
+      if (!match.length) {
+        throw new Error(`Chunking function must return a non-empty string.`);
+      }
+      if (!buffer.startsWith(match)) {
+        throw new Error(
+          `Chunking function must return a match that is a prefix of the buffer. Received: "${match}" expected to start with "${buffer}"`
+        );
+      }
+      return match;
+    };
+  } else {
+    const chunkingRegex = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking;
+    if (chunkingRegex == null) {
+      throw new InvalidArgumentError2({
+        argument: "chunking",
+        message: `Chunking must be "word" or "line" or a RegExp. Received: ${chunking}`
+      });
+    }
+    detectChunk = (buffer) => {
+      const match = chunkingRegex.exec(buffer);
+      if (!match) {
+        return null;
+      }
+      return buffer.slice(0, match.index) + (match == null ? void 0 : match[0]);
+    };
   }
   return () => {
     let buffer = "";
@@ -4642,10 +4682,9 @@ function smoothStream({
         }
         buffer += chunk.textDelta;
         let match;
-        while ((match = chunkingRegexp.exec(buffer)) != null) {
-          const chunk2 = match[0];
-          controller.enqueue({ type: "text-delta", textDelta: chunk2 });
-          buffer = buffer.slice(chunk2.length);
+        while ((match = detectChunk(buffer)) != null) {
+          controller.enqueue({ type: "text-delta", textDelta: match });
+          buffer = buffer.slice(match.length);
           await delay2(delayInMs);
         }
       }