ai 6.0.23 → 6.0.25

package/dist/index.mjs

@@ -716,6 +716,12 @@ function getStepTimeoutMs(timeout) {
   }
   return timeout.stepMs;
 }
+function getChunkTimeoutMs(timeout) {
+  if (timeout == null || typeof timeout === "number") {
+    return void 0;
+  }
+  return timeout.chunkMs;
+}
 
 // src/prompt/convert-to-language-model-prompt.ts
 import {
@@ -915,7 +921,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.23" : "0.0.0-test";
+var VERSION = true ? "6.0.25" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -2901,12 +2907,16 @@ async function parsePartialJson(jsonText) {
 
 // src/generate-text/output.ts
 var text = () => ({
+  name: "text",
   responseFormat: Promise.resolve({ type: "text" }),
   async parseCompleteOutput({ text: text2 }) {
     return text2;
   },
   async parsePartialOutput({ text: text2 }) {
     return { partial: text2 };
+  },
+  createElementStreamTransform() {
+    return void 0;
   }
 });
 var object = ({
@@ -2916,6 +2926,7 @@ var object = ({
 }) => {
   const schema = asSchema2(inputSchema);
   return {
+    name: "object",
     responseFormat: resolve(schema.jsonSchema).then((jsonSchema2) => ({
       type: "json",
       schema: jsonSchema2,
@@ -2965,6 +2976,9 @@ var object = ({
           };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -2975,6 +2989,7 @@ var array = ({
 }) => {
   const elementSchema = asSchema2(inputElementSchema);
   return {
+    name: "array",
     // JSON schema that describes an array of elements:
     responseFormat: resolve(elementSchema.jsonSchema).then((jsonSchema2) => {
       const { $schema, ...itemSchema } = jsonSchema2;
@@ -3064,6 +3079,18 @@ var array = ({
           return { partial: parsedElements };
         }
       }
+    },
+    createElementStreamTransform() {
+      let publishedElements = 0;
+      return new TransformStream({
+        transform({ partialOutput }, controller) {
+          if (partialOutput != null) {
+            for (; publishedElements < partialOutput.length; publishedElements++) {
+              controller.enqueue(partialOutput[publishedElements]);
+            }
+          }
+        }
+      });
     }
   };
 };
@@ -3073,6 +3100,7 @@ var choice = ({
   description
 }) => {
   return {
+    name: "choice",
     // JSON schema that describes an enumeration:
     responseFormat: Promise.resolve({
       type: "json",
@@ -3139,6 +3167,9 @@ var choice = ({
           }
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3147,6 +3178,7 @@ var json = ({
   description
 } = {}) => {
   return {
+    name: "json",
     responseFormat: Promise.resolve({
       type: "json",
       ...name16 != null && { name: name16 },
@@ -3178,6 +3210,9 @@ var json = ({
           return result.value === void 0 ? void 0 : { partial: result.value };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -4300,7 +4335,8 @@ function asContent({
 
 // src/generate-text/stream-text.ts
 import {
-  getErrorMessage as getErrorMessage7
+  getErrorMessage as getErrorMessage7,
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
 import {
   createIdGenerator as createIdGenerator2,
@@ -5693,7 +5729,9 @@ function streamText({
 }) {
   const totalTimeoutMs = getTotalTimeoutMs(timeout);
   const stepTimeoutMs = getStepTimeoutMs(timeout);
+  const chunkTimeoutMs = getChunkTimeoutMs(timeout);
   const stepAbortController = stepTimeoutMs != null ? new AbortController() : void 0;
+  const chunkAbortController = chunkTimeoutMs != null ? new AbortController() : void 0;
   return new DefaultStreamTextResult({
     model: resolveLanguageModel(model),
     telemetry,
@@ -5703,10 +5741,13 @@ function streamText({
     abortSignal: mergeAbortSignals(
       abortSignal,
       totalTimeoutMs != null ? AbortSignal.timeout(totalTimeoutMs) : void 0,
-      stepAbortController == null ? void 0 : stepAbortController.signal
+      stepAbortController == null ? void 0 : stepAbortController.signal,
+      chunkAbortController == null ? void 0 : chunkAbortController.signal
     ),
     stepTimeoutMs,
     stepAbortController,
+    chunkTimeoutMs,
+    chunkAbortController,
     system,
     prompt,
     messages,
@@ -5803,6 +5844,8 @@ var DefaultStreamTextResult = class {
     abortSignal,
     stepTimeoutMs,
     stepAbortController,
+    chunkTimeoutMs,
+    chunkAbortController,
     system,
     prompt,
     messages,
@@ -6269,6 +6312,29 @@ var DefaultStreamTextResult = class {
           var _a16, _b, _c, _d, _e, _f;
           const includeRawChunks2 = self.includeRawChunks;
           const stepTimeoutId = stepTimeoutMs != null ? setTimeout(() => stepAbortController.abort(), stepTimeoutMs) : void 0;
+          let chunkTimeoutId = void 0;
+          function resetChunkTimeout() {
+            if (chunkTimeoutMs != null) {
+              if (chunkTimeoutId != null) {
+                clearTimeout(chunkTimeoutId);
+              }
+              chunkTimeoutId = setTimeout(
+                () => chunkAbortController.abort(),
+                chunkTimeoutMs
+              );
+            }
+          }
+          function clearChunkTimeout() {
+            if (chunkTimeoutId != null) {
+              clearTimeout(chunkTimeoutId);
+              chunkTimeoutId = void 0;
+            }
+          }
+          function clearStepTimeout() {
+            if (stepTimeoutId != null) {
+              clearTimeout(stepTimeoutId);
+            }
+          }
           stepFinish = new DelayedPromise();
           const stepInputMessages = [...initialMessages, ...responseMessages];
           const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
@@ -6393,6 +6459,7 @@ var DefaultStreamTextResult = class {
               new TransformStream({
                 async transform(chunk, controller) {
                   var _a17, _b2, _c2, _d2, _e2;
+                  resetChunkTimeout();
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
                     return;
@@ -6623,9 +6690,8 @@ var DefaultStreamTextResult = class {
                       pendingDeferredToolCalls.delete(output2.toolCallId);
                     }
                   }
-                  if (stepTimeoutId != null) {
-                    clearTimeout(stepTimeoutId);
-                  }
+                  clearStepTimeout();
+                  clearChunkTimeout();
                   if (
                     // Continue if:
                     // 1. There are client tool calls that have all been executed, OR
@@ -6824,6 +6890,16 @@ var DefaultStreamTextResult = class {
       )
     );
   }
+  get elementStream() {
+    var _a16, _b, _c;
+    const transform = (_a16 = this.outputSpecification) == null ? void 0 : _a16.createElementStreamTransform();
+    if (transform == null) {
+      throw new UnsupportedFunctionalityError2({
+        functionality: `element streams in ${(_c = (_b = this.outputSpecification) == null ? void 0 : _b.name) != null ? _c : "text"} mode`
+      });
+    }
+    return createAsyncIterableStream(this.teeStream().pipeThrough(transform));
+  }
   get output() {
     return this.finalStep.then((step) => {
       var _a16;
@@ -8649,7 +8725,7 @@ import {
   isJSONArray,
   isJSONObject,
   TypeValidationError as TypeValidationError4,
-  UnsupportedFunctionalityError as UnsupportedFunctionalityError2
+  UnsupportedFunctionalityError as UnsupportedFunctionalityError3
 } from "@ai-sdk/provider";
 import {
   asSchema as asSchema4,
@@ -8674,7 +8750,7 @@ var noSchemaOutputStrategy = {
     } : { success: true, value };
   },
   createElementStream() {
-    throw new UnsupportedFunctionalityError2({
+    throw new UnsupportedFunctionalityError3({
       functionality: "element streams in no-schema mode"
     });
   }
@@ -8696,7 +8772,7 @@ var objectOutputStrategy = (schema) => ({
     return safeValidateTypes4({ value, schema });
   },
   createElementStream() {
-    throw new UnsupportedFunctionalityError2({
+    throw new UnsupportedFunctionalityError3({
       functionality: "element streams in object mode"
     });
   }
@@ -8885,7 +8961,7 @@ var enumOutputStrategy = (enumValues) => {
       };
     },
     createElementStream() {
-      throw new UnsupportedFunctionalityError2({
+      throw new UnsupportedFunctionalityError3({
         functionality: "element streams in enum mode"
       });
     }
@@ -12116,11 +12192,9 @@ export {
   generateObject,
   generateText,
   getStaticToolName,
-  getStepTimeoutMs,
   getTextFromDataUrl,
   getToolName,
   getToolOrDynamicToolName,
-  getTotalTimeoutMs,
   hasToolCall,
   isDataUIPart,
   isDeepEqualData,