ai 6.0.229 → 6.0.231

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # ai
 
+## 6.0.231
+
+### Patch Changes
+
+- 926f5a0: Allow UI message chunks to include fields added by newer server versions.
+- 4c5a22f: Propagate abort reasons when generation is cancelled during tool execution.
+- 842a383: Prevent tools excluded by `activeTools` from being parsed or executed.
+- Updated dependencies [2dd1aab]
+- Updated dependencies [e567091]
+  - @ai-sdk/gateway@3.0.154
+
+## 6.0.230
+
+### Patch Changes
+
+- 19093fd: fix(ai): call `onInputStart` before `onInputAvailable` during non-streaming tool calls
+- Updated dependencies [19093fd]
+  - @ai-sdk/provider-utils@4.0.40
+  - @ai-sdk/gateway@3.0.153
+
 ## 6.0.229
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,4 +1,3 @@
-import * as _ai_sdk_provider from '@ai-sdk/provider';
 import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, EmbeddingModelV3Middleware, ImageModelV3, ImageModelV2, ImageModelV3ProviderMetadata, ImageModelV2ProviderMetadata, ImageModelV3Middleware, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, SharedV3Warning, LanguageModelV3Source, LanguageModelV3Middleware, RerankingModelV3, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, TranscriptionModelV3, TranscriptionModelV2, JSONObject, ImageModelV3Usage, AISDKError, LanguageModelV3ToolCall, JSONSchema7, LanguageModelV3CallOptions, LanguageModelV3ToolChoice, ProviderV3, JSONParseError, TypeValidationError, Experimental_VideoModelV3, Experimental_VideoModelV3FrameType, EmbeddingModelV3CallOptions, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
 export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, LoadSettingError, NoContentGeneratedError, NoSuchModelError, TooManyEmbeddingValuesForCallError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
 import { GatewayModelId } from '@ai-sdk/gateway';
@@ -1963,136 +1962,7 @@ type InferUIMessageToolCall<UI_MESSAGE extends UIMessage> = ValueOf<{
     dynamic: true;
 });
 
-declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
-    type: "text-start";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "text-delta";
-    id: string;
-    delta: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "text-end";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "error";
-    errorText: string;
-} | {
-    type: "tool-input-start";
-    toolCallId: string;
-    toolName: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-input-delta";
-    toolCallId: string;
-    inputTextDelta: string;
-} | {
-    type: "tool-input-available";
-    toolCallId: string;
-    toolName: string;
-    input: unknown;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-input-error";
-    toolCallId: string;
-    toolName: string;
-    input: unknown;
-    errorText: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-approval-request";
-    approvalId: string;
-    toolCallId: string;
-    signature?: string | undefined;
-} | {
-    type: "tool-output-available";
-    toolCallId: string;
-    output: unknown;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    preliminary?: boolean | undefined;
-} | {
-    type: "tool-output-error";
-    toolCallId: string;
-    errorText: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-} | {
-    type: "tool-output-denied";
-    toolCallId: string;
-} | {
-    type: "reasoning-start";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "reasoning-delta";
-    id: string;
-    delta: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "reasoning-end";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "source-url";
-    sourceId: string;
-    url: string;
-    title?: string | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "source-document";
-    sourceId: string;
-    mediaType: string;
-    title: string;
-    filename?: string | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "file";
-    url: string;
-    mediaType: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: `data-${string}`;
-    data: unknown;
-    id?: string | undefined;
-    transient?: boolean | undefined;
-} | {
-    type: "start-step";
-} | {
-    type: "finish-step";
-} | {
-    type: "start";
-    messageId?: string | undefined;
-    messageMetadata?: unknown;
-} | {
-    type: "finish";
-    finishReason?: "length" | "error" | "stop" | "content-filter" | "tool-calls" | "other" | undefined;
-    messageMetadata?: unknown;
-} | {
-    type: "abort";
-    reason?: string | undefined;
-} | {
-    type: "message-metadata";
-    messageMetadata: unknown;
-}>;
+declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<UIMessageChunk<unknown, UIDataTypes>>;
 type DataUIMessageChunk<DATA_TYPES extends UIDataTypes> = ValueOf<{
     [NAME in keyof DATA_TYPES & string]: {
         type: `data-${NAME}`;

package/dist/index.d.ts

@@ -1,4 +1,3 @@
-import * as _ai_sdk_provider from '@ai-sdk/provider';
 import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, EmbeddingModelV3Middleware, ImageModelV3, ImageModelV2, ImageModelV3ProviderMetadata, ImageModelV2ProviderMetadata, ImageModelV3Middleware, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, SharedV3Warning, LanguageModelV3Source, LanguageModelV3Middleware, RerankingModelV3, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, TranscriptionModelV3, TranscriptionModelV2, JSONObject, ImageModelV3Usage, AISDKError, LanguageModelV3ToolCall, JSONSchema7, LanguageModelV3CallOptions, LanguageModelV3ToolChoice, ProviderV3, JSONParseError, TypeValidationError, Experimental_VideoModelV3, Experimental_VideoModelV3FrameType, EmbeddingModelV3CallOptions, ProviderV2, NoSuchModelError } from '@ai-sdk/provider';
 export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, JSONSchema7, LoadAPIKeyError, LoadSettingError, NoContentGeneratedError, NoSuchModelError, TooManyEmbeddingValuesForCallError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
 import { GatewayModelId } from '@ai-sdk/gateway';
@@ -1963,136 +1962,7 @@ type InferUIMessageToolCall<UI_MESSAGE extends UIMessage> = ValueOf<{
     dynamic: true;
 });
 
-declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
-    type: "text-start";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "text-delta";
-    id: string;
-    delta: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "text-end";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "error";
-    errorText: string;
-} | {
-    type: "tool-input-start";
-    toolCallId: string;
-    toolName: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-input-delta";
-    toolCallId: string;
-    inputTextDelta: string;
-} | {
-    type: "tool-input-available";
-    toolCallId: string;
-    toolName: string;
-    input: unknown;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-input-error";
-    toolCallId: string;
-    toolName: string;
-    input: unknown;
-    errorText: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    title?: string | undefined;
-} | {
-    type: "tool-approval-request";
-    approvalId: string;
-    toolCallId: string;
-    signature?: string | undefined;
-} | {
-    type: "tool-output-available";
-    toolCallId: string;
-    output: unknown;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-    preliminary?: boolean | undefined;
-} | {
-    type: "tool-output-error";
-    toolCallId: string;
-    errorText: string;
-    providerExecuted?: boolean | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-    toolMetadata?: JSONObject | undefined;
-    dynamic?: boolean | undefined;
-} | {
-    type: "tool-output-denied";
-    toolCallId: string;
-} | {
-    type: "reasoning-start";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "reasoning-delta";
-    id: string;
-    delta: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "reasoning-end";
-    id: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "source-url";
-    sourceId: string;
-    url: string;
-    title?: string | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "source-document";
-    sourceId: string;
-    mediaType: string;
-    title: string;
-    filename?: string | undefined;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: "file";
-    url: string;
-    mediaType: string;
-    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
-} | {
-    type: `data-${string}`;
-    data: unknown;
-    id?: string | undefined;
-    transient?: boolean | undefined;
-} | {
-    type: "start-step";
-} | {
-    type: "finish-step";
-} | {
-    type: "start";
-    messageId?: string | undefined;
-    messageMetadata?: unknown;
-} | {
-    type: "finish";
-    finishReason?: "length" | "error" | "stop" | "content-filter" | "tool-calls" | "other" | undefined;
-    messageMetadata?: unknown;
-} | {
-    type: "abort";
-    reason?: string | undefined;
-} | {
-    type: "message-metadata";
-    messageMetadata: unknown;
-}>;
+declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<UIMessageChunk<unknown, UIDataTypes>>;
 type DataUIMessageChunk<DATA_TYPES extends UIDataTypes> = ValueOf<{
     [NAME in keyof DATA_TYPES & string]: {
         type: `data-${NAME}`;

package/dist/index.js

@@ -1282,7 +1282,7 @@ function detectMediaType({
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.229" : "0.0.0-test";
+var VERSION = true ? "6.0.231" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -3114,6 +3114,21 @@ function extractTextContent(content) {
   return parts.map((content2) => content2.text).join("");
 }
 
+// src/generate-text/filter-active-tools.ts
+function filterActiveTools({
+  tools,
+  activeTools
+}) {
+  if (tools == null || activeTools == null) {
+    return tools;
+  }
+  return Object.fromEntries(
+    Object.entries(tools).filter(
+      ([name22]) => activeTools.includes(name22)
+    )
+  );
+}
+
 // src/generate-text/generated-file.ts
 var import_provider_utils10 = require("@ai-sdk/provider-utils");
 var DefaultGeneratedFile = class {
@@ -4650,6 +4665,9 @@ async function generateText({
         const steps = [];
         const pendingDeferredToolCalls = /* @__PURE__ */ new Map();
         do {
+          if (steps.length > 0) {
+            mergedAbortSignal == null ? void 0 : mergedAbortSignal.throwIfAborted();
+          }
           const stepTimeoutId = setAbortTimeout({
             abortController: stepAbortController,
             label: "Step",
@@ -4681,6 +4699,10 @@ async function generateText({
             });
             experimental_context = (_d = prepareStepResult == null ? void 0 : prepareStepResult.experimental_context) != null ? _d : experimental_context;
             const stepActiveTools = (_e = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _e : activeTools;
+            const stepToolSet = filterActiveTools({
+              tools,
+              activeTools: stepActiveTools
+            });
             const { toolChoice: stepToolChoice, tools: stepTools } = await prepareToolsAndToolChoice({
               tools,
               toolChoice: (_f = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _f : toolChoice,
@@ -4833,7 +4855,7 @@ async function generateText({
               ).map(
                 (toolCall) => parseToolCall({
                   toolCall,
-                  tools,
+                  tools: stepToolSet,
                   repairToolCall,
                   system,
                   messages: stepInputMessages
@@ -4845,11 +4867,19 @@ async function generateText({
               if (toolCall.invalid) {
                 continue;
               }
-              const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
+              const tool2 = stepToolSet == null ? void 0 : stepToolSet[toolCall.toolName];
               if (tool2 == null) {
                 continue;
               }
-              if ((tool2 == null ? void 0 : tool2.onInputAvailable) != null) {
+              if (tool2.onInputStart != null) {
+                await tool2.onInputStart({
+                  toolCallId: toolCall.toolCallId,
+                  messages: stepInputMessages,
+                  abortSignal: mergedAbortSignal,
+                  experimental_context
+                });
+              }
+              if (tool2.onInputAvailable != null) {
                 await tool2.onInputAvailable({
                   input: toolCall.input,
                   toolCallId: toolCall.toolCallId,
@@ -4897,13 +4927,13 @@ async function generateText({
             clientToolCalls = stepToolCalls.filter(
               (toolCall) => !toolCall.providerExecuted
             );
-            if (tools != null) {
+            if (stepToolSet != null) {
               clientToolOutputs.push(
                 ...await executeTools({
                   toolCalls: clientToolCalls.filter(
                     (toolCall) => !toolCall.invalid && toolApprovalRequests[toolCall.toolCallId] == null
                   ),
-                  tools,
+                  tools: stepToolSet,
                   tracer,
                   telemetry,
                   messages: stepInputMessages,
@@ -4925,7 +4955,7 @@ async function generateText({
             for (const toolCall of stepToolCalls) {
               if (!toolCall.providerExecuted)
                 continue;
-              const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
+              const tool2 = stepToolSet == null ? void 0 : stepToolSet[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
@@ -4947,12 +4977,12 @@ async function generateText({
               toolCalls: stepToolCalls,
               toolOutputs: clientToolOutputs,
               toolApprovalRequests: Object.values(toolApprovalRequests),
-              tools
+              tools: stepToolSet
             });
             responseMessages.push(
               ...await toResponseMessages({
                 content: stepContent,
-                tools
+                tools: stepToolSet
               })
             );
             const stepRequest = ((_i = include == null ? void 0 : include.requestBody) != null ? _i : true) ? (_j = currentModelResponse.request) != null ? _j : {} : { ...currentModelResponse.request, body: void 0 };
@@ -5510,27 +5540,27 @@ var toolMetadataSchema = import_v47.z.record(
 var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
   () => (0, import_provider_utils17.zodSchema)(
     import_v47.z.union([
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-start"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-delta"),
         id: import_v47.z.string(),
         delta: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-end"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("error"),
         errorText: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-start"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
@@ -5540,12 +5570,12 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         dynamic: import_v47.z.boolean().optional(),
         title: import_v47.z.string().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-delta"),
         toolCallId: import_v47.z.string(),
         inputTextDelta: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-available"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
@@ -5556,7 +5586,7 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         dynamic: import_v47.z.boolean().optional(),
         title: import_v47.z.string().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-error"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
@@ -5568,13 +5598,13 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         errorText: import_v47.z.string(),
         title: import_v47.z.string().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-approval-request"),
         approvalId: import_v47.z.string(),
         toolCallId: import_v47.z.string(),
         signature: import_v47.z.string().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-output-available"),
         toolCallId: import_v47.z.string(),
         output: import_v47.z.unknown(),
@@ -5584,7 +5614,7 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         dynamic: import_v47.z.boolean().optional(),
         preliminary: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-output-error"),
         toolCallId: import_v47.z.string(),
         errorText: import_v47.z.string(),
@@ -5593,34 +5623,34 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         toolMetadata: toolMetadataSchema.optional(),
         dynamic: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-output-denied"),
         toolCallId: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-start"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-delta"),
         id: import_v47.z.string(),
         delta: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-end"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("source-url"),
         sourceId: import_v47.z.string(),
         url: import_v47.z.string(),
         title: import_v47.z.string().optional(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("source-document"),
         sourceId: import_v47.z.string(),
         mediaType: import_v47.z.string(),
@@ -5628,13 +5658,13 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         filename: import_v47.z.string().optional(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("file"),
         url: import_v47.z.string(),
         mediaType: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.custom(
           (value) => typeof value === "string" && value.startsWith("data-"),
           { message: 'Type must start with "data-"' }
@@ -5643,18 +5673,18 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         data: import_v47.z.unknown(),
         transient: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("start-step")
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("finish-step")
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("start"),
         messageId: import_v47.z.string().optional(),
         messageMetadata: import_v47.z.unknown().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("finish"),
         finishReason: import_v47.z.enum([
           "stop",
@@ -5666,11 +5696,11 @@ var uiMessageChunkSchema = (0, import_provider_utils17.lazySchema)(
         ]).optional(),
         messageMetadata: import_v47.z.unknown().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("abort"),
         reason: import_v47.z.string().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("message-metadata"),
         messageMetadata: import_v47.z.unknown()
       })
@@ -7124,6 +7154,7 @@ var DefaultStreamTextResult = class {
     let recordedWarnings = [];
     const recordedSteps = [];
     let recordedNoOutputError;
+    let currentStepToolSet = tools;
     const pendingDeferredToolCalls = /* @__PURE__ */ new Map();
     let rootSpan;
     let activeTextContent = createIdMap();
@@ -7251,7 +7282,7 @@ var DefaultStreamTextResult = class {
         if (part.type === "finish-step") {
           const stepMessages = await toResponseMessages({
             content: recordedContent,
-            tools
+            tools: currentStepToolSet
           });
           const currentStepResult = new DefaultStepResult({
             stepNumber: recordedSteps.length,
@@ -7682,6 +7713,11 @@ var DefaultStreamTextResult = class {
               download: download2
             });
             const stepActiveTools = (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools;
+            const stepToolSet = filterActiveTools({
+              tools,
+              activeTools: stepActiveTools
+            });
+            currentStepToolSet = stepToolSet;
             const { toolChoice: stepToolChoice, tools: stepTools } = await prepareToolsAndToolChoice({
               tools,
               toolChoice: (_e = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _e : toolChoice,
@@ -7781,7 +7817,7 @@ var DefaultStreamTextResult = class {
               })
             );
             const streamWithToolResults = runToolsTransformation({
-              tools,
+              tools: stepToolSet,
               generatorStream: stream2,
               tracer,
               telemetry,
@@ -7932,7 +7968,7 @@ var DefaultStreamTextResult = class {
                       }
                       case "tool-input-start": {
                         activeToolCallToolNames[chunk.id] = chunk.toolName;
-                        const tool2 = tools == null ? void 0 : tools[chunk.toolName];
+                        const tool2 = stepToolSet == null ? void 0 : stepToolSet[chunk.toolName];
                         if ((tool2 == null ? void 0 : tool2.onInputStart) != null) {
                           await tool2.onInputStart({
                             toolCallId: chunk.id,
@@ -7955,7 +7991,7 @@ var DefaultStreamTextResult = class {
                       }
                       case "tool-input-delta": {
                         const toolName = activeToolCallToolNames[chunk.id];
-                        const tool2 = tools == null ? void 0 : tools[toolName];
+                        const tool2 = stepToolSet == null ? void 0 : stepToolSet[toolName];
                         if ((tool2 == null ? void 0 : tool2.onInputDelta) != null) {
                           await tool2.onInputDelta({
                             inputTextDelta: chunk.delta,
@@ -8087,7 +8123,7 @@ var DefaultStreamTextResult = class {
                     for (const toolCall of stepToolCalls) {
                       if (toolCall.providerExecuted !== true)
                         continue;
-                      const tool2 = tools == null ? void 0 : tools[toolCall.toolName];
+                      const tool2 = stepToolSet == null ? void 0 : stepToolSet[toolCall.toolName];
                       if ((tool2 == null ? void 0 : tool2.type) === "provider" && tool2.supportsDeferredResults) {
                         const hasResultInStep = stepToolOutputs.some(
                           (output2) => (output2.type === "tool-result" || output2.type === "tool-error") && output2.toolCallId === toolCall.toolCallId
@@ -8122,7 +8158,7 @@ var DefaultStreamTextResult = class {
                             // use transformed content to create the messages for the next step:
                             recordedSteps[recordedSteps.length - 1].content
                           ),
-                          tools
+                          tools: stepToolSet
                         })
                       );
                       try {