ai 6.0.22 → 6.0.24

package/dist/index.mjs

@@ -710,6 +710,18 @@ function getTotalTimeoutMs(timeout) {
   }
   return timeout.totalMs;
 }
+function getStepTimeoutMs(timeout) {
+  if (timeout == null || typeof timeout === "number") {
+    return void 0;
+  }
+  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 {
@@ -909,7 +921,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.22" : "0.0.0-test";
+var VERSION = true ? "6.0.24" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -3582,9 +3594,12 @@ async function generateText({
   const model = resolveLanguageModel(modelArg);
   const stopConditions = asArray(stopWhen);
   const totalTimeoutMs = getTotalTimeoutMs(timeout);
+  const stepTimeoutMs = getStepTimeoutMs(timeout);
+  const stepAbortController = stepTimeoutMs != null ? new AbortController() : void 0;
   const mergedAbortSignal = mergeAbortSignals(
     abortSignal,
-    totalTimeoutMs != null ? AbortSignal.timeout(totalTimeoutMs) : void 0
+    totalTimeoutMs != null ? AbortSignal.timeout(totalTimeoutMs) : void 0,
+    stepAbortController == null ? void 0 : stepAbortController.signal
   );
   const { maxRetries, retry } = prepareRetries({
     maxRetries: maxRetriesArg,
@@ -3713,263 +3728,270 @@ async function generateText({
         const steps = [];
         const pendingDeferredToolCalls = /* @__PURE__ */ new Map();
         do {
-          const stepInputMessages = [...initialMessages, ...responseMessages];
-          const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
-            model,
-            steps,
-            stepNumber: steps.length,
-            messages: stepInputMessages,
-            experimental_context
-          }));
-          const stepModel = resolveLanguageModel(
-            (_a16 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a16 : model
-          );
-          const promptMessages = await convertToLanguageModelPrompt({
-            prompt: {
-              system: (_b = prepareStepResult == null ? void 0 : prepareStepResult.system) != null ? _b : initialPrompt.system,
-              messages: (_c = prepareStepResult == null ? void 0 : prepareStepResult.messages) != null ? _c : stepInputMessages
-            },
-            supportedUrls: await stepModel.supportedUrls,
-            download: download2
-          });
-          experimental_context = (_d = prepareStepResult == null ? void 0 : prepareStepResult.experimental_context) != null ? _d : experimental_context;
-          const { toolChoice: stepToolChoice, tools: stepTools } = await prepareToolsAndToolChoice({
-            tools,
-            toolChoice: (_e = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _e : toolChoice,
-            activeTools: (_f = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _f : activeTools
-          });
-          currentModelResponse = await retry(
-            () => {
-              var _a17;
-              return recordSpan({
-                name: "ai.generateText.doGenerate",
-                attributes: selectTelemetryAttributes({
-                  telemetry,
-                  attributes: {
-                    ...assembleOperationName({
-                      operationId: "ai.generateText.doGenerate",
-                      telemetry
-                    }),
-                    ...baseTelemetryAttributes,
-                    // model:
-                    "ai.model.provider": stepModel.provider,
-                    "ai.model.id": stepModel.modelId,
-                    // prompt:
-                    "ai.prompt.messages": {
-                      input: () => stringifyForTelemetry(promptMessages)
-                    },
-                    "ai.prompt.tools": {
-                      // convert the language model level tools:
-                      input: () => stepTools == null ? void 0 : stepTools.map((tool2) => JSON.stringify(tool2))
-                    },
-                    "ai.prompt.toolChoice": {
-                      input: () => stepToolChoice != null ? JSON.stringify(stepToolChoice) : void 0
-                    },
-                    // standardized gen-ai llm span attributes:
-                    "gen_ai.system": stepModel.provider,
-                    "gen_ai.request.model": stepModel.modelId,
-                    "gen_ai.request.frequency_penalty": settings.frequencyPenalty,
-                    "gen_ai.request.max_tokens": settings.maxOutputTokens,
-                    "gen_ai.request.presence_penalty": settings.presencePenalty,
-                    "gen_ai.request.stop_sequences": settings.stopSequences,
-                    "gen_ai.request.temperature": (_a17 = settings.temperature) != null ? _a17 : void 0,
-                    "gen_ai.request.top_k": settings.topK,
-                    "gen_ai.request.top_p": settings.topP
+          const stepTimeoutId = stepTimeoutMs != null ? setTimeout(() => stepAbortController.abort(), stepTimeoutMs) : void 0;
+          try {
+            const stepInputMessages = [...initialMessages, ...responseMessages];
+            const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
+              model,
+              steps,
+              stepNumber: steps.length,
+              messages: stepInputMessages,
+              experimental_context
+            }));
+            const stepModel = resolveLanguageModel(
+              (_a16 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a16 : model
+            );
+            const promptMessages = await convertToLanguageModelPrompt({
+              prompt: {
+                system: (_b = prepareStepResult == null ? void 0 : prepareStepResult.system) != null ? _b : initialPrompt.system,
+                messages: (_c = prepareStepResult == null ? void 0 : prepareStepResult.messages) != null ? _c : stepInputMessages
+              },
+              supportedUrls: await stepModel.supportedUrls,
+              download: download2
+            });
+            experimental_context = (_d = prepareStepResult == null ? void 0 : prepareStepResult.experimental_context) != null ? _d : experimental_context;
+            const { toolChoice: stepToolChoice, tools: stepTools } = await prepareToolsAndToolChoice({
+              tools,
+              toolChoice: (_e = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _e : toolChoice,
+              activeTools: (_f = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _f : activeTools
+            });
+            currentModelResponse = await retry(
+              () => {
+                var _a17;
+                return recordSpan({
+                  name: "ai.generateText.doGenerate",
+                  attributes: selectTelemetryAttributes({
+                    telemetry,
+                    attributes: {
+                      ...assembleOperationName({
+                        operationId: "ai.generateText.doGenerate",
+                        telemetry
+                      }),
+                      ...baseTelemetryAttributes,
+                      // model:
+                      "ai.model.provider": stepModel.provider,
+                      "ai.model.id": stepModel.modelId,
+                      // prompt:
+                      "ai.prompt.messages": {
+                        input: () => stringifyForTelemetry(promptMessages)
+                      },
+                      "ai.prompt.tools": {
+                        // convert the language model level tools:
+                        input: () => stepTools == null ? void 0 : stepTools.map((tool2) => JSON.stringify(tool2))
+                      },
+                      "ai.prompt.toolChoice": {
+                        input: () => stepToolChoice != null ? JSON.stringify(stepToolChoice) : void 0
+                      },
+                      // standardized gen-ai llm span attributes:
+                      "gen_ai.system": stepModel.provider,
+                      "gen_ai.request.model": stepModel.modelId,
+                      "gen_ai.request.frequency_penalty": settings.frequencyPenalty,
+                      "gen_ai.request.max_tokens": settings.maxOutputTokens,
+                      "gen_ai.request.presence_penalty": settings.presencePenalty,
+                      "gen_ai.request.stop_sequences": settings.stopSequences,
+                      "gen_ai.request.temperature": (_a17 = settings.temperature) != null ? _a17 : void 0,
+                      "gen_ai.request.top_k": settings.topK,
+                      "gen_ai.request.top_p": settings.topP
+                    }
+                  }),
+                  tracer,
+                  fn: async (span2) => {
+                    var _a18, _b2, _c2, _d2, _e2, _f2, _g2, _h2;
+                    const stepProviderOptions = mergeObjects(
+                      providerOptions,
+                      prepareStepResult == null ? void 0 : prepareStepResult.providerOptions
+                    );
+                    const result = await stepModel.doGenerate({
+                      ...callSettings2,
+                      tools: stepTools,
+                      toolChoice: stepToolChoice,
+                      responseFormat: await (output == null ? void 0 : output.responseFormat),
+                      prompt: promptMessages,
+                      providerOptions: stepProviderOptions,
+                      abortSignal: mergedAbortSignal,
+                      headers: headersWithUserAgent
+                    });
+                    const responseData = {
+                      id: (_b2 = (_a18 = result.response) == null ? void 0 : _a18.id) != null ? _b2 : generateId2(),
+                      timestamp: (_d2 = (_c2 = result.response) == null ? void 0 : _c2.timestamp) != null ? _d2 : /* @__PURE__ */ new Date(),
+                      modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : stepModel.modelId,
+                      headers: (_g2 = result.response) == null ? void 0 : _g2.headers,
+                      body: (_h2 = result.response) == null ? void 0 : _h2.body
+                    };
+                    span2.setAttributes(
+                      await selectTelemetryAttributes({
+                        telemetry,
+                        attributes: {
+                          "ai.response.finishReason": result.finishReason.unified,
+                          "ai.response.text": {
+                            output: () => extractTextContent(result.content)
+                          },
+                          "ai.response.toolCalls": {
+                            output: () => {
+                              const toolCalls = asToolCalls(result.content);
+                              return toolCalls == null ? void 0 : JSON.stringify(toolCalls);
+                            }
+                          },
+                          "ai.response.id": responseData.id,
+                          "ai.response.model": responseData.modelId,
+                          "ai.response.timestamp": responseData.timestamp.toISOString(),
+                          "ai.response.providerMetadata": JSON.stringify(
+                            result.providerMetadata
+                          ),
+                          // TODO rename telemetry attributes to inputTokens and outputTokens
+                          "ai.usage.promptTokens": result.usage.inputTokens.total,
+                          "ai.usage.completionTokens": result.usage.outputTokens.total,
+                          // standardized gen-ai llm span attributes:
+                          "gen_ai.response.finish_reasons": [
+                            result.finishReason.unified
+                          ],
+                          "gen_ai.response.id": responseData.id,
+                          "gen_ai.response.model": responseData.modelId,
+                          "gen_ai.usage.input_tokens": result.usage.inputTokens.total,
+                          "gen_ai.usage.output_tokens": result.usage.outputTokens.total
+                        }
+                      })
+                    );
+                    return { ...result, response: responseData };
                   }
-                }),
-                tracer,
-                fn: async (span2) => {
-                  var _a18, _b2, _c2, _d2, _e2, _f2, _g2, _h2;
-                  const stepProviderOptions = mergeObjects(
-                    providerOptions,
-                    prepareStepResult == null ? void 0 : prepareStepResult.providerOptions
-                  );
-                  const result = await stepModel.doGenerate({
-                    ...callSettings2,
-                    tools: stepTools,
-                    toolChoice: stepToolChoice,
-                    responseFormat: await (output == null ? void 0 : output.responseFormat),
-                    prompt: promptMessages,
-                    providerOptions: stepProviderOptions,
-                    abortSignal: mergedAbortSignal,
-                    headers: headersWithUserAgent
-                  });
-                  const responseData = {
-                    id: (_b2 = (_a18 = result.response) == null ? void 0 : _a18.id) != null ? _b2 : generateId2(),
-                    timestamp: (_d2 = (_c2 = result.response) == null ? void 0 : _c2.timestamp) != null ? _d2 : /* @__PURE__ */ new Date(),
-                    modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : stepModel.modelId,
-                    headers: (_g2 = result.response) == null ? void 0 : _g2.headers,
-                    body: (_h2 = result.response) == null ? void 0 : _h2.body
-                  };
-                  span2.setAttributes(
-                    await selectTelemetryAttributes({
-                      telemetry,
-                      attributes: {
-                        "ai.response.finishReason": result.finishReason.unified,
-                        "ai.response.text": {
-                          output: () => extractTextContent(result.content)
-                        },
-                        "ai.response.toolCalls": {
-                          output: () => {
-                            const toolCalls = asToolCalls(result.content);
-                            return toolCalls == null ? void 0 : JSON.stringify(toolCalls);
-                          }
-                        },
-                        "ai.response.id": responseData.id,
-                        "ai.response.model": responseData.modelId,
-                        "ai.response.timestamp": responseData.timestamp.toISOString(),
-                        "ai.response.providerMetadata": JSON.stringify(
-                          result.providerMetadata
-                        ),
-                        // TODO rename telemetry attributes to inputTokens and outputTokens
-                        "ai.usage.promptTokens": result.usage.inputTokens.total,
-                        "ai.usage.completionTokens": result.usage.outputTokens.total,
-                        // standardized gen-ai llm span attributes:
-                        "gen_ai.response.finish_reasons": [
-                          result.finishReason.unified
-                        ],
-                        "gen_ai.response.id": responseData.id,
-                        "gen_ai.response.model": responseData.modelId,
-                        "gen_ai.usage.input_tokens": result.usage.inputTokens.total,
-                        "gen_ai.usage.output_tokens": result.usage.outputTokens.total
-                      }
-                    })
-                  );
-                  return { ...result, response: responseData };
-                }
-              });
-            }
-          );
-          const stepToolCalls = await Promise.all(
-            currentModelResponse.content.filter(
-              (part) => part.type === "tool-call"
-            ).map(
-              (toolCall) => parseToolCall({
+                });
+              }
+            );
+            const stepToolCalls = await Promise.all(
+              currentModelResponse.content.filter(
+                (part) => part.type === "tool-call"
+              ).map(
+                (toolCall) => parseToolCall({
+                  toolCall,
+                  tools,
+                  repairToolCall,
+                  system,
+                  messages: stepInputMessages
+                })
+              )
+            );
+            const toolApprovalRequests = {};
+            for (const toolCall of stepToolCalls) {
+              if (toolCall.invalid) {
+                continue;
+              }
+              const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
+              if (tool2 == null) {
+                continue;
+              }
+              if ((tool2 == null ? void 0 : tool2.onInputAvailable) != null) {
+                await tool2.onInputAvailable({
+                  input: toolCall.input,
+                  toolCallId: toolCall.toolCallId,
+                  messages: stepInputMessages,
+                  abortSignal: mergedAbortSignal,
+                  experimental_context
+                });
+              }
+              if (await isApprovalNeeded({
+                tool: tool2,
                 toolCall,
-                tools,
-                repairToolCall,
-                system,
-                messages: stepInputMessages
-              })
-            )
-          );
-          const toolApprovalRequests = {};
-          for (const toolCall of stepToolCalls) {
-            if (toolCall.invalid) {
-              continue;
-            }
-            const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
-            if (tool2 == null) {
-              continue;
-            }
-            if ((tool2 == null ? void 0 : tool2.onInputAvailable) != null) {
-              await tool2.onInputAvailable({
-                input: toolCall.input,
-                toolCallId: toolCall.toolCallId,
                 messages: stepInputMessages,
-                abortSignal: mergedAbortSignal,
                 experimental_context
-              });
+              })) {
+                toolApprovalRequests[toolCall.toolCallId] = {
+                  type: "tool-approval-request",
+                  approvalId: generateId2(),
+                  toolCall
+                };
+              }
             }
-            if (await isApprovalNeeded({
-              tool: tool2,
-              toolCall,
-              messages: stepInputMessages,
-              experimental_context
-            })) {
-              toolApprovalRequests[toolCall.toolCallId] = {
-                type: "tool-approval-request",
-                approvalId: generateId2(),
-                toolCall
-              };
+            const invalidToolCalls = stepToolCalls.filter(
+              (toolCall) => toolCall.invalid && toolCall.dynamic
+            );
+            clientToolOutputs = [];
+            for (const toolCall of invalidToolCalls) {
+              clientToolOutputs.push({
+                type: "tool-error",
+                toolCallId: toolCall.toolCallId,
+                toolName: toolCall.toolName,
+                input: toolCall.input,
+                error: getErrorMessage5(toolCall.error),
+                dynamic: true
+              });
             }
-          }
-          const invalidToolCalls = stepToolCalls.filter(
-            (toolCall) => toolCall.invalid && toolCall.dynamic
-          );
-          clientToolOutputs = [];
-          for (const toolCall of invalidToolCalls) {
-            clientToolOutputs.push({
-              type: "tool-error",
-              toolCallId: toolCall.toolCallId,
-              toolName: toolCall.toolName,
-              input: toolCall.input,
-              error: getErrorMessage5(toolCall.error),
-              dynamic: true
-            });
-          }
-          clientToolCalls = stepToolCalls.filter(
-            (toolCall) => !toolCall.providerExecuted
-          );
-          if (tools != null) {
-            clientToolOutputs.push(
-              ...await executeTools({
-                toolCalls: clientToolCalls.filter(
-                  (toolCall) => !toolCall.invalid && toolApprovalRequests[toolCall.toolCallId] == null
-                ),
-                tools,
-                tracer,
-                telemetry,
-                messages: stepInputMessages,
-                abortSignal: mergedAbortSignal,
-                experimental_context
-              })
+            clientToolCalls = stepToolCalls.filter(
+              (toolCall) => !toolCall.providerExecuted
             );
-          }
-          for (const toolCall of stepToolCalls) {
-            if (!toolCall.providerExecuted)
-              continue;
-            const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
-            if ((tool2 == null ? void 0 : tool2.type) === "provider" && tool2.supportsDeferredResults) {
-              const hasResultInResponse = currentModelResponse.content.some(
-                (part) => part.type === "tool-result" && part.toolCallId === toolCall.toolCallId
+            if (tools != null) {
+              clientToolOutputs.push(
+                ...await executeTools({
+                  toolCalls: clientToolCalls.filter(
+                    (toolCall) => !toolCall.invalid && toolApprovalRequests[toolCall.toolCallId] == null
+                  ),
+                  tools,
+                  tracer,
+                  telemetry,
+                  messages: stepInputMessages,
+                  abortSignal: mergedAbortSignal,
+                  experimental_context
+                })
               );
-              if (!hasResultInResponse) {
-                pendingDeferredToolCalls.set(toolCall.toolCallId, {
-                  toolName: toolCall.toolName
-                });
+            }
+            for (const toolCall of stepToolCalls) {
+              if (!toolCall.providerExecuted)
+                continue;
+              const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
+              if ((tool2 == null ? void 0 : tool2.type) === "provider" && tool2.supportsDeferredResults) {
+                const hasResultInResponse = currentModelResponse.content.some(
+                  (part) => part.type === "tool-result" && part.toolCallId === toolCall.toolCallId
+                );
+                if (!hasResultInResponse) {
+                  pendingDeferredToolCalls.set(toolCall.toolCallId, {
+                    toolName: toolCall.toolName
+                  });
+                }
               }
             }
-          }
-          for (const part of currentModelResponse.content) {
-            if (part.type === "tool-result") {
-              pendingDeferredToolCalls.delete(part.toolCallId);
+            for (const part of currentModelResponse.content) {
+              if (part.type === "tool-result") {
+                pendingDeferredToolCalls.delete(part.toolCallId);
+              }
             }
-          }
-          const stepContent = asContent({
-            content: currentModelResponse.content,
-            toolCalls: stepToolCalls,
-            toolOutputs: clientToolOutputs,
-            toolApprovalRequests: Object.values(toolApprovalRequests),
-            tools
-          });
-          responseMessages.push(
-            ...await toResponseMessages({
-              content: stepContent,
+            const stepContent = asContent({
+              content: currentModelResponse.content,
+              toolCalls: stepToolCalls,
+              toolOutputs: clientToolOutputs,
+              toolApprovalRequests: Object.values(toolApprovalRequests),
               tools
-            })
-          );
-          const currentStepResult = new DefaultStepResult({
-            content: stepContent,
-            finishReason: currentModelResponse.finishReason.unified,
-            rawFinishReason: currentModelResponse.finishReason.raw,
-            usage: asLanguageModelUsage(currentModelResponse.usage),
-            warnings: currentModelResponse.warnings,
-            providerMetadata: currentModelResponse.providerMetadata,
-            request: (_g = currentModelResponse.request) != null ? _g : {},
-            response: {
-              ...currentModelResponse.response,
-              // deep clone msgs to avoid mutating past messages in multi-step:
-              messages: structuredClone(responseMessages)
+            });
+            responseMessages.push(
+              ...await toResponseMessages({
+                content: stepContent,
+                tools
+              })
+            );
+            const currentStepResult = new DefaultStepResult({
+              content: stepContent,
+              finishReason: currentModelResponse.finishReason.unified,
+              rawFinishReason: currentModelResponse.finishReason.raw,
+              usage: asLanguageModelUsage(currentModelResponse.usage),
+              warnings: currentModelResponse.warnings,
+              providerMetadata: currentModelResponse.providerMetadata,
+              request: (_g = currentModelResponse.request) != null ? _g : {},
+              response: {
+                ...currentModelResponse.response,
+                // deep clone msgs to avoid mutating past messages in multi-step:
+                messages: structuredClone(responseMessages)
+              }
+            });
+            logWarnings({
+              warnings: (_h = currentModelResponse.warnings) != null ? _h : [],
+              provider: stepModel.provider,
+              model: stepModel.modelId
+            });
+            steps.push(currentStepResult);
+            await (onStepFinish == null ? void 0 : onStepFinish(currentStepResult));
+          } finally {
+            if (stepTimeoutId != null) {
+              clearTimeout(stepTimeoutId);
             }
-          });
-          logWarnings({
-            warnings: (_h = currentModelResponse.warnings) != null ? _h : [],
-            provider: stepModel.provider,
-            model: stepModel.modelId
-          });
-          steps.push(currentStepResult);
-          await (onStepFinish == null ? void 0 : onStepFinish(currentStepResult));
+          }
         } while (
           // Continue if:
           // 1. There are client tool calls that have all been executed, OR
@@ -5676,6 +5698,10 @@ function streamText({
   ...settings
 }) {
   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,
@@ -5684,8 +5710,14 @@ function streamText({
     maxRetries,
     abortSignal: mergeAbortSignals(
       abortSignal,
-      totalTimeoutMs != null ? AbortSignal.timeout(totalTimeoutMs) : void 0
+      totalTimeoutMs != null ? AbortSignal.timeout(totalTimeoutMs) : void 0,
+      stepAbortController == null ? void 0 : stepAbortController.signal,
+      chunkAbortController == null ? void 0 : chunkAbortController.signal
     ),
+    stepTimeoutMs,
+    stepAbortController,
+    chunkTimeoutMs,
+    chunkAbortController,
     system,
     prompt,
     messages,
@@ -5780,6 +5812,10 @@ var DefaultStreamTextResult = class {
     settings,
     maxRetries: maxRetriesArg,
     abortSignal,
+    stepTimeoutMs,
+    stepAbortController,
+    chunkTimeoutMs,
+    chunkAbortController,
     system,
     prompt,
     messages,
@@ -6245,6 +6281,30 @@ 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({
@@ -6369,6 +6429,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;
@@ -6599,6 +6660,8 @@ var DefaultStreamTextResult = class {
                       pendingDeferredToolCalls.delete(output2.toolCallId);
                     }
                   }
+                  clearStepTimeout();
+                  clearChunkTimeout();
                   if (
                     // Continue if:
                     // 1. There are client tool calls that have all been executed, OR
@@ -10182,7 +10245,16 @@ function smoothStream({
   _internal: { delay: delay2 = originalDelay } = {}
 } = {}) {
   let detectChunk;
-  if (typeof chunking === "function") {
+  if (chunking != null && typeof chunking === "object" && "segment" in chunking && typeof chunking.segment === "function") {
+    const segmenter = chunking;
+    detectChunk = (buffer) => {
+      if (buffer.length === 0)
+        return null;
+      const iterator = segmenter.segment(buffer)[Symbol.iterator]();
+      const first = iterator.next().value;
+      return (first == null ? void 0 : first.segment) || null;
+    };
+  } else if (typeof chunking === "function") {
     detectChunk = (buffer) => {
       const match = chunking(buffer);
       if (match == null) {
@@ -10199,11 +10271,11 @@ function smoothStream({
       return match;
     };
   } else {
-    const chunkingRegex = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking;
+    const chunkingRegex = typeof chunking === "string" ? CHUNKING_REGEXPS[chunking] : chunking instanceof RegExp ? chunking : void 0;
     if (chunkingRegex == null) {
       throw new InvalidArgumentError2({
         argument: "chunking",
-        message: `Chunking must be "word" or "line" or a RegExp. Received: ${chunking}`
+        message: `Chunking must be "word", "line", a RegExp, an Intl.Segmenter, or a ChunkDetector function. Received: ${chunking}`
       });
     }
     detectChunk = (buffer) => {
@@ -12083,7 +12155,6 @@ export {
   getTextFromDataUrl,
   getToolName,
   getToolOrDynamicToolName,
-  getTotalTimeoutMs,
   hasToolCall,
   isDataUIPart,
   isDeepEqualData,