ai 7.0.0-beta.20 → 7.0.0-beta.21

package/dist/index.mjs

@@ -1016,6 +1016,12 @@ function getChunkTimeoutMs(timeout) {
   }
   return timeout.chunkMs;
 }
+function getToolTimeoutMs(timeout) {
+  if (timeout == null || typeof timeout === "number") {
+    return void 0;
+  }
+  return timeout.toolMs;
+}
 
 // src/prompt/convert-to-language-model-prompt.ts
 import {
@@ -1262,7 +1268,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.0-beta.20" : "0.0.0-test";
+var VERSION = true ? "7.0.0-beta.21" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -3136,6 +3142,7 @@ async function executeToolCall({
   callId,
   messages,
   abortSignal,
+  toolTimeoutMs,
   experimental_context,
   stepNumber,
   model,
@@ -3162,6 +3169,7 @@ async function executeToolCall({
   };
   let output;
   await notify({ event: baseCallbackEvent, callbacks: onToolCallStart });
+  const toolAbortSignal = toolTimeoutMs != null ? abortSignal != null ? AbortSignal.any([abortSignal, AbortSignal.timeout(toolTimeoutMs)]) : AbortSignal.timeout(toolTimeoutMs) : abortSignal;
   const startTime = now();
   try {
     await executeToolInTelemetryContext({
@@ -3174,7 +3182,7 @@ async function executeToolCall({
           options: {
             toolCallId,
             messages,
-            abortSignal,
+            abortSignal: toolAbortSignal,
             experimental_context
           }
         });
@@ -4448,6 +4456,7 @@ async function generateText({
   const stopConditions = asArray(stopWhen);
   const totalTimeoutMs = getTotalTimeoutMs(timeout);
   const stepTimeoutMs = getStepTimeoutMs(timeout);
+  const toolTimeoutMs = getToolTimeoutMs(timeout);
   const stepAbortController = stepTimeoutMs != null ? new AbortController() : void 0;
   const mergedAbortSignal = mergeAbortSignals(
     abortSignal,
@@ -4529,6 +4538,7 @@ async function generateText({
         callId,
         messages: initialMessages,
         abortSignal: mergedAbortSignal,
+        toolTimeoutMs,
         experimental_context,
         stepNumber: 0,
         model: modelInfo,
@@ -4775,6 +4785,7 @@ async function generateText({
               callId,
               messages: stepInputMessages,
               abortSignal: mergedAbortSignal,
+              toolTimeoutMs,
               experimental_context,
               stepNumber: steps.length,
               model: stepModelInfo,
@@ -4957,6 +4968,7 @@ async function executeTools({
   callId,
   messages,
   abortSignal,
+  toolTimeoutMs,
   experimental_context,
   stepNumber,
   model,
@@ -4973,6 +4985,7 @@ async function executeTools({
         callId,
         messages,
         abortSignal,
+        toolTimeoutMs,
         experimental_context,
         stepNumber,
         model,
@@ -6394,6 +6407,24 @@ function createStitchableStream() {
   };
 }
 
+// src/generate-text/create-stream-text-part-transform.ts
+function createStreamTextPartTransform() {
+  return new TransformStream({
+    async transform(chunk, controller) {
+      if (chunk.type === "text-delta") {
+        controller.enqueue({
+          type: "text-delta",
+          id: chunk.id,
+          text: chunk.delta,
+          providerMetadata: chunk.providerMetadata
+        });
+      } else {
+        controller.enqueue(chunk);
+      }
+    }
+  });
+}
+
 // src/generate-text/run-tools-transformation.ts
 import {
   getErrorMessage as getErrorMessage6
@@ -6407,6 +6438,7 @@ function runToolsTransformation({
   messages,
   abortSignal,
   repairToolCall,
+  toolTimeoutMs,
   experimental_context,
   generateId: generateId2,
   stepNumber,
@@ -6440,6 +6472,7 @@ function runToolsTransformation({
       switch (chunkType) {
         case "stream-start":
         case "text-start":
+        case "text-delta":
         case "text-end":
         case "reasoning-start":
         case "reasoning-end":
@@ -6453,14 +6486,6 @@ function runToolsTransformation({
           controller.enqueue(chunk);
           break;
         }
-        case "text-delta":
-          controller.enqueue({
-            type: "text-delta",
-            id: chunk.id,
-            text: chunk.delta,
-            providerMetadata: chunk.providerMetadata
-          });
-          break;
         case "reasoning-delta":
           controller.enqueue({
             type: "reasoning-delta",
@@ -6570,6 +6595,7 @@ function runToolsTransformation({
                 callId,
                 messages,
                 abortSignal,
+                toolTimeoutMs,
                 experimental_context,
                 stepNumber,
                 model,
@@ -6716,6 +6742,7 @@ function streamText({
   const totalTimeoutMs = getTotalTimeoutMs(timeout);
   const stepTimeoutMs = getStepTimeoutMs(timeout);
   const chunkTimeoutMs = getChunkTimeoutMs(timeout);
+  const toolTimeoutMs = getToolTimeoutMs(timeout);
   const stepAbortController = stepTimeoutMs != null ? new AbortController() : void 0;
   const chunkAbortController = chunkTimeoutMs != null ? new AbortController() : void 0;
   return new DefaultStreamTextResult({
@@ -6734,6 +6761,7 @@ function streamText({
     stepAbortController,
     chunkTimeoutMs,
     chunkAbortController,
+    toolTimeoutMs,
     system,
     prompt,
     messages,
@@ -6841,6 +6869,7 @@ var DefaultStreamTextResult = class {
     stepAbortController,
     chunkTimeoutMs,
     chunkAbortController,
+    toolTimeoutMs,
     system,
     prompt,
     messages,
@@ -7272,6 +7301,7 @@ var DefaultStreamTextResult = class {
                 callId,
                 messages: initialMessages,
                 abortSignal,
+                toolTimeoutMs,
                 experimental_context,
                 stepNumber: recordedSteps.length,
                 model: modelInfo,
@@ -7444,7 +7474,11 @@ var DefaultStreamTextResult = class {
             ]
           });
           const stepStartTimestampMs = now2();
-          const { stream: stream2, response, request } = await retry(
+          const {
+            stream: languageModelStream,
+            response,
+            request
+          } = await retry(
             async () => stepModel.doStream({
               ...callSettings,
               tools: stepTools,
@@ -7457,6 +7491,9 @@ var DefaultStreamTextResult = class {
               includeRawChunks: includeRawChunks2
             })
           );
+          const stream2 = languageModelStream.pipeThrough(
+            createStreamTextPartTransform()
+          );
           const streamWithToolResults = runToolsTransformation({
             tools,
             generatorStream: stream2,
@@ -7466,6 +7503,7 @@ var DefaultStreamTextResult = class {
             messages: stepInputMessages,
             repairToolCall,
             abortSignal,
+            toolTimeoutMs,
             experimental_context,
             generateId: generateId2,
             stepNumber: recordedSteps.length,