ai 5.0.0-canary.0 → 5.0.0-canary.2

package/dist/index.mjs

@@ -1035,13 +1035,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);
@@ -2211,7 +2213,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 };
   },
@@ -2708,7 +2711,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(
@@ -2817,7 +2821,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(
@@ -2873,7 +2878,8 @@ async function generateObject({
             cause: parseResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage(usage)
+            usage: calculateLanguageModelUsage(usage),
+            finishReason
           });
         }
         const validationResult = outputStrategy.validateFinalResult(
@@ -2890,7 +2896,8 @@ async function generateObject({
             cause: validationResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage(usage)
+            usage: calculateLanguageModelUsage(usage),
+            finishReason
           });
         }
         return validationResult.value;
@@ -3498,7 +3505,8 @@ var DefaultStreamObjectResult = class {
                       cause: validationResult.error,
                       text: accumulatedText,
                       response,
-                      usage
+                      usage,
+                      finishReason
                     });
                     self.objectPromise.reject(error);
                   }
@@ -4326,7 +4334,11 @@ async function generateText({
           }
           return output.parseOutput(
             { text: text2 },
-            { response: currentModelResponse.response, usage }
+            {
+              response: currentModelResponse.response,
+              usage,
+              finishReason: currentModelResponse.finishReason
+            }
           );
         },
         toolCalls: currentToolCalls,
@@ -4586,7 +4598,8 @@ var object = ({
           cause: parseResult.error,
           text: text2,
           response: context.response,
-          usage: context.usage
+          usage: context.usage,
+          finishReason: context.finishReason
         });
       }
       const validationResult = safeValidateTypes4({
@@ -4599,7 +4612,8 @@ var object = ({
           cause: validationResult.error,
           text: text2,
           response: context.response,
-          usage: context.usage
+          usage: context.usage,
+          finishReason: context.finishReason
         });
       }
       return validationResult.value;
@@ -4608,23 +4622,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 = "";
@@ -4640,10 +4680,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);
         }
       }
@@ -4661,6 +4700,25 @@ function asArray(value) {
   return value === void 0 ? [] : Array.isArray(value) ? value : [value];
 }
 
+// util/consume-stream.ts
+async function consumeStream({
+  stream,
+  onError
+}) {
+  const reader = stream.getReader();
+  try {
+    while (true) {
+      const { done } = await reader.read();
+      if (done)
+        break;
+    }
+  } catch (error) {
+    onError == null ? void 0 : onError(error);
+  } finally {
+    reader.releaseLock();
+  }
+}
+
 // core/util/merge-streams.ts
 function mergeStreams(stream1, stream2) {
   const reader1 = stream1.getReader();
@@ -5902,9 +5960,15 @@ var DefaultStreamTextResult = class {
       )
     );
   }
-  async consumeStream() {
-    const stream = this.fullStream;
-    for await (const part of stream) {
+  async consumeStream(options) {
+    var _a17;
+    try {
+      await consumeStream({
+        stream: this.fullStream,
+        onError: options == null ? void 0 : options.onError
+      });
+    } catch (error) {
+      (_a17 = options == null ? void 0 : options.onError) == null ? void 0 : _a17.call(options, error);
     }
   }
   get experimental_partialOutputStream() {
@@ -6213,7 +6277,7 @@ function defaultSettingsMiddleware({
   settings
 }) {
   return {
-    middlewareVersion: "v1",
+    middlewareVersion: "v2",
     transformParams: async ({ params }) => {
       var _a17;
       return {
@@ -6258,7 +6322,7 @@ function extractReasoningMiddleware({
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
-    middlewareVersion: "v1",
+    middlewareVersion: "v2",
     wrapGenerate: async ({ doGenerate }) => {
       const { text: rawText, ...rest } = await doGenerate();
       if (rawText == null) {
@@ -6344,7 +6408,7 @@ function extractReasoningMiddleware({
 // core/middleware/simulate-streaming-middleware.ts
 function simulateStreamingMiddleware() {
   return {
-    middlewareVersion: "v1",
+    middlewareVersion: "v2",
     wrapStream: async ({ doGenerate }) => {
       const result = await doGenerate();
       const simulatedStream = new ReadableStream({
@@ -6449,7 +6513,7 @@ var doWrap = ({
     return transformParams ? await transformParams({ params, type }) : params;
   }
   return {
-    specificationVersion: "v1",
+    specificationVersion: "v2",
     provider: providerId != null ? providerId : model.provider,
     modelId: modelId != null ? modelId : model.modelId,
     defaultObjectGenerationMode: model.defaultObjectGenerationMode,