ai 7.0.0-beta.111 → 7.0.0-beta.113

package/dist/internal/index.d.ts

@@ -1,4 +1,4 @@
-import { SystemModelMessage, ModelMessage, ToolSet, ProviderOptions, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, Context, InferToolSetContext, ReasoningPart, ReasoningFilePart, Arrayable } from '@ai-sdk/provider-utils';
+import { SystemModelMessage, ModelMessage, ToolSet, ProviderOptions, Arrayable, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, Context, InferToolSetContext, ReasoningPart, ReasoningFilePart, InferToolContext } from '@ai-sdk/provider-utils';
 export { convertAsyncIteratorToReadableStream } from '@ai-sdk/provider-utils';
 import { LanguageModelV4Prompt, LanguageModelV4, LanguageModelV3, LanguageModelV2, SharedV4Warning, LanguageModelV4Source, LanguageModelV4ToolChoice, LanguageModelV4FunctionTool, LanguageModelV4ProviderTool, LanguageModelV4CallOptions, LanguageModelV4Usage, JSONObject, EmbeddingModelV4Embedding, SharedV4ProviderMetadata, ProviderV4, ProviderV3, ProviderV2 } from '@ai-sdk/provider';
 import { GatewayModelId } from '@ai-sdk/gateway';
@@ -52,6 +52,14 @@ type Prompt = {
      * System message to include in the prompt. Can be used with `prompt` or `messages`.
      */
     system?: string | SystemModelMessage | Array<SystemModelMessage>;
+    /**
+     * Whether system messages are allowed in the `prompt` or `messages` fields.
+     *
+     * When disabled, system messages must be provided through the `system` option.
+     *
+     * @default false
+     */
+    allowSystemInMessages?: boolean;
 } & ({
     /**
      * A prompt. It can be either a text prompt or a list of messages.
@@ -90,7 +98,18 @@ type StandardizedPrompt = {
      */
     messages: ModelMessage[];
 };
-declare function standardizePrompt(prompt: Prompt): Promise<StandardizedPrompt>;
+/**
+ * Converts a prompt input into a standardized prompt with validated model
+ * messages.
+ *
+ * @param prompt - The prompt definition to standardize.
+ * Set `allowSystemInMessages` to true to allow system messages in the
+ * `prompt` or `messages` fields. System messages in the `system` option are
+ * always allowed.
+ * @returns The standardized prompt.
+ * @throws {InvalidPromptError} When the prompt is invalid.
+ */
+declare function standardizePrompt({ allowSystemInMessages, system, prompt, messages, }: Prompt): Promise<StandardizedPrompt>;
 
 declare function convertToLanguageModelPrompt({ prompt, supportedUrls, download, provider, }: {
     prompt: StandardizedPrompt;
@@ -452,7 +471,7 @@ type LogWarningsFunction = (options: {
  *
  * Called when the operation begins, before the embedding model is called.
  */
-interface EmbedOnStartEvent {
+type EmbedStartEvent = {
     /** Unique identifier for this embed call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type (e.g. 'ai.embed' or 'ai.embedMany'). */
@@ -469,21 +488,13 @@ interface EmbedOnStartEvent {
     readonly headers: Record<string, string | undefined> | undefined;
     /** Additional provider-specific options. */
     readonly providerOptions: ProviderOptions | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event passed to the `onFinish` callback for embed and embedMany operations.
  *
  * Called when the operation completes, after the embedding model returns.
  */
-interface EmbedOnFinishEvent {
+type EmbedEndEvent = {
     /** Unique identifier for this embed call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type (e.g. 'ai.embed' or 'ai.embedMany'). */
@@ -510,22 +521,14 @@ interface EmbedOnFinishEvent {
         headers?: Record<string, string>;
         body?: unknown;
     } | undefined> | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event fired when an individual embedding model call (inner operation doEmbed) begins.
  *
  * For `embed`, there is one call. For `embedMany`, there may be multiple
  * calls when values are chunked.
  */
-interface EmbedStartEvent {
+type EmbeddingModelCallStartEvent = {
     /** Unique identifier for this embed call, used to correlate events. */
     readonly callId: string;
     /** Unique identifier for this individual doEmbed invocation, used to correlate start/finish within parallel chunks. */
@@ -538,21 +541,13 @@ interface EmbedStartEvent {
     readonly modelId: string;
     /** The values being embedded in this particular model call. */
     readonly values: Array<string>;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event fired when an individual embedding model call (doEmbed) completes.
  *
  * Contains the embeddings, usage, and any warnings from the model response.
  */
-interface EmbedFinishEvent {
+type EmbeddingModelCallEndEvent = {
     /** Unique identifier for this embed call, used to correlate events. */
     readonly callId: string;
     /** Unique identifier for this individual doEmbed invocation, used to correlate start/finish within parallel chunks. */
@@ -569,7 +564,7 @@ interface EmbedFinishEvent {
     readonly embeddings: Array<Embedding>;
     /** Token usage for this model call. */
     readonly usage: EmbeddingModelUsage;
-}
+};
 
 /**
  * Event passed to the `experimental_onStart` callback of
@@ -579,7 +574,7 @@ interface EmbedFinishEvent {
  *
  * @deprecated
  */
-interface ObjectOnStartEvent {
+interface GenerateObjectStartEvent {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type (e.g. `'ai.generateObject'` or `'ai.streamObject'`). */
@@ -622,14 +617,6 @@ interface ObjectOnStartEvent {
     readonly schemaName: string | undefined;
     /** Optional description of the schema. */
     readonly schemaDescription: string | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
 }
 /**
  * Event passed to the `experimental_onStepStart` callback of
@@ -640,7 +627,7 @@ interface ObjectOnStartEvent {
  *
  * @deprecated
  */
-interface ObjectOnStepStartEvent {
+interface GenerateObjectStepStartEvent {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
     /** Zero-based index of the current step. Always `0` for object generation. */
@@ -653,8 +640,6 @@ interface ObjectOnStepStartEvent {
     readonly providerOptions: ProviderOptions | undefined;
     /** Additional HTTP headers sent with the request. */
     readonly headers: Record<string, string | undefined> | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
     /** The prompt messages in provider format (for telemetry). */
     readonly promptMessages?: LanguageModelV4Prompt;
 }
@@ -667,7 +652,7 @@ interface ObjectOnStepStartEvent {
  *
  * @deprecated
  */
-interface ObjectOnStepFinishEvent {
+interface GenerateObjectStepEndEvent {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
     /** Zero-based index of the current step. Always `0` for object generation. */
@@ -694,8 +679,6 @@ interface ObjectOnStepFinishEvent {
     };
     /** Additional provider-specific metadata. */
     readonly providerMetadata: ProviderMetadata | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
     /** Milliseconds from the start of the stream to the first chunk (streaming only). */
     readonly msToFirstChunk: number | undefined;
 }
@@ -709,7 +692,7 @@ interface ObjectOnStepFinishEvent {
  *
  * @deprecated
  */
-interface ObjectOnFinishEvent<RESULT> {
+interface GenerateObjectEndEvent<RESULT> {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
     /**
@@ -739,8 +722,6 @@ interface ObjectOnFinishEvent<RESULT> {
     };
     /** Additional provider-specific metadata. */
     readonly providerMetadata: ProviderMetadata | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
 }
 
 /**
@@ -760,6 +741,51 @@ type TimeoutConfiguration<TOOLS extends ToolSet> = number | {
     tools?: Partial<Record<`${keyof TOOLS & string}Ms`, number>>;
 };
 
+/**
+ * Tool names that are enabled for a generation step.
+ *
+ * `undefined` means no tool restriction is applied. Tool names are object keys
+ * at runtime, so the type is restricted to the string keys of the configured
+ * tool set.
+ */
+type ActiveTools<TOOLS extends ToolSet> = ReadonlyArray<keyof TOOLS & string> | undefined;
+
+/**
+ * Telemetry configuration.
+ */
+type TelemetryOptions = {
+    /**
+     * Enable or disable telemetry. Enabled by default when a telemetry
+     * integration is registered. Set to `false` to opt out.
+     */
+    isEnabled?: boolean;
+    /**
+     * Enable or disable input recording. Enabled by default.
+     *
+     * You might want to disable input recording to avoid recording sensitive
+     * information, to reduce data transfers, or to increase performance.
+     */
+    recordInputs?: boolean;
+    /**
+     * Enable or disable output recording. Enabled by default.
+     *
+     * You might want to disable output recording to avoid recording sensitive
+     * information, to reduce data transfers, or to increase performance.
+     */
+    recordOutputs?: boolean;
+    /**
+     * Identifier for this function. Used to group telemetry data by function.
+     */
+    functionId?: string;
+    /**
+     * Per-call telemetry integrations that receive lifecycle events during generation.
+     *
+     * When provided, these integrations will take precedence over the globally registered
+     * integrations for this call.
+     */
+    integrations?: Arrayable<Telemetry>;
+};
+
 /**
  * A generated file.
  */
@@ -780,6 +806,40 @@ interface GeneratedFile {
     readonly mediaType: string;
 }
 
+/**
+ * Reasoning output of a text generation. It contains a reasoning.
+ */
+interface ReasoningOutput {
+    type: 'reasoning';
+    /**
+     * The reasoning text.
+     */
+    text: string;
+    /**
+     * Additional provider-specific metadata. They are passed through
+     * to the provider from the AI SDK and enable provider-specific
+     * functionality that can be fully encapsulated in the provider.
+     */
+    providerMetadata?: ProviderMetadata;
+}
+/**
+ * Reasoning file output of a text generation.
+ * It contains a file generated as part of reasoning.
+ */
+interface ReasoningFileOutput {
+    type: 'reasoning-file';
+    /**
+     * The generated file.
+     */
+    file: GeneratedFile;
+    /**
+     * Additional provider-specific metadata. They are passed through
+     * to the provider from the AI SDK and enable provider-specific
+     * functionality that can be fully encapsulated in the provider.
+     */
+    providerMetadata?: ProviderMetadata;
+}
+
 /**
  * Create a union of the given object's values, and optionally specify which keys to get the values from.
  *
@@ -882,41 +942,41 @@ type ToolApprovalRequestOutput<TOOLS extends ToolSet> = {
      * Tool call that the approval request is for.
      */
     toolCall: TypedToolCall<TOOLS>;
+    /**
+     * Flag indicating whether the tool was automatically approved or denied.
+     *
+     * @default false
+     */
+    isAutomatic?: boolean;
 };
 
 /**
- * Reasoning output of a text generation. It contains a reasoning.
+ * Output part that indicates that a tool approval response is available.
  */
-interface ReasoningOutput {
-    type: 'reasoning';
+type ToolApprovalResponseOutput<TOOLS extends ToolSet> = {
+    type: 'tool-approval-response';
     /**
-     * The reasoning text.
+     * ID of the tool approval.
      */
-    text: string;
+    approvalId: string;
     /**
-     * Additional provider-specific metadata. They are passed through
-     * to the provider from the AI SDK and enable provider-specific
-     * functionality that can be fully encapsulated in the provider.
+     * Tool call that the approval response is for.
      */
-    providerMetadata?: ProviderMetadata;
-}
-/**
- * Reasoning file output of a text generation.
- * It contains a file generated as part of reasoning.
- */
-interface ReasoningFileOutput {
-    type: 'reasoning-file';
+    toolCall: TypedToolCall<TOOLS>;
     /**
-     * The generated file.
+     * Flag indicating whether the approval was granted or denied.
      */
-    file: GeneratedFile;
+    approved: boolean;
     /**
-     * Additional provider-specific metadata. They are passed through
-     * to the provider from the AI SDK and enable provider-specific
-     * functionality that can be fully encapsulated in the provider.
+     * Optional reason for the approval or denial.
      */
-    providerMetadata?: ProviderMetadata;
-}
+    reason?: string;
+    /**
+     * Flag indicating whether the tool call is provider-executed.
+     * Only provider-executed tool approval responses should be sent to the model.
+     */
+    providerExecuted?: boolean;
+};
 
 type StaticToolError<TOOLS extends ToolSet> = ValueOf<{
     [NAME in keyof TOOLS]: {
@@ -998,7 +1058,47 @@ type ContentPart<TOOLS extends ToolSet> = {
     type: 'tool-error';
 } & TypedToolError<TOOLS> & {
     providerMetadata?: ProviderMetadata;
-}) | ToolApprovalRequestOutput<TOOLS>;
+}) | ToolApprovalRequestOutput<TOOLS> | ToolApprovalResponseOutput<TOOLS>;
+
+/**
+ * Common model information used across callback events.
+ */
+type ModelInfo = {
+    /** The provider identifier (e.g., 'openai', 'anthropic'). */
+    readonly provider: string;
+    /** The specific model identifier (e.g., 'gpt-4o'). */
+    readonly modelId: string;
+};
+/**
+ * Event passed to the `onLanguageModelCallStart` callback.
+ *
+ * Called immediately before the provider model call begins.
+ * Unlike `onStepStart`, this only represents model invocation work.
+ */
+type LanguageModelCallStartEvent = ModelInfo & {
+    /** Unique identifier for this generation call, used to correlate events. */
+    readonly callId: string;
+    /** Prepared tool definitions for the model call, if any. */
+    readonly tools: ReadonlyArray<Record<string, unknown>> | undefined;
+} & StandardizedPrompt & LanguageModelCallOptions;
+/**
+ * Event passed to the `onLanguageModelCallEnd` callback.
+ *
+ * Called after the model response has been normalized and parsed, but before
+ * any client-side tool execution begins.
+ */
+type LanguageModelCallEndEvent<TOOLS extends ToolSet = ToolSet> = ModelInfo & {
+    /** Unique identifier for this generation call, used to correlate events. */
+    readonly callId: string;
+    /** The unified reason why the model call finished. */
+    readonly finishReason: FinishReason;
+    /** The token usage reported by the model call. */
+    readonly usage: LanguageModelUsage;
+    /** The content parts produced by the model call. */
+    readonly content: ReadonlyArray<ContentPart<TOOLS>>;
+    /** The provider-returned response id for this model call. */
+    readonly responseId: string;
+};
 
 /**
  * A message that was generated during the generation process.
@@ -1027,10 +1127,6 @@ type StepResult<TOOLS extends ToolSet, RUNTIME_CONTEXT extends Context = Context
         /** The ID of the model. */
         readonly modelId: string;
     };
-    /**
-     * Identifier from telemetry settings for grouping related operations.
-     */
-    readonly functionId: string | undefined;
     /**
      * Tool context.
      */
@@ -1130,20 +1226,6 @@ type StepResult<TOOLS extends ToolSet, RUNTIME_CONTEXT extends Context = Context
     readonly providerMetadata: ProviderMetadata | undefined;
 };
 
-/**
- * A predicate that decides whether a tool-calling loop should stop after the
- * current step.
- *
- * A tool calling loop continues until one of the following conditions is met:
- * - The model returns a finish reason other than `tool-calls`
- * - A tool without an execute function is called
- * - A tool call needs approval
- * - One of the provided stop conditions returns `true`
- */
-type StopCondition<TOOLS extends ToolSet, RUNTIME_CONTEXT extends Context = Context> = (options: {
-    steps: Array<StepResult<TOOLS, RUNTIME_CONTEXT>>;
-}) => PromiseLike<boolean> | boolean;
-
 /**
  * Tool output when the tool execution has been denied (for static tools).
  */
@@ -1240,6 +1322,7 @@ type TextStreamToolOutputDeniedPart<TOOLS extends ToolSet> = {
     type: 'tool-output-denied';
 } & StaticToolOutputDenied<TOOLS>;
 type TextStreamToolApprovalRequestPart<TOOLS extends ToolSet> = ToolApprovalRequestOutput<TOOLS>;
+type TextStreamToolApprovalResponsePart<TOOLS extends ToolSet> = ToolApprovalResponseOutput<TOOLS>;
 type TextStreamStartStepPart = {
     type: 'start-step';
     request: LanguageModelRequestMetadata;
@@ -1274,69 +1357,119 @@ type TextStreamRawPart = {
     type: 'raw';
     rawValue: unknown;
 };
-type TextStreamPart<TOOLS extends ToolSet> = TextStreamTextStartPart | TextStreamTextEndPart | TextStreamTextDeltaPart | TextStreamReasoningStartPart | TextStreamReasoningEndPart | TextStreamReasoningDeltaPart | TextStreamCustomPart | TextStreamToolInputStartPart | TextStreamToolInputEndPart | TextStreamToolInputDeltaPart | TextStreamSourcePart | TextStreamFilePart | TextStreamReasoningFilePart | TextStreamToolCallPart<TOOLS> | TextStreamToolResultPart<TOOLS> | TextStreamToolErrorPart<TOOLS> | TextStreamToolOutputDeniedPart<TOOLS> | TextStreamToolApprovalRequestPart<TOOLS> | TextStreamStartStepPart | TextStreamFinishStepPart | TextStreamStartPart | TextStreamFinishPart | TextStreamAbortPart | TextStreamErrorPart | TextStreamRawPart;
+type TextStreamPart<TOOLS extends ToolSet> = TextStreamTextStartPart | TextStreamTextEndPart | TextStreamTextDeltaPart | TextStreamReasoningStartPart | TextStreamReasoningEndPart | TextStreamReasoningDeltaPart | TextStreamCustomPart | TextStreamToolInputStartPart | TextStreamToolInputEndPart | TextStreamToolInputDeltaPart | TextStreamSourcePart | TextStreamFilePart | TextStreamReasoningFilePart | TextStreamToolCallPart<TOOLS> | TextStreamToolResultPart<TOOLS> | TextStreamToolErrorPart<TOOLS> | TextStreamToolOutputDeniedPart<TOOLS> | TextStreamToolApprovalRequestPart<TOOLS> | TextStreamToolApprovalResponsePart<TOOLS> | TextStreamStartStepPart | TextStreamFinishStepPart | TextStreamStartPart | TextStreamFinishPart | TextStreamAbortPart | TextStreamErrorPart | TextStreamRawPart;
 
+type ToolOutput<TOOLS extends ToolSet> = TypedToolResult<TOOLS> | TypedToolError<TOOLS>;
+
+/**
+ * Resolves a single tool's context type, falling back to `undefined` when the
+ * tool does not declare a `contextSchema`.
+ */
+type ToolContextFor<TOOL extends ToolSet[keyof ToolSet]> = [
+    InferToolContext<TOOL>
+] extends [never] ? undefined : InferToolContext<TOOL>;
+type BaseToolExecutionStartFields = {
+    /** Unique identifier for this generation call, used to correlate events. */
+    readonly callId: string;
+    /**
+     * Messages that were sent to the language model to initiate the response that contained the tool call.
+     * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
+     */
+    readonly messages: ModelMessage[];
+};
+/**
+ * Precise start event union for statically known tools.
+ *
+ * Each union member ties a specific `toolCall.toolName` to that tool's
+ * validated `toolContext` type.
+ */
+type StaticToolExecutionStartEvent<TOOLS extends ToolSet> = ValueOf<{
+    [NAME in keyof TOOLS]: BaseToolExecutionStartFields & {
+        readonly toolCall: Extract<StaticToolCall<TOOLS>, {
+            toolName: NAME;
+        }>;
+        readonly toolContext: ToolContextFor<TOOLS[NAME]>;
+    };
+}>;
+/**
+ * Start event shape for dynamic or untyped tool calls.
+ */
+type DynamicToolExecutionStartEvent = BaseToolExecutionStartFields & {
+    readonly toolCall: DynamicToolCall;
+    readonly toolContext: unknown;
+};
+/**
+ * Broad start event shape used for the default `ToolSet` specialization.
+ *
+ * This keeps generic collectors ergonomic when the caller is not working with
+ * a concrete tool set and therefore cannot benefit from per-tool narrowing.
+ */
+type WidenedToolExecutionStartEvent = BaseToolExecutionStartFields & {
+    readonly toolCall: StaticToolCall<ToolSet> | DynamicToolCall;
+    readonly toolContext: unknown;
+};
 /**
  * Event passed to the `onToolExecutionStart` callback.
  *
  * Called when a tool execution begins, before the tool's `execute` function is invoked.
  */
-interface ToolExecutionStartEvent<TOOLS extends ToolSet = ToolSet> {
+type ToolExecutionStartEvent<TOOLS extends ToolSet = ToolSet> = [
+    ToolSet
+] extends [TOOLS] ? WidenedToolExecutionStartEvent : StaticToolExecutionStartEvent<TOOLS> | DynamicToolExecutionStartEvent;
+type BaseToolExecutionEndFields = {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
-    /** Zero-based index of the current step where this tool call occurs. */
-    readonly stepNumber: number | undefined;
-    /** The provider identifier (e.g., 'openai', 'anthropic'). */
-    readonly provider: string | undefined;
-    /** The specific model identifier (e.g., 'gpt-4o'). */
-    readonly modelId: string | undefined;
-    /** The full tool call object. */
-    readonly toolCall: TypedToolCall<TOOLS>;
-    /** The conversation messages available at tool execution time. */
-    readonly messages: Array<ModelMessage>;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-    /** User-defined context object flowing through the generation. */
-    readonly context: InferToolSetContext<TOOLS>;
-}
+    /** Execution time of the tool call in milliseconds. */
+    readonly durationMs: number;
+    /**
+     * Messages that were sent to the language model to initiate the response that contained the tool call.
+     * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
+     */
+    readonly messages: ModelMessage[];
+};
+/**
+ * Precise end event union for statically known tools.
+ *
+ * Each union member preserves the link between `toolCall.toolName`, the
+ * corresponding validated `toolContext`, and the tool execution result.
+ */
+type StaticToolExecutionEndEvent<TOOLS extends ToolSet> = ValueOf<{
+    [NAME in keyof TOOLS]: BaseToolExecutionEndFields & {
+        readonly toolCall: Extract<StaticToolCall<TOOLS>, {
+            toolName: NAME;
+        }>;
+        readonly toolContext: ToolContextFor<TOOLS[NAME]>;
+        readonly toolOutput: ToolOutput<TOOLS>;
+    };
+}>;
+/**
+ * End event shape for dynamic or untyped tool calls.
+ */
+type DynamicToolExecutionEndEvent<TOOLS extends ToolSet> = BaseToolExecutionEndFields & {
+    readonly toolCall: DynamicToolCall;
+    readonly toolContext: unknown;
+    readonly toolOutput: ToolOutput<TOOLS>;
+};
+/**
+ * Broad end event shape used for the default `ToolSet` specialization.
+ *
+ * This provides an assignable catch-all event type for generic consumers while
+ * the concrete-tool specialization retains full per-tool narrowing.
+ */
+type WidenedToolExecutionEndEvent = BaseToolExecutionEndFields & {
+    readonly toolCall: StaticToolCall<ToolSet> | DynamicToolCall;
+    readonly toolContext: unknown;
+    readonly toolOutput: ToolOutput<ToolSet>;
+};
 /**
  * Event passed to the `onToolExecutionEnd` callback.
  *
  * Called when a tool execution completes, either successfully or with an error.
- * Uses a discriminated union on the `success` field.
+ * Uses the `toolOutput.type` discriminator to distinguish success and error.
  */
-type ToolExecutionEndEvent<TOOLS extends ToolSet = ToolSet> = {
-    /** Unique identifier for this generation call, used to correlate events. */
-    readonly callId: string;
-    /** Zero-based index of the current step where this tool call occurred. */
-    readonly stepNumber: number | undefined;
-    /** The provider identifier (e.g., 'openai', 'anthropic'). */
-    readonly provider: string | undefined;
-    /** The specific model identifier (e.g., 'gpt-4o'). */
-    readonly modelId: string | undefined;
-    /** The full tool call object. */
-    readonly toolCall: TypedToolCall<TOOLS>;
-    /** The conversation messages available at tool execution time. */
-    readonly messages: Array<ModelMessage>;
-    /** Execution time of the tool call in milliseconds. */
-    readonly durationMs: number;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-    /** User-defined context object flowing through the generation. */
-    readonly context: InferToolSetContext<TOOLS>;
-} & ({
-    /** Indicates the tool call succeeded. */
-    readonly success: true;
-    /** The tool's return value. */
-    readonly output: unknown;
-    readonly error?: never;
-} | {
-    /** Indicates the tool call failed. */
-    readonly success: false;
-    readonly output?: never;
-    /** The error that occurred during tool execution. */
-    readonly error: unknown;
-});
+type ToolExecutionEndEvent<TOOLS extends ToolSet = ToolSet> = [
+    ToolSet
+] extends [TOOLS] ? WidenedToolExecutionEndEvent : StaticToolExecutionEndEvent<TOOLS> | DynamicToolExecutionEndEvent<TOOLS>;
 
 type EnrichedStreamPart<TOOLS extends ToolSet, PARTIAL_OUTPUT> = {
     part: TextStreamPart<TOOLS>;
@@ -1381,7 +1514,7 @@ interface Output<OUTPUT = any, PARTIAL = any, ELEMENT = any> {
  *
  * Called when the generation operation begins, before any LLM calls.
  */
-interface OnStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context, OUTPUT extends Output = Output> {
+type GenerateTextStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context, OUTPUT extends Output = Output> = {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type (e.g. 'ai.generateText' or 'ai.streamText'). */
@@ -1390,34 +1523,12 @@ interface OnStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends
     readonly provider: string;
     /** The specific model identifier (e.g., 'gpt-4o'). */
     readonly modelId: string;
-    /** The system message(s) provided to the model. */
-    readonly system: string | SystemModelMessage | Array<SystemModelMessage> | undefined;
-    /** The prompt string or array of messages if using the prompt option. */
-    readonly prompt: string | Array<ModelMessage> | undefined;
-    /** The messages array if using the messages option. */
-    readonly messages: Array<ModelMessage> | undefined;
     /** The tools available for this generation. */
     readonly tools: TOOLS | undefined;
     /** The tool choice strategy for this generation. */
     readonly toolChoice: ToolChoice<NoInfer<TOOLS>> | undefined;
     /** Limits which tools are available for the model to call. */
-    readonly activeTools: Array<keyof TOOLS> | undefined;
-    /** Maximum number of tokens to generate. */
-    readonly maxOutputTokens: number | undefined;
-    /** Sampling temperature for generation. */
-    readonly temperature: number | undefined;
-    /** Top-p (nucleus) sampling parameter. */
-    readonly topP: number | undefined;
-    /** Top-k sampling parameter. */
-    readonly topK: number | undefined;
-    /** Presence penalty for generation. */
-    readonly presencePenalty: number | undefined;
-    /** Frequency penalty for generation. */
-    readonly frequencyPenalty: number | undefined;
-    /** Sequences that will stop generation. */
-    readonly stopSequences: string[] | undefined;
-    /** Random seed for reproducible generation. */
-    readonly seed: number | undefined;
+    readonly activeTools: ActiveTools<TOOLS>;
     /** Maximum number of retries for failed requests. */
     readonly maxRetries: number;
     /**
@@ -1429,21 +1540,8 @@ interface OnStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends
     readonly headers: Record<string, string | undefined> | undefined;
     /** Additional provider-specific options. */
     readonly providerOptions: ProviderOptions | undefined;
-    /**
-     * Condition(s) for stopping the generation.
-     * When the condition is an array, any of the conditions can be met to stop.
-     */
-    readonly stopWhen: Arrayable<StopCondition<NoInfer<TOOLS>, RUNTIME_CONTEXT>>;
     /** The output specification for structured outputs, if configured. */
     readonly output: OUTPUT | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
     /**
      * Tool context.
      */
@@ -1452,58 +1550,34 @@ interface OnStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends
      * User-defined runtime context.
      */
     readonly runtimeContext: RUNTIME_CONTEXT;
-}
+} & LanguageModelCallOptions & StandardizedPrompt;
 /**
  * Event passed to the `onStepStart` callback.
  *
  * Called when a step (LLM call) begins, before the provider is called.
  * Each step represents a single LLM invocation.
  */
-interface OnStepStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context, OUTPUT extends Output = Output> {
+type GenerateTextStepStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context, OUTPUT extends Output = Output> = {
     /** Unique identifier for this generation call, used to correlate events. */
     readonly callId: string;
-    /** Zero-based index of the current step. */
-    readonly stepNumber: number;
     /** The provider identifier (e.g., 'openai', 'anthropic'). */
     readonly provider: string;
     /** The specific model identifier (e.g., 'gpt-4o'). */
     readonly modelId: string;
-    /**
-     * The system message for this step.
-     */
-    readonly system: string | SystemModelMessage | Array<SystemModelMessage> | undefined;
-    /**
-     * The messages that will be sent to the model for this step.
-     * Uses the user-facing `ModelMessage` format.
-     * May be overridden by prepareStep.
-     */
-    readonly messages: Array<ModelMessage>;
+    /** Zero-based index of the current step. */
+    readonly stepNumber: number;
     /** The tools available for this generation. */
     readonly tools: TOOLS | undefined;
     /** The tool choice configuration for this step. */
     readonly toolChoice: ToolChoice<NoInfer<TOOLS>> | undefined;
     /** Limits which tools are available for this step. */
-    readonly activeTools: Array<keyof TOOLS> | undefined;
+    readonly activeTools: ActiveTools<TOOLS>;
     /** Array of results from previous steps (empty for first step). */
     readonly steps: ReadonlyArray<StepResult<TOOLS, RUNTIME_CONTEXT>>;
     /** Additional provider-specific options for this step. */
     readonly providerOptions: ProviderOptions | undefined;
-    /**
-     * Timeout configuration for the generation.
-     * Can be a number (milliseconds) or an object with totalMs, stepMs, chunkMs, toolMs, and per-tool overrides via tools.
-     */
-    readonly timeout: TimeoutConfiguration<TOOLS> | undefined;
-    /** Additional HTTP headers sent with the request. */
-    readonly headers: Record<string, string | undefined> | undefined;
-    /**
-     * Condition(s) for stopping the generation.
-     * When the condition is an array, any of the conditions can be met to stop.
-     */
-    readonly stopWhen: Arrayable<StopCondition<TOOLS, RUNTIME_CONTEXT>>;
     /** The output specification for structured outputs, if configured. */
     readonly output: OUTPUT | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
     /**
      * Runtime context. May be updated from `prepareStep` between steps.
      */
@@ -1512,7 +1586,7 @@ interface OnStepStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT exte
      * Tool context. May be updated from `prepareStep` between steps.
      */
     readonly toolsContext: InferToolSetContext<TOOLS>;
-}
+} & StandardizedPrompt;
 /**
  * Event passed to the `onChunk` callback.
  *
@@ -1520,34 +1594,32 @@ interface OnStepStartEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT exte
  * The chunk is either a content part (text-delta, tool-call, etc.) or
  * a stream lifecycle marker (`ai.stream.firstChunk` / `ai.stream.finish`).
  */
-interface OnChunkEvent<TOOLS extends ToolSet = ToolSet> {
+type StreamTextChunkEvent<TOOLS extends ToolSet = ToolSet> = {
     readonly chunk: TextStreamPart<TOOLS> | {
         readonly type: 'ai.stream.firstChunk' | 'ai.stream.finish';
         readonly callId: string;
         readonly stepNumber: number;
         readonly attributes?: Record<string, unknown>;
     };
-}
+};
 /**
  * Event passed to the `onStepFinish` callback.
  *
  * Called when a step (LLM call) completes.
  * Includes the StepResult for that step along with the call identifier.
  */
-type OnStepFinishEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context> = StepResult<TOOLS, RUNTIME_CONTEXT>;
+type GenerateTextStepEndEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context> = StepResult<TOOLS, RUNTIME_CONTEXT>;
 /**
  * Event passed to the `onFinish` callback.
  *
  * Called when the entire generation completes (all steps finished).
  * Includes the final step's result along with aggregated data from all steps.
  */
-type OnFinishEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context> = StepResult<TOOLS, RUNTIME_CONTEXT> & {
+type GenerateTextEndEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Context = Context> = StepResult<TOOLS, RUNTIME_CONTEXT> & {
     /** Array containing results from all steps in the generation. */
     readonly steps: StepResult<TOOLS, RUNTIME_CONTEXT>[];
     /** Aggregated token usage across all steps. */
     readonly totalUsage: LanguageModelUsage;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
 };
 
 /**
@@ -1555,7 +1627,7 @@ type OnFinishEvent<TOOLS extends ToolSet = ToolSet, RUNTIME_CONTEXT extends Cont
  *
  * Called when the operation begins, before the reranking model is called.
  */
-interface RerankOnStartEvent {
+type RerankStartEvent = {
     /** Unique identifier for this rerank call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type ('ai.rerank'). */
@@ -1575,21 +1647,13 @@ interface RerankOnStartEvent {
     readonly headers: Record<string, string | undefined> | undefined;
     /** Additional provider-specific options. */
     readonly providerOptions: ProviderOptions | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event passed to the `onFinish` callback for rerank operations.
  *
  * Called when the operation completes, after the reranking model returns.
  */
-interface RerankOnFinishEvent {
+type RerankEndEvent = {
     /** Unique identifier for this rerank call, used to correlate events. */
     readonly callId: string;
     /** Identifies the operation type ('ai.rerank'). */
@@ -1619,19 +1683,11 @@ interface RerankOnFinishEvent {
         headers?: Record<string, string>;
         body?: unknown;
     };
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event fired when an individual reranking model call (inner doRerank) begins.
  */
-interface RerankStartEvent {
+type RerankingModelCallStartEvent = {
     /** Unique identifier for this rerank call, used to correlate events. */
     readonly callId: string;
     /** Identifies the inner operation ('ai.rerank.doRerank'). */
@@ -1648,21 +1704,13 @@ interface RerankStartEvent {
     readonly query: string;
     /** Number of top documents to return. */
     readonly topN: number | undefined;
-    /** Whether telemetry is enabled. Defaults to `true`. */
-    readonly isEnabled: boolean | undefined;
-    /** Whether to record inputs in telemetry. Enabled by default. */
-    readonly recordInputs: boolean | undefined;
-    /** Whether to record outputs in telemetry. Enabled by default. */
-    readonly recordOutputs: boolean | undefined;
-    /** Identifier from telemetry settings for grouping related operations. */
-    readonly functionId: string | undefined;
-}
+};
 /**
  * Event fired when an individual reranking model call (doRerank) completes.
  *
  * Contains the ranking results from the model response.
  */
-interface RerankFinishEvent {
+type RerankingModelCallEndEvent = {
     /** Unique identifier for this rerank call, used to correlate events. */
     readonly callId: string;
     /** Identifies the inner operation ('ai.rerank.doRerank'). */
@@ -1678,8 +1726,11 @@ interface RerankFinishEvent {
         index: number;
         relevanceScore: number;
     }>;
-}
+};
 
+type InferTelemetryEvent<EVENT> = EVENT & Omit<TelemetryOptions, 'integrations' | 'isEnabled'>;
+type OperationStartEvent = GenerateTextStartEvent | GenerateObjectStartEvent | EmbedStartEvent | RerankStartEvent;
+type OperationFinishEvent = GenerateTextEndEvent<ToolSet> | GenerateObjectEndEvent<unknown> | EmbedEndEvent | RerankEndEvent;
 /**
  * Implement this interface to create custom telemetry integrations.
  * Methods can be sync or return a PromiseLike.
@@ -1692,7 +1743,7 @@ interface Telemetry {
      *
      * Use the `operationId` field to distinguish between operation types.
      */
-    onStart?: Callback<OnStartEvent | ObjectOnStartEvent | EmbedOnStartEvent | RerankOnStartEvent>;
+    onStart?: Callback<InferTelemetryEvent<OperationStartEvent>>;
     /**
      * Called when an individual step (single LLM invocation) begins.
      * A generation may consist of multiple steps (e.g. when tool calls trigger
@@ -1702,12 +1753,23 @@ interface Telemetry {
      * The event includes the step number, accumulated previous step results,
      * and the messages that will be sent to the model.
      */
-    onStepStart?: Callback<OnStepStartEvent>;
+    onStepStart?: Callback<InferTelemetryEvent<GenerateTextStepStartEvent>>;
+    /**
+     * Called immediately before the provider model call begins.
+     * Unlike `onStepStart`, this callback is scoped to model work only and
+     * excludes any later client-side tool execution.
+     */
+    onLanguageModelCallStart?: Callback<InferTelemetryEvent<LanguageModelCallStartEvent>>;
+    /**
+     * Called after the model response has been normalized and parsed, but before
+     * any client-side tool execution begins.
+     */
+    onLanguageModelCallEnd?: Callback<InferTelemetryEvent<LanguageModelCallEndEvent>>;
     /**
      * Called when a tool execution begins, before the tool's `execute` function
      * is invoked. Use this to create tool-level spans or log tool invocations.
      */
-    onToolExecutionStart?: Callback<ToolExecutionStartEvent>;
+    onToolExecutionStart?: Callback<InferTelemetryEvent<ToolExecutionStartEvent>>;
     /**
      * Called when a tool execution completes, either successfully or with an error.
      * The event uses a discriminated union on the `success` field — check
@@ -1715,54 +1777,54 @@ interface Telemetry {
      *
      * The event includes execution duration (`durationMs`) for performance tracking.
      */
-    onToolExecutionEnd?: Callback<ToolExecutionEndEvent>;
+    onToolExecutionEnd?: Callback<InferTelemetryEvent<ToolExecutionEndEvent>>;
     /**
      * Called for each chunk received during streaming.
      * Only relevant for `streamText` — not called during `generateText`.
      */
-    onChunk?: Callback<OnChunkEvent>;
+    onChunk?: Callback<StreamTextChunkEvent>;
     /**
      * Called when an individual step (single LLM invocation) completes.
      * The event is a `StepResult` containing the model's response, tool calls
      * and results, usage statistics, finish reason, and optional request/response
      * bodies.
      */
-    onStepFinish?: Callback<OnStepFinishEvent>;
+    onStepFinish?: Callback<InferTelemetryEvent<GenerateTextStepEndEvent>>;
     /**
      * Called when an object generation step (single LLM invocation) begins.
      * For generateObject/streamObject there is always exactly one step.
      *
      * @deprecated
      */
-    onObjectStepStart?: Callback<ObjectOnStepStartEvent>;
+    onObjectStepStart?: Callback<InferTelemetryEvent<GenerateObjectStepStartEvent>>;
     /**
      * Called when an object generation step (single LLM invocation) completes,
      * with the raw result before JSON parsing and schema validation.
      *
      * @deprecated
      */
-    onObjectStepFinish?: Callback<ObjectOnStepFinishEvent>;
+    onObjectStepFinish?: Callback<InferTelemetryEvent<GenerateObjectStepEndEvent>>;
     /**
      * Called when an individual embedding model call (doEmbed) begins.
      * For `embed`, there is one call. For `embedMany`, there may be multiple
      * calls when values are chunked.
      */
-    onEmbedStart?: Callback<EmbedStartEvent>;
+    onEmbedStart?: Callback<InferTelemetryEvent<EmbeddingModelCallStartEvent>>;
     /**
      * Called when an individual embedding model call (doEmbed) completes.
      * Contains the embeddings, usage, and any warnings from the model response.
      */
-    onEmbedFinish?: Callback<EmbedFinishEvent>;
+    onEmbedFinish?: Callback<InferTelemetryEvent<EmbeddingModelCallEndEvent>>;
     /**
      * Called when an individual reranking model call (doRerank) begins.
      * There is one call per `rerank` invocation.
      */
-    onRerankStart?: Callback<RerankStartEvent>;
+    onRerankStart?: Callback<InferTelemetryEvent<RerankingModelCallStartEvent>>;
     /**
      * Called when an individual reranking model call (doRerank) completes.
      * Contains the ranking results from the model response.
      */
-    onRerankFinish?: Callback<RerankFinishEvent>;
+    onRerankFinish?: Callback<InferTelemetryEvent<RerankingModelCallEndEvent>>;
     /**
      * Called when an operation completes. Fired for text generation
      * (generateText/streamText), object generation (generateObject/streamObject),
@@ -1770,7 +1832,7 @@ interface Telemetry {
      *
      * Use the event shape or `operationId` to distinguish between operation types.
      */
-    onFinish?: Callback<OnFinishEvent<ToolSet> | ObjectOnFinishEvent<unknown> | EmbedOnFinishEvent | RerankOnFinishEvent>;
+    onFinish?: Callback<InferTelemetryEvent<OperationFinishEvent>>;
     /**
      * Called when an unrecoverable error occurs during the generation lifecycle.
      * The error value is untyped — it may be an `Error` instance, an `AISDKError`,