ai 6.0.23 → 6.0.25

package/dist/index.js

@@ -96,11 +96,9 @@ __export(src_exports, {
   generateObject: () => generateObject,
   generateText: () => generateText,
   getStaticToolName: () => getStaticToolName,
-  getStepTimeoutMs: () => getStepTimeoutMs,
   getTextFromDataUrl: () => getTextFromDataUrl,
   getToolName: () => getToolName,
   getToolOrDynamicToolName: () => getToolOrDynamicToolName,
-  getTotalTimeoutMs: () => getTotalTimeoutMs,
   hasToolCall: () => hasToolCall,
   isDataUIPart: () => isDataUIPart,
   isDeepEqualData: () => isDeepEqualData,
@@ -826,6 +824,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
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
@@ -1020,7 +1024,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_provider_utils4 = require("@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 }) => {
@@ -2991,12 +2995,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 = ({
@@ -3006,6 +3014,7 @@ var object = ({
 }) => {
   const schema = (0, import_provider_utils13.asSchema)(inputSchema);
   return {
+    name: "object",
     responseFormat: (0, import_provider_utils13.resolve)(schema.jsonSchema).then((jsonSchema2) => ({
       type: "json",
       schema: jsonSchema2,
@@ -3055,6 +3064,9 @@ var object = ({
           };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3065,6 +3077,7 @@ var array = ({
 }) => {
   const elementSchema = (0, import_provider_utils13.asSchema)(inputElementSchema);
   return {
+    name: "array",
     // JSON schema that describes an array of elements:
     responseFormat: (0, import_provider_utils13.resolve)(elementSchema.jsonSchema).then((jsonSchema2) => {
       const { $schema, ...itemSchema } = jsonSchema2;
@@ -3154,6 +3167,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]);
+            }
+          }
+        }
+      });
     }
   };
 };
@@ -3163,6 +3188,7 @@ var choice = ({
   description
 }) => {
   return {
+    name: "choice",
     // JSON schema that describes an enumeration:
     responseFormat: Promise.resolve({
       type: "json",
@@ -3229,6 +3255,9 @@ var choice = ({
           }
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -3237,6 +3266,7 @@ var json = ({
   description
 } = {}) => {
   return {
+    name: "json",
     responseFormat: Promise.resolve({
       type: "json",
       ...name16 != null && { name: name16 },
@@ -3268,6 +3298,9 @@ var json = ({
           return result.value === void 0 ? void 0 : { partial: result.value };
         }
       }
+    },
+    createElementStreamTransform() {
+      return void 0;
     }
   };
 };
@@ -5771,7 +5804,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,
@@ -5781,10 +5816,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,
@@ -5881,6 +5919,8 @@ var DefaultStreamTextResult = class {
     abortSignal,
     stepTimeoutMs,
     stepAbortController,
+    chunkTimeoutMs,
+    chunkAbortController,
     system,
     prompt,
     messages,
@@ -6347,6 +6387,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 import_provider_utils19.DelayedPromise();
           const stepInputMessages = [...initialMessages, ...responseMessages];
           const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
@@ -6471,6 +6534,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;
@@ -6701,9 +6765,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
@@ -6902,6 +6965,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 import_provider24.UnsupportedFunctionalityError({
+        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;
@@ -12154,11 +12227,9 @@ var TextStreamChatTransport = class extends HttpChatTransport {
   generateObject,
   generateText,
   getStaticToolName,
-  getStepTimeoutMs,
   getTextFromDataUrl,
   getToolName,
   getToolOrDynamicToolName,
-  getTotalTimeoutMs,
   hasToolCall,
   isDataUIPart,
   isDeepEqualData,