@ai-sdk/workflow 1.0.0-beta.8 → 1.0.0-beta.94

package/dist/index.d.mts

@@ -1,5 +1,6 @@
 import { LanguageModelV4, SharedV4ProviderOptions, LanguageModelV4CallOptions, LanguageModelV4Prompt, LanguageModelV4StreamPart } from '@ai-sdk/provider';
-import { ToolSet, LanguageModel, SystemModelMessage, ToolChoice, StopCondition, LanguageModelResponseMetadata, LanguageModelUsage, FinishReason, ToolCallRepairFunction, StepResult, StreamTextOnStepFinishCallback, ModelMessage, Experimental_LanguageModelStreamPart, UIMessage, UIMessageChunk, ChatTransport, PrepareSendMessagesRequest, PrepareReconnectToStreamRequest, ChatRequestOptions } from 'ai';
+import { Context, HasRequiredKey, InferToolSetContext } from '@ai-sdk/provider-utils';
+import { ToolSet, LanguageModel, Instructions, ToolChoice, TelemetryOptions as TelemetryOptions$1, StopCondition, ActiveTools, LanguageModelResponseMetadata, LanguageModelUsage, FinishReason, ToolCallRepairFunction, StepResult, GenerateTextOnStepEndCallback, ModelMessage, Experimental_LanguageModelStreamPart, UIMessage, UIMessageChunk, ChatTransport, PrepareSendMessagesRequest, PrepareReconnectToStreamRequest, ChatRequestOptions } from 'ai';
 export { Experimental_LanguageModelStreamPart as ModelCallStreamPart, Output, ToolCallRepairFunction } from 'ai';
 
 /**
@@ -15,14 +16,21 @@ type CompatibleLanguageModel = LanguageModelV4;
 
 /**
  * Callback function to be called after each step completes.
- * Alias for the AI SDK's StreamTextOnStepFinishCallback, using
+ * Alias for the AI SDK's GenerateTextOnStepEndCallback, using
  * WorkflowAgent-consistent naming.
  */
-type WorkflowAgentOnStepFinishCallback<TTools extends ToolSet = ToolSet> = StreamTextOnStepFinishCallback<TTools, any>;
+type WorkflowAgentOnStepEndCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = GenerateTextOnStepEndCallback<TTools, TRuntimeContext>;
+/**
+ * Callback function to be called after each step completes.
+ * Deprecated alias for `WorkflowAgentOnStepEndCallback`.
+ *
+ * @deprecated Use `WorkflowAgentOnStepEndCallback` instead.
+ */
+type WorkflowAgentOnStepFinishCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = WorkflowAgentOnStepEndCallback<TTools, TRuntimeContext>;
 /**
  * Infer the type of the tools of a workflow agent.
  */
-type InferWorkflowAgentTools<WORKFLOW_AGENT> = WORKFLOW_AGENT extends WorkflowAgent<infer TOOLS> ? TOOLS : never;
+type InferWorkflowAgentTools<WORKFLOW_AGENT> = WORKFLOW_AGENT extends WorkflowAgent<infer TOOLS, any> ? TOOLS : never;
 /**
  * Infer the UI message type of a workflow agent.
  */
@@ -52,41 +60,12 @@ interface OutputSpecification<OUTPUT, PARTIAL> {
  * Provider-specific options type. This is equivalent to SharedV4ProviderOptions from @ai-sdk/provider.
  */
 type ProviderOptions = SharedV4ProviderOptions;
-/**
- * Telemetry settings for observability.
- */
-interface TelemetrySettings {
-    /**
-     * Enable or disable telemetry. Defaults to true.
-     */
-    isEnabled?: boolean;
-    /**
-     * Identifier for this function. Used to group telemetry data by function.
-     */
-    functionId?: string;
-    /**
-     * Additional information to include in the telemetry data.
-     */
-    metadata?: Record<string, string | number | boolean | Array<string | number | boolean> | null | undefined>;
-    /**
-     * 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;
-    /**
-     * Custom tracer for the telemetry.
-     */
-    tracer?: unknown;
-}
+type WorkflowAgentToolsContextParameter<TTools extends ToolSet> = HasRequiredKey<InferToolSetContext<TTools>> extends true ? {
+    toolsContext: InferToolSetContext<TTools>;
+} : {
+    toolsContext?: never;
+};
+type TelemetryOptions<TRuntimeContext extends Context = Context, TTools extends ToolSet = ToolSet> = TelemetryOptions$1<TRuntimeContext, TTools>;
 /**
  * A transformation that is applied to the stream.
  */
@@ -182,7 +161,7 @@ interface GenerationSettings {
 /**
  * Information passed to the prepareStep callback.
  */
-interface PrepareStepInfo<TTools extends ToolSet = ToolSet> {
+interface PrepareStepInfo<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> {
     /**
      * The current model configuration (string or function).
      * The function should return a LanguageModelV4 instance.
@@ -195,22 +174,31 @@ interface PrepareStepInfo<TTools extends ToolSet = ToolSet> {
     /**
      * All previous steps with their results.
      */
-    steps: StepResult<TTools, any>[];
+    steps: StepResult<TTools, TRuntimeContext>[];
     /**
      * The messages that will be sent to the model.
      * This is the LanguageModelV4Prompt format used internally.
      */
     messages: LanguageModelV4Prompt;
     /**
-     * The context passed via the experimental_context setting (experimental).
+     * The runtime context that flows through the agent loop.
+     * Treat the value as immutable; return a new `runtimeContext` from
+     * `prepareStep` to update it for the current and subsequent steps.
      */
-    experimental_context: unknown;
+    runtimeContext: TRuntimeContext;
+    /**
+     * Per-tool context, keyed by tool name. Each tool receives only its own
+     * validated entry as `context` during execution.
+     * Treat the value as immutable; return a new `toolsContext` from
+     * `prepareStep` to update it for the current and subsequent steps.
+     */
+    toolsContext: InferToolSetContext<TTools>;
 }
 /**
  * Return type from the prepareStep callback.
  * All properties are optional - only return the ones you want to override.
  */
-interface PrepareStepResult extends Partial<GenerationSettings> {
+interface PrepareStepResult<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> extends Partial<GenerationSettings> {
     /**
      * Override the model for this step.
      */
@@ -234,26 +222,39 @@ interface PrepareStepResult extends Partial<GenerationSettings> {
      */
     activeTools?: string[];
     /**
-     * Context that is passed into tool execution. Experimental.
-     * Changing the context will affect the context in this step and all subsequent steps.
+     * Updated runtime context for the current and subsequent steps.
+     * Returning a value replaces the agent's runtime context.
+     */
+    runtimeContext?: TRuntimeContext;
+    /**
+     * Updated per-tool context for the current and subsequent steps.
+     * Returning a value replaces the agent's tools context.
      */
-    experimental_context?: unknown;
+    toolsContext?: InferToolSetContext<TTools>;
 }
 /**
  * Callback function called before each step in the agent loop.
  * Use this to modify settings, manage context, or implement dynamic behavior.
  */
-type PrepareStepCallback<TTools extends ToolSet = ToolSet> = (info: PrepareStepInfo<TTools>) => PrepareStepResult | Promise<PrepareStepResult>;
+type PrepareStepCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = (info: PrepareStepInfo<TTools, TRuntimeContext>) => PrepareStepResult<TTools, TRuntimeContext> | undefined | Promise<PrepareStepResult<TTools, TRuntimeContext> | undefined>;
 /**
  * Options passed to the prepareCall callback.
  */
-interface PrepareCallOptions<TTools extends ToolSet = ToolSet> extends Partial<GenerationSettings> {
+interface PrepareCallOptions<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> extends Partial<GenerationSettings> {
     model: LanguageModel;
     tools: TTools;
-    instructions?: string | SystemModelMessage | Array<SystemModelMessage>;
+    instructions?: Instructions;
     toolChoice?: ToolChoice<TTools>;
-    experimental_telemetry?: TelemetrySettings;
-    experimental_context?: unknown;
+    telemetry?: TelemetryOptions<TRuntimeContext, TTools>;
+    /**
+     * Runtime context that flows through the agent loop.
+     * Treat as immutable; return a new `runtimeContext` to update it for the call.
+     */
+    runtimeContext?: TRuntimeContext;
+    /**
+     * Per-tool context, keyed by tool name.
+     */
+    toolsContext?: InferToolSetContext<TTools>;
     messages: ModelMessage[];
 }
 /**
@@ -262,15 +263,15 @@ interface PrepareCallOptions<TTools extends ToolSet = ToolSet> extends Partial<G
  * Note: `tools` cannot be overridden via prepareCall because they are
  * bound at construction time for type safety.
  */
-type PrepareCallResult<TTools extends ToolSet = ToolSet> = Partial<Omit<PrepareCallOptions<TTools>, 'tools'>>;
+type PrepareCallResult<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = Partial<Omit<PrepareCallOptions<TTools, TRuntimeContext>, 'tools'>>;
 /**
  * Callback called once before the agent loop starts to transform call parameters.
  */
-type PrepareCallCallback<TTools extends ToolSet = ToolSet> = (options: PrepareCallOptions<TTools>) => PrepareCallResult<TTools> | Promise<PrepareCallResult<TTools>>;
+type PrepareCallCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = (options: PrepareCallOptions<TTools, TRuntimeContext>) => PrepareCallResult<TTools, TRuntimeContext> | Promise<PrepareCallResult<TTools, TRuntimeContext>>;
 /**
  * Configuration options for creating a {@link WorkflowAgent} instance.
  */
-interface WorkflowAgentOptions<TTools extends ToolSet = ToolSet> extends GenerationSettings {
+type WorkflowAgentOptions<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = GenerationSettings & WorkflowAgentToolsContextParameter<TTools> & {
     /**
      * The id of the agent.
      */
@@ -292,7 +293,7 @@ interface WorkflowAgentOptions<TTools extends ToolSet = ToolSet> extends Generat
      * Agent instructions. Can be a string, a SystemModelMessage, or an array of SystemModelMessages.
      * Supports provider-specific options (e.g., caching) when using the SystemModelMessage form.
      */
-    instructions?: string | SystemModelMessage | Array<SystemModelMessage>;
+    instructions?: Instructions;
     /**
      * Optional system prompt to guide the agent's behavior.
      * @deprecated Use `instructions` instead.
@@ -303,17 +304,23 @@ interface WorkflowAgentOptions<TTools extends ToolSet = ToolSet> extends Generat
      */
     toolChoice?: ToolChoice<TTools>;
     /**
-     * Optional telemetry configuration (experimental).
+     * Optional telemetry configuration.
      */
-    experimental_telemetry?: TelemetrySettings;
+    telemetry?: TelemetryOptions<TRuntimeContext, TTools>;
     /**
-     * Default context that is passed into tool execution for every stream call on this agent.
+     * Default runtime context for every stream call on this agent.
+     *
+     * The runtime context flows through `prepareStep`, lifecycle callbacks,
+     * and step results.
+     * Treat as immutable; return a new `runtimeContext` from `prepareStep`
+     * to update it between steps.
+     *
+     * In workflow context, keep values serializable so they can cross workflow
+     * and step boundaries.
      *
-     * Per-stream `experimental_context` values passed to `stream()` override this default.
-     * Experimental (can break in patch releases).
-     * @default undefined
+     * Per-stream `runtimeContext` values passed to `stream()` override this default.
      */
-    experimental_context?: unknown;
+    runtimeContext?: TRuntimeContext;
     /**
      * Default stop condition for the agent loop. When the condition is an array,
      * any of the conditions can be met to stop the generation.
@@ -327,7 +334,7 @@ interface WorkflowAgentOptions<TTools extends ToolSet = ToolSet> extends Generat
      *
      * Per-stream `activeTools` values passed to `stream()` override this default.
      */
-    activeTools?: Array<keyof NoInfer<TTools>>;
+    activeTools?: ActiveTools<NoInfer<TTools>>;
     /**
      * Default output specification for structured outputs.
      * Use `Output.object({ schema })` for structured output or `Output.text()` for text output.
@@ -354,46 +361,58 @@ interface WorkflowAgentOptions<TTools extends ToolSet = ToolSet> extends Generat
      *
      * Per-stream `prepareStep` values passed to `stream()` override this default.
      */
-    prepareStep?: PrepareStepCallback<TTools>;
+    prepareStep?: PrepareStepCallback<TTools, TRuntimeContext>;
     /**
      * Callback function to be called after each step completes.
      */
-    onStepFinish?: WorkflowAgentOnStepFinishCallback<ToolSet>;
+    onStepEnd?: WorkflowAgentOnStepEndCallback<TTools, TRuntimeContext>;
+    /**
+     * Callback function to be called after each step completes.
+     *
+     * @deprecated Use `onStepEnd` instead.
+     */
+    onStepFinish?: WorkflowAgentOnStepFinishCallback<TTools, TRuntimeContext>;
     /**
      * Callback that is called when the LLM response and all request tool executions are finished.
      */
-    onFinish?: WorkflowAgentOnFinishCallback<ToolSet>;
+    onEnd?: WorkflowAgentOnEndCallback<TTools, TRuntimeContext>;
+    /**
+     * Callback that is called when the LLM response and all request tool executions are finished.
+     *
+     * @deprecated Use `onEnd` instead.
+     */
+    onFinish?: WorkflowAgentOnFinishCallback<TTools, TRuntimeContext>;
     /**
      * Callback called when the agent starts streaming, before any LLM calls.
      */
-    experimental_onStart?: WorkflowAgentOnStartCallback;
+    experimental_onStart?: WorkflowAgentOnStartCallback<TTools, TRuntimeContext>;
     /**
      * Callback called before each step (LLM call) begins.
      */
-    experimental_onStepStart?: WorkflowAgentOnStepStartCallback;
+    experimental_onStepStart?: WorkflowAgentOnStepStartCallback<TTools, TRuntimeContext>;
     /**
      * Callback called before a tool's execute function runs.
      */
-    experimental_onToolCallStart?: WorkflowAgentOnToolCallStartCallback;
+    onToolExecutionStart?: WorkflowAgentOnToolExecutionStartCallback<TTools>;
     /**
      * Callback called after a tool execution completes.
      */
-    experimental_onToolCallFinish?: WorkflowAgentOnToolCallFinishCallback;
+    onToolExecutionEnd?: WorkflowAgentOnToolExecutionEndCallback<TTools>;
     /**
      * Prepare the parameters for the stream call.
      * Called once before the agent loop starts. Use this to transform
      * model, tools, instructions, or other settings based on runtime context.
      */
-    prepareCall?: PrepareCallCallback<TTools>;
-}
+    prepareCall?: PrepareCallCallback<TTools, TRuntimeContext>;
+};
 /**
  * Callback that is called when the LLM response and all request tool executions are finished.
  */
-type WorkflowAgentOnFinishCallback<TTools extends ToolSet = ToolSet, OUTPUT = never> = (event: {
+type WorkflowAgentOnEndCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context, OUTPUT = never> = (event: {
     /**
      * Details for all steps.
      */
-    readonly steps: StepResult<TTools, any>[];
+    readonly steps: StepResult<TTools, TRuntimeContext>[];
     /**
      * The final messages including all tool calls and results.
      */
@@ -406,20 +425,34 @@ type WorkflowAgentOnFinishCallback<TTools extends ToolSet = ToolSet, OUTPUT = ne
      * The finish reason from the last step.
      */
     readonly finishReason: FinishReason;
+    /**
+     * The total token usage across all steps.
+     */
+    readonly usage: LanguageModelUsage;
     /**
      * The total token usage across all steps.
      */
     readonly totalUsage: LanguageModelUsage;
     /**
-     * Context that is passed into tool execution.
+     * The runtime context at the end of the agent loop.
      */
-    readonly experimental_context: unknown;
+    readonly runtimeContext: TRuntimeContext;
+    /**
+     * The per-tool context at the end of the agent loop.
+     */
+    readonly toolsContext: InferToolSetContext<TTools>;
     /**
      * The generated structured output. It uses the `output` specification.
      * Only available when `output` is specified.
      */
     readonly output: OUTPUT;
 }) => PromiseLike<void> | void;
+/**
+ * Callback that is called when the LLM response and all request tool executions are finished.
+ *
+ * @deprecated Use `WorkflowAgentOnEndCallback` instead.
+ */
+type WorkflowAgentOnFinishCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context, OUTPUT = never> = WorkflowAgentOnEndCallback<TTools, TRuntimeContext, OUTPUT>;
 /**
  * Callback that is invoked when an error occurs during streaming.
  */
@@ -438,16 +471,20 @@ type WorkflowAgentOnAbortCallback<TTools extends ToolSet = ToolSet> = (event: {
 /**
  * Callback that is called when the agent starts streaming, before any LLM calls.
  */
-type WorkflowAgentOnStartCallback = (event: {
+type WorkflowAgentOnStartCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = (event: {
     /** The model being used */
     readonly model: LanguageModel;
     /** The messages being sent */
     readonly messages: ModelMessage[];
+    /** Shared runtime context for this agent loop */
+    readonly runtimeContext: TRuntimeContext;
+    /** Per-tool context map for this agent loop */
+    readonly toolsContext: InferToolSetContext<TTools>;
 }) => PromiseLike<void> | void;
 /**
  * Callback that is called before each step (LLM call) begins.
  */
-type WorkflowAgentOnStepStartCallback<TTools extends ToolSet = ToolSet> = (event: {
+type WorkflowAgentOnStepStartCallback<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> = (event: {
     /** The current step number (0-based) */
     readonly stepNumber: number;
     /** The model being used for this step */
@@ -455,29 +492,41 @@ type WorkflowAgentOnStepStartCallback<TTools extends ToolSet = ToolSet> = (event
     /** The messages being sent for this step */
     readonly messages: ModelMessage[];
     /** Results from all previously finished steps */
-    readonly steps: ReadonlyArray<StepResult<TTools, any>>;
+    readonly steps: ReadonlyArray<StepResult<TTools, TRuntimeContext>>;
+    /** Shared runtime context for this step */
+    readonly runtimeContext: TRuntimeContext;
+    /** Per-tool context map for this step */
+    readonly toolsContext: InferToolSetContext<TTools>;
 }) => PromiseLike<void> | void;
 /**
  * Callback that is called before a tool's execute function runs.
  */
-type WorkflowAgentOnToolCallStartCallback = (event: {
+type WorkflowAgentOnToolExecutionStartCallback<TTools extends ToolSet = ToolSet> = (event: {
     /** The tool call being executed */
     readonly toolCall: ToolCall;
     /** The current step number (0-based) */
     readonly stepNumber: number;
+    /** Messages sent to the language model for the step that produced the call */
+    readonly messages: ModelMessage[];
+    /** Tool-specific context passed to the tool */
+    readonly toolContext: InferToolSetContext<TTools>[keyof InferToolSetContext<TTools>] | undefined;
 }) => PromiseLike<void> | void;
 /**
  * Callback that is called after a tool execution completes.
  * Uses a discriminated union pattern: check `success` to determine
  * whether `output` or `error` is available.
  */
-type WorkflowAgentOnToolCallFinishCallback = (event: {
+type WorkflowAgentOnToolExecutionEndCallback<TTools extends ToolSet = ToolSet> = (event: {
     /** The tool call that was executed */
     readonly toolCall: ToolCall;
     /** The current step number (0-based) */
     readonly stepNumber: number;
     /** Execution time in milliseconds */
     readonly durationMs: number;
+    /** Messages sent to the language model for the step that produced the call */
+    readonly messages: ModelMessage[];
+    /** Tool-specific context passed to the tool */
+    readonly toolContext: InferToolSetContext<TTools>[keyof InferToolSetContext<TTools>] | undefined;
     /** Whether the tool call succeeded */
     readonly success: true;
     /** The tool result */
@@ -490,6 +539,10 @@ type WorkflowAgentOnToolCallFinishCallback = (event: {
     readonly stepNumber: number;
     /** Execution time in milliseconds */
     readonly durationMs: number;
+    /** Messages sent to the language model for the step that produced the call */
+    readonly messages: ModelMessage[];
+    /** Tool-specific context passed to the tool */
+    readonly toolContext: InferToolSetContext<TTools>[keyof InferToolSetContext<TTools>] | undefined;
     /** Whether the tool call succeeded */
     readonly success: false;
     /** The error that occurred */
@@ -499,7 +552,7 @@ type WorkflowAgentOnToolCallFinishCallback = (event: {
 /**
  * Options for the {@link WorkflowAgent.stream} method.
  */
-type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never, PARTIAL_OUTPUT = never> = Partial<GenerationSettings> & ({
+type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context, OUTPUT = never, PARTIAL_OUTPUT = never> = Partial<GenerationSettings> & ({
     /**
      * A prompt. It can be either a text prompt or a list of messages.
      *
@@ -558,12 +611,6 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
      * When the condition is an array, any of the conditions can be met to stop the generation.
      */
     stopWhen?: StopCondition<NoInfer<ToolSet>, any> | Array<StopCondition<NoInfer<ToolSet>, any>>;
-    /**
-     * Maximum number of sequential LLM calls (steps), e.g. when you use tool calls.
-     * A maximum number can be set to prevent infinite loops in the case of misconfigured tools.
-     * By default, it's unlimited (the agent loops until completion).
-     */
-    maxSteps?: number;
     /**
      * The tool choice strategy. Default: 'auto'.
      * Overrides the toolChoice from the constructor if provided.
@@ -573,17 +620,34 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
      * Limits the tools that are available for the model to call without
      * changing the tool call and result types in the result.
      */
-    activeTools?: Array<keyof NoInfer<TTools>>;
+    activeTools?: ActiveTools<NoInfer<TTools>>;
     /**
-     * Optional telemetry configuration (experimental).
+     * Optional telemetry configuration.
      */
-    experimental_telemetry?: TelemetrySettings;
+    telemetry?: TelemetryOptions<TRuntimeContext, TTools>;
     /**
-     * Context that is passed into tool execution.
-     * Experimental (can break in patch releases).
-     * @default undefined
+     * Runtime context that flows through the agent loop.
+     *
+     * Treat as immutable; return a new `runtimeContext` from `prepareStep`
+     * to update it between steps.
+     *
+     * In workflow context, keep values serializable so they can cross workflow
+     * and step boundaries.
+     *
+     * Overrides the constructor-level `runtimeContext` if provided.
+     */
+    runtimeContext?: TRuntimeContext;
+    /**
+     * Per-tool context, keyed by tool name. Each tool receives only its own
+     * validated entry as `context` during execution. Tools that declare a
+     * `contextSchema` validate their entry against the schema.
+     *
+     * In workflow context, keep values serializable so they can cross workflow
+     * and step boundaries.
+     *
+     * Overrides the constructor-level `toolsContext` if provided.
      */
-    experimental_context?: unknown;
+    toolsContext?: InferToolSetContext<TTools>;
     /**
      * Optional specification for parsing structured outputs from the LLM response.
      * Use `Output.object({ schema })` for structured output or `Output.text()` for text output.
@@ -633,7 +697,13 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
     /**
      * Callback function to be called after each step completes.
      */
-    onStepFinish?: WorkflowAgentOnStepFinishCallback<TTools>;
+    onStepEnd?: WorkflowAgentOnStepEndCallback<TTools, TRuntimeContext>;
+    /**
+     * Callback function to be called after each step completes.
+     *
+     * @deprecated Use `onStepEnd` instead.
+     */
+    onStepFinish?: WorkflowAgentOnStepFinishCallback<TTools, TRuntimeContext>;
     /**
      * Callback that is invoked when an error occurs during streaming.
      * You can use it to log errors.
@@ -643,7 +713,14 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
      * Callback that is called when the LLM response and all request tool executions
      * (for tools that have an `execute` function) are finished.
      */
-    onFinish?: WorkflowAgentOnFinishCallback<TTools, OUTPUT>;
+    onEnd?: WorkflowAgentOnEndCallback<TTools, TRuntimeContext, OUTPUT>;
+    /**
+     * Callback that is called when the LLM response and all request tool executions
+     * (for tools that have an `execute` function) are finished.
+     *
+     * @deprecated Use `onEnd` instead.
+     */
+    onFinish?: WorkflowAgentOnFinishCallback<TTools, TRuntimeContext, OUTPUT>;
     /**
      * Callback that is called when the operation is aborted.
      */
@@ -651,19 +728,19 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
     /**
      * Callback called when the agent starts streaming, before any LLM calls.
      */
-    experimental_onStart?: WorkflowAgentOnStartCallback;
+    experimental_onStart?: WorkflowAgentOnStartCallback<TTools, TRuntimeContext>;
     /**
      * Callback called before each step (LLM call) begins.
      */
-    experimental_onStepStart?: WorkflowAgentOnStepStartCallback;
+    experimental_onStepStart?: WorkflowAgentOnStepStartCallback<TTools, TRuntimeContext>;
     /**
      * Callback called before a tool's execute function runs.
      */
-    experimental_onToolCallStart?: WorkflowAgentOnToolCallStartCallback;
+    onToolExecutionStart?: WorkflowAgentOnToolExecutionStartCallback<TTools>;
     /**
      * Callback called after a tool execution completes.
      */
-    experimental_onToolCallFinish?: WorkflowAgentOnToolCallFinishCallback;
+    onToolExecutionEnd?: WorkflowAgentOnToolExecutionEndCallback<TTools>;
     /**
      * Callback function called before each step in the agent loop.
      * Use this to modify settings, manage context, or inject messages dynamically.
@@ -682,7 +759,7 @@ type WorkflowAgentStreamOptions<TTools extends ToolSet = ToolSet, OUTPUT = never
      * }
      * ```
      */
-    prepareStep?: PrepareStepCallback<TTools>;
+    prepareStep?: PrepareStepCallback<TTools, TRuntimeContext>;
     /**
      * Timeout in milliseconds for the stream operation.
      * When specified, creates an AbortSignal that will abort the operation after the given time.
@@ -798,7 +875,7 @@ interface WorkflowAgentStreamResult<TTools extends ToolSet = ToolSet, OUTPUT = n
  * });
  * ```
  */
-declare class WorkflowAgent<TBaseTools extends ToolSet = ToolSet> {
+declare class WorkflowAgent<TBaseTools extends ToolSet = ToolSet, TRuntimeContext extends Context = Context> {
     /**
      * The id of the agent.
      */
@@ -812,23 +889,24 @@ declare class WorkflowAgent<TBaseTools extends ToolSet = ToolSet> {
     private generationSettings;
     private toolChoice?;
     private telemetry?;
-    private experimentalContext;
+    private runtimeContext?;
+    private toolsContext?;
     private stopWhen?;
     private activeTools?;
     private output?;
     private experimentalRepairToolCall?;
     private experimentalDownload?;
     private prepareStep?;
-    private constructorOnStepFinish?;
-    private constructorOnFinish?;
+    private constructorOnStepEnd?;
+    private constructorOnEnd?;
     private constructorOnStart?;
     private constructorOnStepStart?;
-    private constructorOnToolCallStart?;
-    private constructorOnToolCallFinish?;
+    private constructorOnToolExecutionStart?;
+    private constructorOnToolExecutionEnd?;
     private prepareCall?;
-    constructor(options: WorkflowAgentOptions<TBaseTools>);
+    constructor(options: WorkflowAgentOptions<TBaseTools, TRuntimeContext>);
     generate(): void;
-    stream<TTools extends TBaseTools = TBaseTools, OUTPUT = never, PARTIAL_OUTPUT = never>(options: WorkflowAgentStreamOptions<TTools, OUTPUT, PARTIAL_OUTPUT>): Promise<WorkflowAgentStreamResult<TTools, OUTPUT>>;
+    stream<TTools extends TBaseTools = TBaseTools, OUTPUT = never, PARTIAL_OUTPUT = never>(options: WorkflowAgentStreamOptions<TTools, TRuntimeContext, OUTPUT, PARTIAL_OUTPUT>): Promise<WorkflowAgentStreamResult<TTools, OUTPUT>>;
 }
 
 /**
@@ -997,4 +1075,4 @@ declare class WorkflowChatTransport<UI_MESSAGE extends UIMessage> implements Cha
     private onFinish;
 }
 
-export { type CompatibleLanguageModel, type DownloadFunction, type GenerationSettings, type InferWorkflowAgentTools, type InferWorkflowAgentUIMessage, type OutputSpecification, type PrepareCallCallback, type PrepareCallOptions, type PrepareCallResult, type PrepareStepCallback, type PrepareStepInfo, type PrepareStepResult, type ProviderOptions, type ReconnectToStreamOptions, type SendMessagesOptions, type StreamTextTransform, type TelemetrySettings, WorkflowAgent, type WorkflowAgentOnAbortCallback, type WorkflowAgentOnErrorCallback, type WorkflowAgentOnFinishCallback, type WorkflowAgentOnStartCallback, type WorkflowAgentOnStepFinishCallback, type WorkflowAgentOnStepStartCallback, type WorkflowAgentOnToolCallFinishCallback, type WorkflowAgentOnToolCallStartCallback, type WorkflowAgentOptions, type WorkflowAgentStreamOptions, type WorkflowAgentStreamResult, WorkflowChatTransport, type WorkflowChatTransportOptions, createModelCallToUIChunkTransform, toUIMessageChunk };
+export { type CompatibleLanguageModel, type DownloadFunction, type GenerationSettings, type InferWorkflowAgentTools, type InferWorkflowAgentUIMessage, type OutputSpecification, type PrepareCallCallback, type PrepareCallOptions, type PrepareCallResult, type PrepareStepCallback, type PrepareStepInfo, type PrepareStepResult, type ProviderOptions, type ReconnectToStreamOptions, type SendMessagesOptions, type StreamTextTransform, type TelemetryOptions, WorkflowAgent, type WorkflowAgentOnAbortCallback, type WorkflowAgentOnEndCallback, type WorkflowAgentOnErrorCallback, type WorkflowAgentOnFinishCallback, type WorkflowAgentOnStartCallback, type WorkflowAgentOnStepEndCallback, type WorkflowAgentOnStepFinishCallback, type WorkflowAgentOnStepStartCallback, type WorkflowAgentOnToolExecutionEndCallback, type WorkflowAgentOnToolExecutionStartCallback, type WorkflowAgentOptions, type WorkflowAgentStreamOptions, type WorkflowAgentStreamResult, WorkflowChatTransport, type WorkflowChatTransportOptions, createModelCallToUIChunkTransform, toUIMessageChunk };