langchain 1.4.1 → 1.4.4

package/dist/agents/middleware/types.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"types.cjs","names":[],"sources":["../../../src/agents/middleware/types.ts"],"sourcesContent":["/* oxlint-disable @typescript-eslint/no-explicit-any */\nimport type {\n  InteropZodObject,\n  InteropZodDefault,\n  InteropZodOptional,\n  InferInteropZodInput,\n  InferInteropZodOutput,\n} from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type {\n  AnnotationRoot,\n  StateSchema,\n  InferStateSchemaValue,\n  InferStateSchemaUpdate,\n  StateDefinitionInit,\n} from \"@langchain/langgraph\";\nimport type {\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n} from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command } from \"@langchain/langgraph\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\n\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\n\ntype PromiseOrValue<T> = T | Promise<T>;\n\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n\n/**\n * Type bag that encapsulates all middleware type parameters.\n *\n * This interface bundles all the generic type parameters used throughout the middleware system\n * into a single configuration object. This pattern simplifies type signatures and makes\n * it easier to add new type parameters without changing multiple function signatures.\n *\n * @typeParam TSchema - The middleware state schema type. Can be a `StateDefinitionInit`\n *   (including `InteropZodObject`, `StateSchema`, or `AnnotationRoot`) or `undefined`.\n *\n * @typeParam TContextSchema - The middleware context schema type. Can be an `InteropZodObject`,\n *   `InteropZodDefault`, `InteropZodOptional`, or `undefined`.\n *\n * @typeParam TFullContext - The full context type available to middleware hooks.\n *\n * @typeParam TTools - The tools array type registered by the middleware.\n *\n * @example\n * ```typescript\n * // Define a type configuration\n * type MyMiddlewareTypes = MiddlewareTypeConfig<\n *   typeof myStateSchema,\n *   typeof myContextSchema,\n *   MyContextType,\n *   typeof myTools\n * >;\n * ```\n */\nexport interface MiddlewareTypeConfig<\n  TSchema extends StateDefinitionInit | undefined =\n    | StateDefinitionInit\n    | undefined,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined =\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined,\n  TFullContext = any,\n  TTools extends readonly (ClientTool | ServerTool)[] = readonly (\n    | ClientTool\n    | ServerTool\n  )[],\n> {\n  /** The middleware state schema type */\n  Schema: TSchema;\n  /** The middleware context schema type */\n  ContextSchema: TContextSchema;\n  /** The full context type */\n  FullContext: TFullContext;\n  /** The tools array type */\n  Tools: TTools;\n}\n\n/**\n * Default type configuration for middleware.\n * Used when no explicit type parameters are provided.\n */\nexport type DefaultMiddlewareTypeConfig = MiddlewareTypeConfig;\n\nexport type InferSchemaValueType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodOutput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaValue<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type InferSchemaUpdateType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaInput<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type NormalizedSchemaInput<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaValueType<TSchema>;\n\nexport type NormalizedSchemaUpdate<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaUpdateType<TSchema>;\n\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> =\n  | (TState & {\n      jumpTo?: JumpToTarget;\n    })\n  | void;\n\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<\n  TState extends Record<string, unknown> = Record<string, unknown>,\n  TContext = unknown,\n> {\n  /**\n   * The tool call to be executed\n   */\n  toolCall: ToolCall;\n  /**\n   * The BaseTool instance being invoked.\n   * Provides access to tool metadata like name, description, schema, etc.\n   *\n   * This will be `undefined` for dynamically registered tools that aren't\n   * declared upfront when creating the agent. In such cases, middleware\n   * should provide the tool implementation by spreading the request with\n   * the tool property.\n   *\n   * @example Dynamic tool handling\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   if (request.toolCall.name === \"dynamic_tool\" && !request.tool) {\n   *     // Provide the tool implementation for dynamically registered tools\n   *     return handler({ ...request, tool: myDynamicTool });\n   *   }\n   *   return handler(request);\n   * }\n   * ```\n   */\n  tool: ClientTool | ServerTool | undefined;\n  /**\n   * The current agent state (includes both middleware state and built-in state).\n   */\n  state: TState & AgentBuiltInState;\n  /**\n   * The runtime context containing metadata, signal, writer, interrupt, etc.\n   */\n  runtime: Runtime<TContext>;\n}\n\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<\n  TSchema extends Record<string, unknown> = AgentBuiltInState,\n  TContext = unknown,\n> = (\n  request: ToolCallRequest<TSchema, TContext>\n) => PromiseOrValue<ToolMessage | Command>;\n\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>,\n  handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>\n) => PromiseOrValue<ToolMessage | Command>;\n\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: Omit<\n    ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,\n    /**\n     * allow to reset the system prompt or system message\n     */\n    \"systemPrompt\" | \"systemMessage\"\n  > & { systemPrompt?: string; systemMessage?: SystemMessage }\n) => PromiseOrValue<AIMessage>;\n\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,\n  handler: WrapModelCallHandler<TSchema, TContext>\n) => PromiseOrValue<AIMessage | Command>;\n\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeAgentHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | BeforeAgentHandler<TSchema, TContext>\n  | {\n      hook: BeforeAgentHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeModelHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | BeforeModelHandler<TSchema, TContext>\n  | {\n      hook: BeforeModelHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterModelHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | AfterModelHandler<TSchema, TContext>\n  | {\n      hook: AfterModelHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterAgentHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | AfterAgentHandler<TSchema, TContext>\n  | {\n      hook: AfterAgentHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Unique symbol used to brand middleware instances.\n * This prevents functions from being accidentally assignable to AgentMiddleware\n * since functions have a 'name' property that would otherwise make them structurally compatible.\n */\nexport const MIDDLEWARE_BRAND: symbol = Symbol.for(\"AgentMiddleware\");\n\n/**\n * Base middleware interface.\n *\n * @typeParam TSchema - The middleware state schema type\n * @typeParam TContextSchema - The middleware context schema type\n * @typeParam TFullContext - The full context type available to hooks\n * @typeParam TTools - The tools array type registered by the middleware\n *\n * @example\n * ```typescript\n * const middleware = createMiddleware({\n *   name: \"myMiddleware\",\n *   stateSchema: z.object({ count: z.number() }),\n *   tools: [myTool],\n * });\n * ```\n */\nexport interface AgentMiddleware<\n  TSchema extends StateDefinitionInit | undefined = any,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined = any,\n  TFullContext = any,\n  TTools extends readonly (ClientTool | ServerTool)[] = readonly (\n    | ClientTool\n    | ServerTool\n  )[],\n> {\n  /**\n   * Brand property to distinguish middleware instances from plain objects or functions.\n   * This is required and prevents accidental assignment of functions to middleware arrays.\n   */\n  readonly [MIDDLEWARE_BRAND]: true;\n\n  /**\n   * Type marker for extracting the MiddlewareTypeConfig from a middleware instance.\n   * This is a phantom property used only for type inference.\n   * @internal\n   */\n  readonly \"~middlewareTypes\"?: MiddlewareTypeConfig<\n    TSchema,\n    TContextSchema,\n    TFullContext,\n    TTools\n  >;\n\n  /**\n   * The name of the middleware.\n   */\n  name: string;\n\n  /**\n   * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:\n   * - A Zod object (InteropZodObject)\n   * - A StateSchema from LangGraph (supports ReducedValue, UntrackedValue)\n   * - An AnnotationRoot\n   * - Undefined\n   */\n  stateSchema?: TSchema;\n\n  /**\n   * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:\n   * - A Zod object\n   * - A Zod optional object\n   * - A Zod default object\n   * - Undefined\n   */\n  contextSchema?: TContextSchema;\n\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: TTools;\n  /**\n   * Wraps tool execution with custom logic. This allows you to:\n   * - Modify tool call parameters before execution\n   * - Handle errors and retry with different parameters\n   * - Post-process tool results\n   * - Implement caching, logging, authentication, or other cross-cutting concerns\n   * - Return Command objects for advanced control flow\n   *\n   * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n   * along with a handler function to execute the actual tool.\n   *\n   * @param request - The tool call request containing toolCall, state, and runtime.\n   * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n   * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n   *\n   * @example\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   console.log(`Calling tool: ${request.tool.name}`);\n   *   console.log(`Tool description: ${request.tool.description}`);\n   *\n   *   try {\n   *     // Execute the tool\n   *     const result = await handler(request);\n   *     console.log(`Tool ${request.tool.name} succeeded`);\n   *     return result;\n   *   } catch (error) {\n   *     console.error(`Tool ${request.tool.name} failed:`, error);\n   *     // Could return a custom error message or retry\n   *     throw error;\n   *   }\n   * }\n   * ```\n   *\n   * @example Authentication\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   // Check if user is authorized for this tool\n   *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n   *     return new ToolMessage({\n   *       content: \"Unauthorized to call this tool\",\n   *       tool_call_id: request.toolCall.id,\n   *     });\n   *   }\n   *   return handler(request);\n   * }\n   * ```\n   *\n   * @example Caching\n   * ```ts\n   * const cache = new Map();\n   * wrapToolCall: async (request, handler) => {\n   *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n   *   if (cache.has(cacheKey)) {\n   *     return cache.get(cacheKey);\n   *   }\n   *   const result = await handler(request);\n   *   cache.set(cacheKey, result);\n   *   return result;\n   * }\n   * ```\n   */\n  wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n\n  /**\n   * Wraps the model invocation with custom logic. This allows you to:\n   * - Modify the request before calling the model\n   * - Handle errors and retry with different parameters\n   * - Post-process the response\n   * - Implement custom caching, logging, or other cross-cutting concerns\n   *\n   * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n   * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n   * @returns The response from the model (or a modified version).\n   *\n   * @example\n   * ```ts\n   * wrapModelCall: async (request, handler) => {\n   *   // Modify request before calling\n   *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n   *\n   *   try {\n   *     // Call the model\n   *     return await handler(modifiedRequest);\n   *   } catch (error) {\n   *     // Handle errors and retry with fallback\n   *     const fallbackRequest = { ...request, model: fallbackModel };\n   *     return await handler(fallbackRequest);\n   *   }\n   * }\n   * ```\n   */\n  wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n   * It allows to modify the state of the agent before any model calls or tool executions.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n   * It allows to modify the state of the agent.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n   * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  afterModel?: AfterModelHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n   * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n  [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\n\n/**\n * Helper type to resolve a MiddlewareTypeConfig from either:\n * - A MiddlewareTypeConfig directly\n * - An AgentMiddleware instance (using `typeof middleware`)\n */\nexport type ResolveMiddlewareTypeConfig<T> = T extends {\n  \"~middlewareTypes\"?: infer Types;\n}\n  ? Types extends MiddlewareTypeConfig\n    ? Types\n    : never\n  : T extends MiddlewareTypeConfig\n    ? T\n    : never;\n\n/**\n * Helper type to extract any property from a MiddlewareTypeConfig or AgentMiddleware.\n *\n * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from\n * @typeParam K - The property key to extract (\"Schema\" | \"ContextSchema\" | \"FullContext\" | \"Tools\")\n */\nexport type InferMiddlewareType<\n  T,\n  K extends keyof MiddlewareTypeConfig,\n> = ResolveMiddlewareTypeConfig<T>[K];\n\n/**\n * Shorthand helper to extract the Schema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareSchema<T> = InferMiddlewareType<T, \"Schema\">;\n\n/**\n * Shorthand helper to extract the ContextSchema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareContextSchema<T> = InferMiddlewareType<\n  T,\n  \"ContextSchema\"\n>;\n\n/**\n * Shorthand helper to extract the FullContext type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareFullContext<T> = InferMiddlewareType<\n  T,\n  \"FullContext\"\n>;\n\n/**\n * Shorthand helper to extract the Tools type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareToolsFromConfig<T> = InferMiddlewareType<T, \"Tools\">;\n\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> =\n  T extends AnyAnnotationRoot\n    ? ToAnnotationRoot<T>[\"State\"]\n    : T extends InteropZodObject\n      ? InferInteropZodInput<T>\n      : {};\n\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer TSchema, any, any, any>\n    ? TSchema extends StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaValue<TFields>>\n      : TSchema extends InteropZodObject\n        ? FilterPrivateProps<InferInteropZodOutput<TSchema>>\n        : TSchema extends StateDefinitionInit\n          ? FilterPrivateProps<InferSchemaValue<TSchema>>\n          : {}\n    : {};\n\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer TSchema, any, any, any>\n    ? TSchema extends StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaUpdate<TFields>>\n      : TSchema extends InteropZodObject\n        ? FilterPrivateProps<InferInteropZodInput<TSchema>>\n        : TSchema extends StateDefinitionInit\n          ? FilterPrivateProps<InferSchemaInput<TSchema>>\n          : {}\n    : {};\n\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T extends readonly AgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AgentMiddleware[]\n          ? InferMiddlewareState<First> & InferMiddlewareStates<Rest>\n          : InferMiddlewareState<First>\n        : {}\n      : {};\n\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AgentMiddleware[]\n          ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest>\n          : InferMiddlewareInputState<First>\n        : {}\n      : {};\n\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AgentMiddleware[]> =\n  InferMiddlewareStates<T> & AgentBuiltInState;\n\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AgentMiddleware[]> =\n  InferMiddlewareInputStates<T> & AgentBuiltInState;\n\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> =\n  T extends AgentMiddleware<any, infer TContextSchema, any, any>\n    ? TContextSchema extends InteropZodObject\n      ? InferInteropZodInput<TContextSchema>\n      : {}\n    : {};\n\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> =\n  T extends AgentMiddleware<any, infer TContextSchema, any, any>\n    ? TContextSchema extends InteropZodOptional<infer Inner>\n      ? InferInteropZodInput<Inner> | undefined\n      : TContextSchema extends InteropZodObject\n        ? InferInteropZodInput<TContextSchema>\n        : {}\n    : {};\n\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AgentMiddleware[]\n          ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest>\n          : InferMiddlewareContext<First>\n        : {}\n      : {};\n\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined]\n  ? [B] extends [undefined]\n    ? undefined\n    : B | undefined\n  : [B] extends [undefined]\n    ? A | undefined\n    : [A] extends [B]\n      ? A\n      : [B] extends [A]\n        ? B\n        : A & B;\n\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AgentMiddleware[]\n          ? MergeContextTypes<\n              InferMiddlewareContextInput<First>,\n              InferMiddlewareContextInputs<Rest>\n            >\n          : InferMiddlewareContextInput<First>\n        : {}\n      : {};\n\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject,\n> = ContextSchema extends InteropZodObject\n  ? InferInteropZodInput<ContextSchema>\n  : ContextSchema extends AnyAnnotationRoot\n    ? ToAnnotationRoot<ContextSchema>[\"State\"]\n    : {};\n\nexport type ToAnnotationRoot<A extends StateDefinitionInit> =\n  A extends AnyAnnotationRoot\n    ? A\n    : A extends InteropZodObject\n      ? InteropZodToStateDefinition<A>\n      : never;\n\nexport type InferSchemaValue<A extends StateDefinitionInit | undefined> =\n  A extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodOutput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"State\"]\n        : {};\n\nexport type InferSchemaInput<A extends StateDefinitionInit | undefined> =\n  A extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodInput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"Update\"]\n        : {};\n"],"mappings":";;;;;;AAsWA,MAAa,mBAA2B,OAAO,IAAI,kBAAkB"}
+{"version":3,"file":"types.cjs","names":[],"sources":["../../../src/agents/middleware/types.ts"],"sourcesContent":["/* oxlint-disable @typescript-eslint/no-explicit-any */\nimport type {\n  InteropZodObject,\n  InteropZodDefault,\n  InteropZodOptional,\n  InferInteropZodInput,\n  InferInteropZodOutput,\n} from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type {\n  AnnotationRoot,\n  StateSchema,\n  InferStateSchemaValue,\n  InferStateSchemaUpdate,\n  StateDefinitionInit,\n} from \"@langchain/langgraph\";\nimport type {\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n} from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command, StreamTransformer } from \"@langchain/langgraph\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\n\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\n\ntype PromiseOrValue<T> = T | Promise<T>;\n\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n\n/**\n * Type bag that encapsulates all middleware type parameters.\n *\n * This interface bundles all the generic type parameters used throughout the middleware system\n * into a single configuration object. This pattern simplifies type signatures and makes\n * it easier to add new type parameters without changing multiple function signatures.\n *\n * @typeParam TSchema - The middleware state schema type. Can be a `StateDefinitionInit`\n *   (including `InteropZodObject`, `StateSchema`, or `AnnotationRoot`) or `undefined`.\n *\n * @typeParam TContextSchema - The middleware context schema type. Can be an `InteropZodObject`,\n *   `InteropZodDefault`, `InteropZodOptional`, or `undefined`.\n *\n * @typeParam TFullContext - The full context type available to middleware hooks.\n *\n * @typeParam TTools - The tools array type registered by the middleware.\n *\n * @typeParam TStreamTransformers - The stream transformer factories registered\n *   by the middleware.\n *\n * @example\n * ```typescript\n * // Define a type configuration\n * type MyMiddlewareTypes = MiddlewareTypeConfig<\n *   typeof myStateSchema,\n *   typeof myContextSchema,\n *   MyContextType,\n *   typeof myTools\n * >;\n * ```\n */\nexport interface MiddlewareTypeConfig<\n  TSchema extends StateDefinitionInit | undefined =\n    | StateDefinitionInit\n    | undefined,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined =\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined,\n  TFullContext = any,\n  TTools extends readonly (ClientTool | ServerTool)[] = readonly (\n    | ClientTool\n    | ServerTool\n  )[],\n  TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> =\n    ReadonlyArray<() => StreamTransformer<any>>,\n> {\n  /** The middleware state schema type */\n  Schema: TSchema;\n  /** The middleware context schema type */\n  ContextSchema: TContextSchema;\n  /** The full context type */\n  FullContext: TFullContext;\n  /** The tools array type */\n  Tools: TTools;\n  /** The stream transformer factories registered by the middleware */\n  StreamTransformers: TStreamTransformers;\n}\n\n/**\n * Default type configuration for middleware.\n * Used when no explicit type parameters are provided.\n */\nexport type DefaultMiddlewareTypeConfig = MiddlewareTypeConfig;\n\nexport type InferSchemaValueType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodOutput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaValue<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type InferSchemaUpdateType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaInput<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type NormalizedSchemaInput<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaValueType<TSchema>;\n\nexport type NormalizedSchemaUpdate<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaUpdateType<TSchema>;\n\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> =\n  | (TState & {\n      jumpTo?: JumpToTarget;\n    })\n  | void;\n\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<\n  TState extends Record<string, unknown> = Record<string, unknown>,\n  TContext = unknown,\n> {\n  /**\n   * The tool call to be executed\n   */\n  toolCall: ToolCall;\n  /**\n   * The BaseTool instance being invoked.\n   * Provides access to tool metadata like name, description, schema, etc.\n   *\n   * This will be `undefined` for dynamically registered tools that aren't\n   * declared upfront when creating the agent. In such cases, middleware\n   * should provide the tool implementation by spreading the request with\n   * the tool property.\n   *\n   * @example Dynamic tool handling\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   if (request.toolCall.name === \"dynamic_tool\" && !request.tool) {\n   *     // Provide the tool implementation for dynamically registered tools\n   *     return handler({ ...request, tool: myDynamicTool });\n   *   }\n   *   return handler(request);\n   * }\n   * ```\n   */\n  tool: ClientTool | ServerTool | undefined;\n  /**\n   * The current agent state (includes both middleware state and built-in state).\n   */\n  state: TState & AgentBuiltInState;\n  /**\n   * The runtime context containing metadata, signal, writer, interrupt, etc.\n   */\n  runtime: Runtime<TContext>;\n}\n\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<\n  TSchema extends Record<string, unknown> = AgentBuiltInState,\n  TContext = unknown,\n> = (\n  request: ToolCallRequest<TSchema, TContext>\n) => PromiseOrValue<ToolMessage | Command>;\n\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>,\n  handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>\n) => PromiseOrValue<ToolMessage | Command>;\n\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: Omit<\n    ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,\n    /**\n     * allow to reset the system prompt or system message\n     */\n    \"systemPrompt\" | \"systemMessage\"\n  > & { systemPrompt?: string; systemMessage?: SystemMessage }\n) => PromiseOrValue<AIMessage>;\n\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> = (\n  request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,\n  handler: WrapModelCallHandler<TSchema, TContext>\n) => PromiseOrValue<AIMessage | Command>;\n\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeAgentHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | BeforeAgentHandler<TSchema, TContext>\n  | {\n      hook: BeforeAgentHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeModelHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | BeforeModelHandler<TSchema, TContext>\n  | {\n      hook: BeforeModelHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterModelHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | AfterModelHandler<TSchema, TContext>\n  | {\n      hook: AfterModelHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterAgentHandler<TSchema, TContext> = (\n  state: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;\n\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<\n  TSchema extends StateDefinitionInit | undefined = undefined,\n  TContext = unknown,\n> =\n  | AfterAgentHandler<TSchema, TContext>\n  | {\n      hook: AfterAgentHandler<TSchema, TContext>;\n      canJumpTo?: JumpToTarget[];\n    };\n\n/**\n * Unique symbol used to brand middleware instances.\n * This prevents functions from being accidentally assignable to AgentMiddleware\n * since functions have a 'name' property that would otherwise make them structurally compatible.\n */\nexport const MIDDLEWARE_BRAND: symbol = Symbol.for(\"AgentMiddleware\");\n\n/**\n * Widened middleware type for agent configuration arrays. Accepts middleware\n * instances with any registered tools or stream transformers while preserving\n * precise inference when middleware arrays are passed with `const`.\n */\nexport type AnyAgentMiddleware = AgentMiddleware<\n  StateDefinitionInit | undefined,\n  | InteropZodObject\n  | InteropZodDefault<InteropZodObject>\n  | InteropZodOptional<InteropZodObject>\n  | undefined,\n  any,\n  readonly (ClientTool | ServerTool)[],\n  ReadonlyArray<() => StreamTransformer<any>>\n>;\n\n/**\n * Base middleware interface.\n *\n * @typeParam TSchema - The middleware state schema type\n * @typeParam TContextSchema - The middleware context schema type\n * @typeParam TFullContext - The full context type available to hooks\n * @typeParam TTools - The tools array type registered by the middleware\n * @typeParam TStreamTransformers - The stream transformer factories registered\n *   by the middleware\n *\n * @example\n * ```typescript\n * const middleware = createMiddleware({\n *   name: \"myMiddleware\",\n *   stateSchema: z.object({ count: z.number() }),\n *   tools: [myTool],\n * });\n * ```\n */\nexport interface AgentMiddleware<\n  TSchema extends StateDefinitionInit | undefined = any,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined = any,\n  TFullContext = any,\n  TTools extends readonly (ClientTool | ServerTool)[] = readonly (\n    | ClientTool\n    | ServerTool\n  )[],\n  TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> =\n    ReadonlyArray<() => StreamTransformer<any>>,\n> {\n  /**\n   * Brand property to distinguish middleware instances from plain objects or functions.\n   * This is required and prevents accidental assignment of functions to middleware arrays.\n   */\n  readonly [MIDDLEWARE_BRAND]: true;\n\n  /**\n   * Type marker for extracting the MiddlewareTypeConfig from a middleware instance.\n   * This is a phantom property used only for type inference.\n   * @internal\n   */\n  readonly \"~middlewareTypes\"?: MiddlewareTypeConfig<\n    TSchema,\n    TContextSchema,\n    TFullContext,\n    TTools,\n    TStreamTransformers\n  >;\n\n  /**\n   * The name of the middleware.\n   */\n  name: string;\n\n  /**\n   * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:\n   * - A Zod object (InteropZodObject)\n   * - A StateSchema from LangGraph (supports ReducedValue, UntrackedValue)\n   * - An AnnotationRoot\n   * - Undefined\n   */\n  stateSchema?: TSchema;\n\n  /**\n   * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:\n   * - A Zod object\n   * - A Zod optional object\n   * - A Zod default object\n   * - Undefined\n   */\n  contextSchema?: TContextSchema;\n\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: TTools;\n\n  /**\n   * Stream transformer factories registered by the middleware.\n   * These are merged with `createAgent({ streamTransformers })` when the agent\n   * is compiled.\n   */\n  streamTransformers?: TStreamTransformers;\n\n  /**\n   * Wraps tool execution with custom logic. This allows you to:\n   * - Modify tool call parameters before execution\n   * - Handle errors and retry with different parameters\n   * - Post-process tool results\n   * - Implement caching, logging, authentication, or other cross-cutting concerns\n   * - Return Command objects for advanced control flow\n   *\n   * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n   * along with a handler function to execute the actual tool.\n   *\n   * @param request - The tool call request containing toolCall, state, and runtime.\n   * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n   * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n   *\n   * @example\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   console.log(`Calling tool: ${request.tool.name}`);\n   *   console.log(`Tool description: ${request.tool.description}`);\n   *\n   *   try {\n   *     // Execute the tool\n   *     const result = await handler(request);\n   *     console.log(`Tool ${request.tool.name} succeeded`);\n   *     return result;\n   *   } catch (error) {\n   *     console.error(`Tool ${request.tool.name} failed:`, error);\n   *     // Could return a custom error message or retry\n   *     throw error;\n   *   }\n   * }\n   * ```\n   *\n   * @example Authentication\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   // Check if user is authorized for this tool\n   *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n   *     return new ToolMessage({\n   *       content: \"Unauthorized to call this tool\",\n   *       tool_call_id: request.toolCall.id,\n   *     });\n   *   }\n   *   return handler(request);\n   * }\n   * ```\n   *\n   * @example Caching\n   * ```ts\n   * const cache = new Map();\n   * wrapToolCall: async (request, handler) => {\n   *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n   *   if (cache.has(cacheKey)) {\n   *     return cache.get(cacheKey);\n   *   }\n   *   const result = await handler(request);\n   *   cache.set(cacheKey, result);\n   *   return result;\n   * }\n   * ```\n   */\n  wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n\n  /**\n   * Wraps the model invocation with custom logic. This allows you to:\n   * - Modify the request before calling the model\n   * - Handle errors and retry with different parameters\n   * - Post-process the response\n   * - Implement custom caching, logging, or other cross-cutting concerns\n   *\n   * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n   * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n   * @returns The response from the model (or a modified version).\n   *\n   * @example\n   * ```ts\n   * wrapModelCall: async (request, handler) => {\n   *   // Modify request before calling\n   *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n   *\n   *   try {\n   *     // Call the model\n   *     return await handler(modifiedRequest);\n   *   } catch (error) {\n   *     // Handle errors and retry with fallback\n   *     const fallbackRequest = { ...request, model: fallbackModel };\n   *     return await handler(fallbackRequest);\n   *   }\n   * }\n   * ```\n   */\n  wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n   * It allows to modify the state of the agent before any model calls or tool executions.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n   * It allows to modify the state of the agent.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n   * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  afterModel?: AfterModelHook<TSchema, TFullContext>;\n\n  /**\n   * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n   * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n   *\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified middleware state or undefined to pass through\n   */\n  afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n  [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\n\n/**\n * Helper type to resolve a MiddlewareTypeConfig from either:\n * - A MiddlewareTypeConfig directly\n * - An AgentMiddleware instance (using `typeof middleware`)\n */\nexport type ResolveMiddlewareTypeConfig<T> = T extends {\n  \"~middlewareTypes\"?: infer Types;\n}\n  ? Types extends MiddlewareTypeConfig\n    ? Types\n    : never\n  : T extends MiddlewareTypeConfig\n    ? T\n    : never;\n\n/**\n * Helper type to extract any property from a MiddlewareTypeConfig or AgentMiddleware.\n *\n * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from\n * @typeParam K - The property key to extract (\"Schema\" | \"ContextSchema\" | \"FullContext\" | \"Tools\" | \"StreamTransformers\")\n */\nexport type InferMiddlewareType<\n  T,\n  K extends keyof MiddlewareTypeConfig,\n> = ResolveMiddlewareTypeConfig<T>[K];\n\n/**\n * Shorthand helper to extract the Schema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareSchema<T> = InferMiddlewareType<T, \"Schema\">;\n\n/**\n * Shorthand helper to extract the ContextSchema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareContextSchema<T> = InferMiddlewareType<\n  T,\n  \"ContextSchema\"\n>;\n\n/**\n * Shorthand helper to extract the FullContext type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareFullContext<T> = InferMiddlewareType<\n  T,\n  \"FullContext\"\n>;\n\n/**\n * Shorthand helper to extract the Tools type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareToolsFromConfig<T> = InferMiddlewareType<T, \"Tools\">;\n\n/**\n * Shorthand helper to extract the StreamTransformers type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareStreamTransformersFromConfig<T> =\n  InferMiddlewareType<T, \"StreamTransformers\">;\n\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> =\n  T extends AnyAnnotationRoot\n    ? ToAnnotationRoot<T>[\"State\"]\n    : T extends InteropZodObject\n      ? InferInteropZodInput<T>\n      : {};\n\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer TSchema, any, any, any, any>\n    ? TSchema extends StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaValue<TFields>>\n      : TSchema extends InteropZodObject\n        ? FilterPrivateProps<InferInteropZodOutput<TSchema>>\n        : TSchema extends StateDefinitionInit\n          ? FilterPrivateProps<InferSchemaValue<TSchema>>\n          : {}\n    : {};\n\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer TSchema, any, any, any, any>\n    ? TSchema extends StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaUpdate<TFields>>\n      : TSchema extends InteropZodObject\n        ? FilterPrivateProps<InferInteropZodInput<TSchema>>\n        : TSchema extends StateDefinitionInit\n          ? FilterPrivateProps<InferSchemaInput<TSchema>>\n          : {}\n    : {};\n\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T extends readonly AnyAgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AnyAgentMiddleware[]\n          ? InferMiddlewareState<First> & InferMiddlewareStates<Rest>\n          : InferMiddlewareState<First>\n        : {}\n      : {};\n\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<\n  T extends readonly AnyAgentMiddleware[],\n> = T extends readonly []\n  ? {}\n  : T extends readonly [infer First, ...infer Rest]\n    ? First extends AgentMiddleware\n      ? Rest extends readonly AnyAgentMiddleware[]\n        ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest>\n        : InferMiddlewareInputState<First>\n      : {}\n    : {};\n\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AnyAgentMiddleware[]> =\n  InferMiddlewareStates<T> & AgentBuiltInState;\n\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AnyAgentMiddleware[]> =\n  InferMiddlewareInputStates<T> & AgentBuiltInState;\n\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> =\n  T extends AgentMiddleware<any, infer TContextSchema, any, any, any>\n    ? TContextSchema extends InteropZodObject\n      ? InferInteropZodInput<TContextSchema>\n      : {}\n    : {};\n\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> =\n  T extends AgentMiddleware<any, infer TContextSchema, any, any, any>\n    ? TContextSchema extends InteropZodOptional<infer Inner>\n      ? InferInteropZodInput<Inner> | undefined\n      : TContextSchema extends InteropZodObject\n        ? InferInteropZodInput<TContextSchema>\n        : {}\n    : {};\n\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AnyAgentMiddleware[]> =\n  T extends readonly []\n    ? {}\n    : T extends readonly [infer First, ...infer Rest]\n      ? First extends AgentMiddleware\n        ? Rest extends readonly AnyAgentMiddleware[]\n          ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest>\n          : InferMiddlewareContext<First>\n        : {}\n      : {};\n\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined]\n  ? [B] extends [undefined]\n    ? undefined\n    : B | undefined\n  : [B] extends [undefined]\n    ? A | undefined\n    : [A] extends [B]\n      ? A\n      : [B] extends [A]\n        ? B\n        : A & B;\n\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<\n  T extends readonly AnyAgentMiddleware[],\n> = T extends readonly []\n  ? {}\n  : T extends readonly [infer First, ...infer Rest]\n    ? First extends AgentMiddleware\n      ? Rest extends readonly AnyAgentMiddleware[]\n        ? MergeContextTypes<\n            InferMiddlewareContextInput<First>,\n            InferMiddlewareContextInputs<Rest>\n          >\n        : InferMiddlewareContextInput<First>\n      : {}\n    : {};\n\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject,\n> = ContextSchema extends InteropZodObject\n  ? InferInteropZodInput<ContextSchema>\n  : ContextSchema extends AnyAnnotationRoot\n    ? ToAnnotationRoot<ContextSchema>[\"State\"]\n    : {};\n\nexport type ToAnnotationRoot<A extends StateDefinitionInit> =\n  A extends AnyAnnotationRoot\n    ? A\n    : A extends InteropZodObject\n      ? InteropZodToStateDefinition<A>\n      : never;\n\nexport type InferSchemaValue<A extends StateDefinitionInit | undefined> = [\n  A,\n] extends [undefined]\n  ? {}\n  : A extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodOutput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"State\"]\n        : {};\n\nexport type InferSchemaInput<A extends StateDefinitionInit | undefined> = [\n  A,\n] extends [undefined]\n  ? {}\n  : A extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodInput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"Update\"]\n        : {};\n"],"mappings":";;;;;;AA6WA,MAAa,mBAA2B,OAAO,IAAI,kBAAkB"}

package/dist/agents/middleware/types.d.cts

@@ -4,7 +4,7 @@ import { ModelRequest } from "../nodes/types.cjs";
 import { AIMessage, SystemMessage, ToolMessage } from "@langchain/core/messages";
 import { ClientTool, ServerTool } from "@langchain/core/tools";
 import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
-import { AnnotationRoot, Command, InferStateSchemaUpdate, InferStateSchemaValue, StateDefinitionInit, StateSchema } from "@langchain/langgraph";
+import { AnnotationRoot, Command, InferStateSchemaUpdate, InferStateSchemaValue, StateDefinitionInit, StateSchema, StreamTransformer } from "@langchain/langgraph";
 import { ToolCall as ToolCall$1 } from "@langchain/core/messages/tool";
 import { InteropZodToStateDefinition } from "@langchain/langgraph/zod";
 
@@ -28,6 +28,9 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  *
  * @typeParam TTools - The tools array type registered by the middleware.
  *
+ * @typeParam TStreamTransformers - The stream transformer factories registered
+ *   by the middleware.
+ *
  * @example
  * ```typescript
  * // Define a type configuration
@@ -39,7 +42,7 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  * >;
  * ```
  */
-interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[], TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> = ReadonlyArray<() => StreamTransformer<any>>> {
   /** The middleware state schema type */
   Schema: TSchema;
   /** The middleware context schema type */
@@ -48,6 +51,8 @@ interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined =
   FullContext: TFullContext;
   /** The tools array type */
   Tools: TTools;
+  /** The stream transformer factories registered by the middleware */
+  StreamTransformers: TStreamTransformers;
 }
 /**
  * Default type configuration for middleware.
@@ -221,6 +226,12 @@ type AfterAgentHook<TSchema extends StateDefinitionInit | undefined = undefined,
  * since functions have a 'name' property that would otherwise make them structurally compatible.
  */
 declare const MIDDLEWARE_BRAND: symbol;
+/**
+ * Widened middleware type for agent configuration arrays. Accepts middleware
+ * instances with any registered tools or stream transformers while preserving
+ * precise inference when middleware arrays are passed with `const`.
+ */
+type AnyAgentMiddleware = AgentMiddleware<StateDefinitionInit | undefined, InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, any, readonly (ClientTool | ServerTool)[], ReadonlyArray<() => StreamTransformer<any>>>;
 /**
  * Base middleware interface.
  *
@@ -228,6 +239,8 @@ declare const MIDDLEWARE_BRAND: symbol;
  * @typeParam TContextSchema - The middleware context schema type
  * @typeParam TFullContext - The full context type available to hooks
  * @typeParam TTools - The tools array type registered by the middleware
+ * @typeParam TStreamTransformers - The stream transformer factories registered
+ *   by the middleware
  *
  * @example
  * ```typescript
@@ -238,13 +251,13 @@ declare const MIDDLEWARE_BRAND: symbol;
  * });
  * ```
  */
-interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[], TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> = ReadonlyArray<() => StreamTransformer<any>>> {
   /**
    * Type marker for extracting the MiddlewareTypeConfig from a middleware instance.
    * This is a phantom property used only for type inference.
    * @internal
    */
-  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools>;
+  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools, TStreamTransformers>;
   /**
    * The name of the middleware.
    */
@@ -269,6 +282,12 @@ interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any,
    * Additional tools registered by the middleware.
    */
   tools?: TTools;
+  /**
+   * Stream transformer factories registered by the middleware.
+   * These are merged with `createAgent({ streamTransformers })` when the agent
+   * is compiled.
+   */
+  streamTransformers?: TStreamTransformers;
   /**
    * Wraps tool execution with custom logic. This allows you to:
    * - Modify tool call parameters before execution
@@ -414,7 +433,7 @@ type ResolveMiddlewareTypeConfig<T> = T extends {
  * Helper type to extract any property from a MiddlewareTypeConfig or AgentMiddleware.
  *
  * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from
- * @typeParam K - The property key to extract ("Schema" | "ContextSchema" | "FullContext" | "Tools")
+ * @typeParam K - The property key to extract ("Schema" | "ContextSchema" | "FullContext" | "Tools" | "StreamTransformers")
  */
 type InferMiddlewareType<T, K extends keyof MiddlewareTypeConfig> = ResolveMiddlewareTypeConfig<T>[K];
 /**
@@ -433,47 +452,51 @@ type InferMiddlewareFullContext<T> = InferMiddlewareType<T, "FullContext">;
  * Shorthand helper to extract the Tools type from a MiddlewareTypeConfig or AgentMiddleware.
  */
 type InferMiddlewareToolsFromConfig<T> = InferMiddlewareType<T, "Tools">;
+/**
+ * Shorthand helper to extract the StreamTransformers type from a MiddlewareTypeConfig or AgentMiddleware.
+ */
+type InferMiddlewareStreamTransformersFromConfig<T> = InferMiddlewareType<T, "StreamTransformers">;
 type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>["State"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};
 /**
  * Helper type to infer the state schema type from a middleware
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaValue<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaValue<TSchema>> : {} : {};
+type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaValue<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaValue<TSchema>> : {} : {};
 /**
  * Helper type to infer the input state schema type from a middleware (all properties optional)
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaUpdate<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};
+type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaUpdate<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (just the middleware states)
  */
-type InferMiddlewareStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};
+type InferMiddlewareStates<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};
 /**
  * Helper type to infer merged input state from an array of middleware (with optional defaults)
  */
-type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};
+type InferMiddlewareInputStates<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (includes built-in state)
  */
-type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;
+type InferMergedState<T extends readonly AnyAgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;
 /**
  * Helper type to infer merged input state from an array of middleware (includes built-in state)
  */
-type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;
+type InferMergedInputState<T extends readonly AnyAgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;
 /**
  * Helper type to infer the context schema type from a middleware
  */
-type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any> ? TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
+type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any, any> ? TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
 /**
  * Helper type to infer the input context schema type from a middleware (with optional defaults)
  */
-type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any> ? TContextSchema extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
+type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any, any> ? TContextSchema extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
 /**
  * Helper type to infer merged context from an array of middleware
  */
-type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};
+type InferMiddlewareContexts<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};
 /**
  * Helper to merge two context types, preserving undefined unions
  */
@@ -481,14 +504,14 @@ type MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined]
 /**
  * Helper type to infer merged input context from an array of middleware (with optional defaults)
  */
-type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};
+type InferMiddlewareContextInputs<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};
 /**
  * Helper type to extract input type from context schema (with optional defaults)
  */
 type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>["State"] : {};
 type ToAnnotationRoot<A extends StateDefinitionInit> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? InteropZodToStateDefinition<A> : never;
-type InferSchemaValue<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
-type InferSchemaInput<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodInput<A> : A extends AnyAnnotationRoot ? A["Update"] : {};
+type InferSchemaValue<A extends StateDefinitionInit | undefined> = [A] extends [undefined] ? {} : A extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
+type InferSchemaInput<A extends StateDefinitionInit | undefined> = [A] extends [undefined] ? {} : A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodInput<A> : A extends AnyAnnotationRoot ? A["Update"] : {};
 //#endregion
-export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, InferSchemaUpdateType, InferSchemaValue, InferSchemaValueType, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, NormalizedSchemaUpdate, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
+export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareStreamTransformersFromConfig, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, InferSchemaUpdateType, InferSchemaValue, InferSchemaValueType, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, NormalizedSchemaUpdate, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
 //# sourceMappingURL=types.d.cts.map

package/dist/agents/middleware/types.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA6BK,cAAA,MAAoB,CAAA,GAAI,OAAA,CAAQ,CAAA;AAAA,KAEzB,iBAAA,GAAoB,cAAA;;;;;;;;;;;;;;AAAhC;;;;;AA8BA;;;;;;;;;;UAAiB,oBAAA,iBACC,mBAAA,eACZ,mBAAA,qCAGA,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,gBAEnB,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,4DAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EANkB;EAUtB,MAAA,EAAQ,OAAA;EATe;EAWvB,aAAA,EAAe,cAAA;EARU;EAUzB,WAAA,EAAa,YAAA;EATT;EAWJ,KAAA,EAAO,MAAA;AAAA;;;;;KAOG,2BAAA,GAA8B,oBAAA;AAAA,KAE9B,oBAAA,aAAiC,OAAA,oBACzC,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,aAAkC,OAAA,oBAC1C,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,sBAAA,CAAuB,OAAA,IAAW,iBAAA,GAClC,OAAA,SAAgB,gBAAA,GACd,oBAAA,CAAqB,OAAA,IAAW,iBAAA,GAChC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,iBACM,mBAAA,8BACd,oBAAA,CAAqB,OAAA;AAAA,KAEb,sBAAA,iBACM,mBAAA,8BACd,qBAAA,CAAsB,OAAA;;;;KAKd,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EA/DzC;;;EAqEA,QAAA,EAAU,UAAA;EAnEN;;;;;;;;;;;AAiBN;;;;;AAEA;;;;EAqEE,IAAA,EAAM,UAAA,GAAa,UAAA;EAnEjB;;;EAuEF,KAAA,EAAO,MAAA,GAAS,iBAAA;EAtEqB;;;EA0ErC,OAAA,EAAS,SAAA,CAAQ,QAAA;AAAA;;;;;KAOP,eAAA,iBACM,MAAA,oBAA0B,iBAAA,yBAG1C,OAAA,EAAS,eAAA,CAAgB,OAAA,EAAS,QAAA,MAC/B,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;KAMtB,gBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACzD,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,MACtD,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;;;;KAStB,oBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,IAAA,CACP,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA;;;;;EAKzC,YAAA;EAAuB,aAAA,GAAgB,aAAA;AAAA,MAC1C,cAAA,CAAe,SAAA;;;;;;;AA/GpB;;;;;;;KA8HY,iBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACtD,OAAA,EAAS,oBAAA,CAAqB,OAAA,EAAS,QAAA,MACpC,cAAA,CAAe,SAAA,GAAY,OAAA;;;;;;;;;KAU3B,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;AAjLlB;;;;;;;;KA6LK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;AAjNnE;;;;KAwNY,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;AArNlB;;;cA6Na,gBAAA;;;;;;;;;;;;;;;;;;UAmBI,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EA9Ne;;;;;EAAA,SA4OV,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA;EAxOe;;;EA8OjB,IAAA;EAvOyB;;;;;;;EAgPzB,WAAA,GAAc,OAAA;EA3OkB;;;;;;;EAoPhC,aAAA,GAAgB,cAAA;EArPP;;;EA0PT,KAAA,GAAQ,MAAA;EAzPL;;;;;AAML;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAeA;;;;;;;;;;;;;;;;;;;;;;EAmSE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EAzRZ;;;;;;;AAgB/B;;;;;;;;;;;;;;;;;;;;;EAuSE,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EAnSW;;;;;;;;EA6StD,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA3SF;;AAAE;;;;;;EAqTvC,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAxS0B;;;;;;;;EAkTjE,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EApT9B;;;;;;;;EA8TP,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;;KAMlC,kBAAA,oBACS,CAAA,IAAK,CAAA,gCAAiC,CAAA,GAAI,CAAA,CAAE,CAAA;;;;;;KAQ9C,2BAAA,MAAiC,CAAA;EAC3C,kBAAA;AAAA,IAEE,KAAA,SAAc,oBAAA,GACZ,KAAA,WAEF,CAAA,SAAU,oBAAA,GACR,CAAA;;;;;;;KASM,mBAAA,oBAEM,oBAAA,IACd,2BAAA,CAA4B,CAAA,EAAG,CAAA;;;;KAKvB,qBAAA,MAA2B,mBAAA,CAAoB,CAAA;;;;KAK/C,4BAAA,MAAkC,mBAAA,CAC5C,CAAA;;AA1VI;;KAiWM,0BAAA,MAAgC,mBAAA,CAC1C,CAAA;;;;KAOU,8BAAA,MAAoC,mBAAA,CAAoB,CAAA;AAAA,KAExD,gBAAA,WAA2B,iBAAA,GAAoB,gBAAA,IACzD,CAAA,SAAU,iBAAA,GACN,gBAAA,CAAiB,CAAA,aACjB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA;;;;;;KAQjB,oBAAA,WAA+B,eAAA,IACzC,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;;KASpC,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,sBAAA,CAAuB,OAAA,KAC1C,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOpC,qBAAA,oBAAyC,eAAA,MACnD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,oBAAA,CAAqB,KAAA,IAAS,qBAAA,CAAsB,IAAA,IACpD,oBAAA,CAAqB,KAAA;;;;KAOrB,0BAAA,oBAA8C,eAAA,MACxD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,yBAAA,CAA0B,KAAA,IAAS,0BAAA,CAA2B,IAAA,IAC9D,yBAAA,CAA0B,KAAA;;AAtZtC;;KA6ZY,gBAAA,oBAAoC,eAAA,MAC9C,qBAAA,CAAsB,CAAA,IAAK,iBAAA;;;;KAKjB,qBAAA,oBAAyC,eAAA,MACnD,0BAAA,CAA2B,CAAA,IAAK,iBAAA;;;;KAKtB,sBAAA,WAAiC,eAAA,IAC3C,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOjB,2BAAA,WAAsC,eAAA,IAChD,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,kBAAA,gBACrB,oBAAA,CAAqB,KAAA,gBACrB,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOnB,uBAAA,oBAA2C,eAAA,MACrD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,sBAAA,CAAuB,KAAA,IAAS,uBAAA,CAAwB,IAAA,IACxD,sBAAA,CAAuB,KAAA;;;;KAO9B,iBAAA,UAA2B,CAAA,yBAC3B,CAAA,oCAEC,CAAA,gBACD,CAAA,wBACC,CAAA,gBACC,CAAA,WAAY,CAAA,IACX,CAAA,IACC,CAAA,WAAY,CAAA,IACX,CAAA,GACA,CAAA,GAAI,CAAA;;;;KAKF,4BAAA,oBAAgD,eAAA,MAC1D,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,iBAAA,CACE,2BAAA,CAA4B,KAAA,GAC5B,4BAAA,CAA6B,IAAA,KAE/B,2BAAA,CAA4B,KAAA;AA9dlC;;;AAAA,KAqeM,iBAAA,uBACY,iBAAA,GAAoB,gBAAA,IACxC,aAAA,SAAsB,gBAAA,GACtB,oBAAA,CAAqB,aAAA,IACrB,aAAA,SAAsB,iBAAA,GACpB,gBAAA,CAAiB,aAAA;AAAA,KAGX,gBAAA,WAA2B,mBAAA,IACrC,CAAA,SAAU,iBAAA,GACN,CAAA,GACA,CAAA,SAAU,gBAAA,GACR,2BAAA,CAA4B,CAAA;AAAA,KAGxB,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,qBAAA,CAAsB,OAAA,IACtB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA;AAAA,KAGE,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA,IACrB,CAAA,SAAU,iBAAA,GACR,CAAA"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA6BK,cAAA,MAAoB,CAAA,GAAI,OAAA,CAAQ,CAAA;AAAA,KAEzB,iBAAA,GAAoB,cAAA;;;;;;;;;;;;;;AAAhC;;;;;AAiCA;;;;;;;;;;;;;UAAiB,oBAAA,iBACC,mBAAA,eACZ,mBAAA,qCAGA,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,gBAEnB,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,4DAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA,iCAEwB,aAAA,OAAoB,iBAAA,SAC9C,aAAA,OAAoB,iBAAA;EARlB;EAWJ,MAAA,EAAQ,OAAA;EAR8B;EAUtC,aAAA,EAAe,cAAA;EARX;EAUJ,WAAA,EAAa,YAAA;EARe;EAU5B,KAAA,EAAO,MAAA;EATL;EAWF,kBAAA,EAAoB,mBAAA;AAAA;;;;;KAOV,2BAAA,GAA8B,oBAAA;AAAA,KAE9B,oBAAA,aAAiC,OAAA,oBACzC,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,aAAkC,OAAA,oBAC1C,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,sBAAA,CAAuB,OAAA,IAAW,iBAAA,GAClC,OAAA,SAAgB,gBAAA,GACd,oBAAA,CAAqB,OAAA,IAAW,iBAAA,GAChC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,iBACM,mBAAA,8BACd,oBAAA,CAAqB,OAAA;AAAA,KAEb,sBAAA,iBACM,mBAAA,8BACd,qBAAA,CAAsB,OAAA;;;;KAKd,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EAnEzC;;;EAyEA,QAAA,EAAU,UAAA;EAvEN;;;;;;;;;;;;;;;;;;AAqBN;;EAuEE,IAAA,EAAM,UAAA,GAAa,UAAA;EAvEqB;;AAE1C;EAyEE,KAAA,EAAO,MAAA,GAAS,iBAAA;EAzEc;;;EA6E9B,OAAA,EAAS,SAAA,CAAQ,QAAA;AAAA;;;;;KAOP,eAAA,iBACM,MAAA,oBAA0B,iBAAA,yBAG1C,OAAA,EAAS,eAAA,CAAgB,OAAA,EAAS,QAAA,MAC/B,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;KAMtB,gBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACzD,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,MACtD,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;;;;KAStB,oBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,IAAA,CACP,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA;;;;;EAKzC,YAAA;EAAuB,aAAA,GAAgB,aAAA;AAAA,MAC1C,cAAA,CAAe,SAAA;;;;;;;;;;;;;;KAeR,iBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACtD,OAAA,EAAS,oBAAA,CAAqB,OAAA,EAAS,QAAA,MACpC,cAAA,CAAe,SAAA,GAAY,OAAA;;;;;;;;;KAU3B,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;AAjLlB;KA6LK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;AA1MlB;;;;;KAqNK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;;;cAQL,gBAAA;;;;AA7Nb;;KAoOY,kBAAA,GAAqB,eAAA,CAC/B,mBAAA,cACE,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,8BAGX,UAAA,GAAa,UAAA,KACvB,aAAA,OAAoB,iBAAA;;;;;;;;;;;;;;;;;;;;UAsBL,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA,iCAEwB,aAAA,OAAoB,iBAAA,SAC9C,aAAA,OAAoB,iBAAA;EA/OtB;;;;;EAAA,SA4PS,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA,EACA,mBAAA;EA7PuB;;AAO3B;EA4PE,IAAA;EA5PyB;;;;;;;EAqQzB,WAAA,GAAc,OAAA;EAhQX;;;;;;;EAyQH,aAAA,GAAgB,cAAA;EA1QS;;;EA+QzB,KAAA,GAAQ,MAAA;EA9QU;;;;AAMpB;EA+QE,kBAAA,GAAqB,mBAAA;EA/QK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAe5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgUE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EAtS/B;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoUV,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EA9TzB;;;;AAAqB;;;;EAwUvC,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA5TtB;;;;;;;;EAsUjB,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAxUjB;;;;;;;;EAkVtB,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EA/UnB;;;;;;AAOpB;;EAkVE,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;;KAMlC,kBAAA,oBACS,CAAA,IAAK,CAAA,gCAAiC,CAAA,GAAI,CAAA,CAAE,CAAA;;;;;;KAQ9C,2BAAA,MAAiC,CAAA;EAC3C,kBAAA;AAAA,IAEE,KAAA,SAAc,oBAAA,GACZ,KAAA,WAEF,CAAA,SAAU,oBAAA,GACR,CAAA;;;;;;;KASM,mBAAA,oBAEM,oBAAA,IACd,2BAAA,CAA4B,CAAA,EAAG,CAAA;;;AA5W7B;KAiXM,qBAAA,MAA2B,mBAAA,CAAoB,CAAA;;;;KAK/C,4BAAA,MAAkC,mBAAA,CAC5C,CAAA;;;;KAOU,0BAAA,MAAgC,mBAAA,CAC1C,CAAA;;;;KAOU,8BAAA,MAAoC,mBAAA,CAAoB,CAAA;;;;KAKxD,2CAAA,MACV,mBAAA,CAAoB,CAAA;AAAA,KAEV,gBAAA,WAA2B,iBAAA,GAAoB,gBAAA,IACzD,CAAA,SAAU,iBAAA,GACN,gBAAA,CAAiB,CAAA,aACjB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA;;;;;;KAQjB,oBAAA,WAA+B,eAAA,IACzC,CAAA,SAAU,eAAA,sCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;AA7YhD;KAsZY,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,sCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,sBAAA,CAAuB,OAAA,KAC1C,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOpC,qBAAA,oBAAyC,kBAAA,MACnD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,oBAAA,CAAqB,KAAA,IAAS,qBAAA,CAAsB,IAAA,IACpD,oBAAA,CAAqB,KAAA;;;;KAOrB,0BAAA,oBACS,kBAAA,MACjB,CAAA,4BAEA,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,yBAAA,CAA0B,KAAA,IAAS,0BAAA,CAA2B,IAAA,IAC9D,yBAAA,CAA0B,KAAA;;;;KAOxB,gBAAA,oBAAoC,kBAAA,MAC9C,qBAAA,CAAsB,CAAA,IAAK,iBAAA;;;;KAKjB,qBAAA,oBAAyC,kBAAA,MACnD,0BAAA,CAA2B,CAAA,IAAK,iBAAA;;;;KAKtB,sBAAA,WAAiC,eAAA,IAC3C,CAAA,SAAU,eAAA,6CACN,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOjB,2BAAA,WAAsC,eAAA,IAChD,CAAA,SAAU,eAAA,6CACN,cAAA,SAAuB,kBAAA,gBACrB,oBAAA,CAAqB,KAAA,gBACrB,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;AApdzB;;KA2dM,uBAAA,oBAA2C,kBAAA,MACrD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,sBAAA,CAAuB,KAAA,IAAS,uBAAA,CAAwB,IAAA,IACxD,sBAAA,CAAuB,KAAA;;;;KAO9B,iBAAA,UAA2B,CAAA,yBAC3B,CAAA,oCAEC,CAAA,gBACD,CAAA,wBACC,CAAA,gBACC,CAAA,WAAY,CAAA,IACX,CAAA,IACC,CAAA,WAAY,CAAA,IACX,CAAA,GACA,CAAA,GAAI,CAAA;;;;KAKF,4BAAA,oBACS,kBAAA,MACjB,CAAA,4BAEA,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,iBAAA,CACE,2BAAA,CAA4B,KAAA,GAC5B,4BAAA,CAA6B,IAAA,KAE/B,2BAAA,CAA4B,KAAA;;;;KAO1B,iBAAA,uBACY,iBAAA,GAAoB,gBAAA,IACxC,aAAA,SAAsB,gBAAA,GACtB,oBAAA,CAAqB,aAAA,IACrB,aAAA,SAAsB,iBAAA,GACpB,gBAAA,CAAiB,aAAA;AAAA,KAGX,gBAAA,WAA2B,mBAAA,IACrC,CAAA,SAAU,iBAAA,GACN,CAAA,GACA,CAAA,SAAU,gBAAA,GACR,2BAAA,CAA4B,CAAA;AAAA,KAGxB,gBAAA,WAA2B,mBAAA,iBACrC,CAAA,6BAGE,CAAA,SAAU,WAAA,kBACR,qBAAA,CAAsB,OAAA,IACtB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA;AAAA,KAGE,gBAAA,WAA2B,mBAAA,iBACrC,CAAA,6BAGE,CAAA,SAAU,WAAA,kBACR,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA,IACrB,CAAA,SAAU,iBAAA,GACR,CAAA"}

package/dist/agents/middleware/types.d.ts

@@ -3,7 +3,7 @@ import { AgentBuiltInState, Runtime as Runtime$1 } from "../runtime.js";
 import { ModelRequest } from "../nodes/types.js";
 import { AIMessage, SystemMessage, ToolMessage } from "@langchain/core/messages";
 import { ClientTool, ServerTool } from "@langchain/core/tools";
-import { AnnotationRoot, Command, InferStateSchemaUpdate, InferStateSchemaValue, StateDefinitionInit, StateSchema } from "@langchain/langgraph";
+import { AnnotationRoot, Command, InferStateSchemaUpdate, InferStateSchemaValue, StateDefinitionInit, StateSchema, StreamTransformer } from "@langchain/langgraph";
 import { InteropZodToStateDefinition } from "@langchain/langgraph/zod";
 import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
 import { ToolCall as ToolCall$1 } from "@langchain/core/messages/tool";
@@ -28,6 +28,9 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  *
  * @typeParam TTools - The tools array type registered by the middleware.
  *
+ * @typeParam TStreamTransformers - The stream transformer factories registered
+ *   by the middleware.
+ *
  * @example
  * ```typescript
  * // Define a type configuration
@@ -39,7 +42,7 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  * >;
  * ```
  */
-interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[], TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> = ReadonlyArray<() => StreamTransformer<any>>> {
   /** The middleware state schema type */
   Schema: TSchema;
   /** The middleware context schema type */
@@ -48,6 +51,8 @@ interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined =
   FullContext: TFullContext;
   /** The tools array type */
   Tools: TTools;
+  /** The stream transformer factories registered by the middleware */
+  StreamTransformers: TStreamTransformers;
 }
 /**
  * Default type configuration for middleware.
@@ -221,6 +226,12 @@ type AfterAgentHook<TSchema extends StateDefinitionInit | undefined = undefined,
  * since functions have a 'name' property that would otherwise make them structurally compatible.
  */
 declare const MIDDLEWARE_BRAND: symbol;
+/**
+ * Widened middleware type for agent configuration arrays. Accepts middleware
+ * instances with any registered tools or stream transformers while preserving
+ * precise inference when middleware arrays are passed with `const`.
+ */
+type AnyAgentMiddleware = AgentMiddleware<StateDefinitionInit | undefined, InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, any, readonly (ClientTool | ServerTool)[], ReadonlyArray<() => StreamTransformer<any>>>;
 /**
  * Base middleware interface.
  *
@@ -228,6 +239,8 @@ declare const MIDDLEWARE_BRAND: symbol;
  * @typeParam TContextSchema - The middleware context schema type
  * @typeParam TFullContext - The full context type available to hooks
  * @typeParam TTools - The tools array type registered by the middleware
+ * @typeParam TStreamTransformers - The stream transformer factories registered
+ *   by the middleware
  *
  * @example
  * ```typescript
@@ -238,13 +251,13 @@ declare const MIDDLEWARE_BRAND: symbol;
  * });
  * ```
  */
-interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[], TStreamTransformers extends ReadonlyArray<() => StreamTransformer<any>> = ReadonlyArray<() => StreamTransformer<any>>> {
   /**
    * Type marker for extracting the MiddlewareTypeConfig from a middleware instance.
    * This is a phantom property used only for type inference.
    * @internal
    */
-  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools>;
+  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools, TStreamTransformers>;
   /**
    * The name of the middleware.
    */
@@ -269,6 +282,12 @@ interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any,
    * Additional tools registered by the middleware.
    */
   tools?: TTools;
+  /**
+   * Stream transformer factories registered by the middleware.
+   * These are merged with `createAgent({ streamTransformers })` when the agent
+   * is compiled.
+   */
+  streamTransformers?: TStreamTransformers;
   /**
    * Wraps tool execution with custom logic. This allows you to:
    * - Modify tool call parameters before execution
@@ -414,7 +433,7 @@ type ResolveMiddlewareTypeConfig<T> = T extends {
  * Helper type to extract any property from a MiddlewareTypeConfig or AgentMiddleware.
  *
  * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from
- * @typeParam K - The property key to extract ("Schema" | "ContextSchema" | "FullContext" | "Tools")
+ * @typeParam K - The property key to extract ("Schema" | "ContextSchema" | "FullContext" | "Tools" | "StreamTransformers")
  */
 type InferMiddlewareType<T, K extends keyof MiddlewareTypeConfig> = ResolveMiddlewareTypeConfig<T>[K];
 /**
@@ -433,47 +452,51 @@ type InferMiddlewareFullContext<T> = InferMiddlewareType<T, "FullContext">;
  * Shorthand helper to extract the Tools type from a MiddlewareTypeConfig or AgentMiddleware.
  */
 type InferMiddlewareToolsFromConfig<T> = InferMiddlewareType<T, "Tools">;
+/**
+ * Shorthand helper to extract the StreamTransformers type from a MiddlewareTypeConfig or AgentMiddleware.
+ */
+type InferMiddlewareStreamTransformersFromConfig<T> = InferMiddlewareType<T, "StreamTransformers">;
 type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>["State"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};
 /**
  * Helper type to infer the state schema type from a middleware
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaValue<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaValue<TSchema>> : {} : {};
+type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaValue<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaValue<TSchema>> : {} : {};
 /**
  * Helper type to infer the input state schema type from a middleware (all properties optional)
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaUpdate<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};
+type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaUpdate<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (just the middleware states)
  */
-type InferMiddlewareStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};
+type InferMiddlewareStates<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};
 /**
  * Helper type to infer merged input state from an array of middleware (with optional defaults)
  */
-type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};
+type InferMiddlewareInputStates<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (includes built-in state)
  */
-type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;
+type InferMergedState<T extends readonly AnyAgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;
 /**
  * Helper type to infer merged input state from an array of middleware (includes built-in state)
  */
-type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;
+type InferMergedInputState<T extends readonly AnyAgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;
 /**
  * Helper type to infer the context schema type from a middleware
  */
-type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any> ? TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
+type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any, any> ? TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
 /**
  * Helper type to infer the input context schema type from a middleware (with optional defaults)
  */
-type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any> ? TContextSchema extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
+type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer TContextSchema, any, any, any> ? TContextSchema extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};
 /**
  * Helper type to infer merged context from an array of middleware
  */
-type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};
+type InferMiddlewareContexts<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};
 /**
  * Helper to merge two context types, preserving undefined unions
  */
@@ -481,14 +504,14 @@ type MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined]
 /**
  * Helper type to infer merged input context from an array of middleware (with optional defaults)
  */
-type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};
+type InferMiddlewareContextInputs<T extends readonly AnyAgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AnyAgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};
 /**
  * Helper type to extract input type from context schema (with optional defaults)
  */
 type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>["State"] : {};
 type ToAnnotationRoot<A extends StateDefinitionInit> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? InteropZodToStateDefinition<A> : never;
-type InferSchemaValue<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
-type InferSchemaInput<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodInput<A> : A extends AnyAnnotationRoot ? A["Update"] : {};
+type InferSchemaValue<A extends StateDefinitionInit | undefined> = [A] extends [undefined] ? {} : A extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
+type InferSchemaInput<A extends StateDefinitionInit | undefined> = [A] extends [undefined] ? {} : A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodInput<A> : A extends AnyAnnotationRoot ? A["Update"] : {};
 //#endregion
-export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, InferSchemaUpdateType, InferSchemaValue, InferSchemaValueType, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, NormalizedSchemaUpdate, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
+export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareStreamTransformersFromConfig, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, InferSchemaUpdateType, InferSchemaValue, InferSchemaValueType, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, NormalizedSchemaUpdate, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
 //# sourceMappingURL=types.d.ts.map

package/dist/agents/middleware/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA6BK,cAAA,MAAoB,CAAA,GAAI,OAAA,CAAQ,CAAA;AAAA,KAEzB,iBAAA,GAAoB,cAAA;;;;;;;;;;;;;;AAAhC;;;;;AA8BA;;;;;;;;;;UAAiB,oBAAA,iBACC,mBAAA,eACZ,mBAAA,qCAGA,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,gBAEnB,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,4DAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EANkB;EAUtB,MAAA,EAAQ,OAAA;EATe;EAWvB,aAAA,EAAe,cAAA;EARU;EAUzB,WAAA,EAAa,YAAA;EATT;EAWJ,KAAA,EAAO,MAAA;AAAA;;;;;KAOG,2BAAA,GAA8B,oBAAA;AAAA,KAE9B,oBAAA,aAAiC,OAAA,oBACzC,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,aAAkC,OAAA,oBAC1C,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,sBAAA,CAAuB,OAAA,IAAW,iBAAA,GAClC,OAAA,SAAgB,gBAAA,GACd,oBAAA,CAAqB,OAAA,IAAW,iBAAA,GAChC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,iBACM,mBAAA,8BACd,oBAAA,CAAqB,OAAA;AAAA,KAEb,sBAAA,iBACM,mBAAA,8BACd,qBAAA,CAAsB,OAAA;;;;KAKd,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EA/DzC;;;EAqEA,QAAA,EAAU,UAAA;EAnEN;;;;;;;;;;;AAiBN;;;;;AAEA;;;;EAqEE,IAAA,EAAM,UAAA,GAAa,UAAA;EAnEjB;;;EAuEF,KAAA,EAAO,MAAA,GAAS,iBAAA;EAtEqB;;;EA0ErC,OAAA,EAAS,SAAA,CAAQ,QAAA;AAAA;;;;;KAOP,eAAA,iBACM,MAAA,oBAA0B,iBAAA,yBAG1C,OAAA,EAAS,eAAA,CAAgB,OAAA,EAAS,QAAA,MAC/B,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;KAMtB,gBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACzD,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,MACtD,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;;;;KAStB,oBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,IAAA,CACP,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA;;;;;EAKzC,YAAA;EAAuB,aAAA,GAAgB,aAAA;AAAA,MAC1C,cAAA,CAAe,SAAA;;;;;;;AA/GpB;;;;;;;KA8HY,iBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACtD,OAAA,EAAS,oBAAA,CAAqB,OAAA,EAAS,QAAA,MACpC,cAAA,CAAe,SAAA,GAAY,OAAA;;;;;;;;;KAU3B,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;AAjLlB;;;;;;;;KA6LK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;AAjNnE;;;;KAwNY,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;AArNlB;;;cA6Na,gBAAA;;;;;;;;;;;;;;;;;;UAmBI,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EA9Ne;;;;;EAAA,SA4OV,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA;EAxOe;;;EA8OjB,IAAA;EAvOyB;;;;;;;EAgPzB,WAAA,GAAc,OAAA;EA3OkB;;;;;;;EAoPhC,aAAA,GAAgB,cAAA;EArPP;;;EA0PT,KAAA,GAAQ,MAAA;EAzPL;;;;;AAML;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAeA;;;;;;;;;;;;;;;;;;;;;;EAmSE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EAzRZ;;;;;;;AAgB/B;;;;;;;;;;;;;;;;;;;;;EAuSE,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EAnSW;;;;;;;;EA6StD,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA3SF;;AAAE;;;;;;EAqTvC,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAxS0B;;;;;;;;EAkTjE,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EApT9B;;;;;;;;EA8TP,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;;KAMlC,kBAAA,oBACS,CAAA,IAAK,CAAA,gCAAiC,CAAA,GAAI,CAAA,CAAE,CAAA;;;;;;KAQ9C,2BAAA,MAAiC,CAAA;EAC3C,kBAAA;AAAA,IAEE,KAAA,SAAc,oBAAA,GACZ,KAAA,WAEF,CAAA,SAAU,oBAAA,GACR,CAAA;;;;;;;KASM,mBAAA,oBAEM,oBAAA,IACd,2BAAA,CAA4B,CAAA,EAAG,CAAA;;;;KAKvB,qBAAA,MAA2B,mBAAA,CAAoB,CAAA;;;;KAK/C,4BAAA,MAAkC,mBAAA,CAC5C,CAAA;;AA1VI;;KAiWM,0BAAA,MAAgC,mBAAA,CAC1C,CAAA;;;;KAOU,8BAAA,MAAoC,mBAAA,CAAoB,CAAA;AAAA,KAExD,gBAAA,WAA2B,iBAAA,GAAoB,gBAAA,IACzD,CAAA,SAAU,iBAAA,GACN,gBAAA,CAAiB,CAAA,aACjB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA;;;;;;KAQjB,oBAAA,WAA+B,eAAA,IACzC,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;;KASpC,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,sBAAA,CAAuB,OAAA,KAC1C,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOpC,qBAAA,oBAAyC,eAAA,MACnD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,oBAAA,CAAqB,KAAA,IAAS,qBAAA,CAAsB,IAAA,IACpD,oBAAA,CAAqB,KAAA;;;;KAOrB,0BAAA,oBAA8C,eAAA,MACxD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,yBAAA,CAA0B,KAAA,IAAS,0BAAA,CAA2B,IAAA,IAC9D,yBAAA,CAA0B,KAAA;;AAtZtC;;KA6ZY,gBAAA,oBAAoC,eAAA,MAC9C,qBAAA,CAAsB,CAAA,IAAK,iBAAA;;;;KAKjB,qBAAA,oBAAyC,eAAA,MACnD,0BAAA,CAA2B,CAAA,IAAK,iBAAA;;;;KAKtB,sBAAA,WAAiC,eAAA,IAC3C,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOjB,2BAAA,WAAsC,eAAA,IAChD,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,kBAAA,gBACrB,oBAAA,CAAqB,KAAA,gBACrB,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOnB,uBAAA,oBAA2C,eAAA,MACrD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,sBAAA,CAAuB,KAAA,IAAS,uBAAA,CAAwB,IAAA,IACxD,sBAAA,CAAuB,KAAA;;;;KAO9B,iBAAA,UAA2B,CAAA,yBAC3B,CAAA,oCAEC,CAAA,gBACD,CAAA,wBACC,CAAA,gBACC,CAAA,WAAY,CAAA,IACX,CAAA,IACC,CAAA,WAAY,CAAA,IACX,CAAA,GACA,CAAA,GAAI,CAAA;;;;KAKF,4BAAA,oBAAgD,eAAA,MAC1D,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,iBAAA,CACE,2BAAA,CAA4B,KAAA,GAC5B,4BAAA,CAA6B,IAAA,KAE/B,2BAAA,CAA4B,KAAA;AA9dlC;;;AAAA,KAqeM,iBAAA,uBACY,iBAAA,GAAoB,gBAAA,IACxC,aAAA,SAAsB,gBAAA,GACtB,oBAAA,CAAqB,aAAA,IACrB,aAAA,SAAsB,iBAAA,GACpB,gBAAA,CAAiB,aAAA;AAAA,KAGX,gBAAA,WAA2B,mBAAA,IACrC,CAAA,SAAU,iBAAA,GACN,CAAA,GACA,CAAA,SAAU,gBAAA,GACR,2BAAA,CAA4B,CAAA;AAAA,KAGxB,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,qBAAA,CAAsB,OAAA,IACtB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA;AAAA,KAGE,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA,IACrB,CAAA,SAAU,iBAAA,GACR,CAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA6BK,cAAA,MAAoB,CAAA,GAAI,OAAA,CAAQ,CAAA;AAAA,KAEzB,iBAAA,GAAoB,cAAA;;;;;;;;;;;;;;AAAhC;;;;;AAiCA;;;;;;;;;;;;;UAAiB,oBAAA,iBACC,mBAAA,eACZ,mBAAA,qCAGA,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,gBAEnB,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,4DAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA,iCAEwB,aAAA,OAAoB,iBAAA,SAC9C,aAAA,OAAoB,iBAAA;EARlB;EAWJ,MAAA,EAAQ,OAAA;EAR8B;EAUtC,aAAA,EAAe,cAAA;EARX;EAUJ,WAAA,EAAa,YAAA;EARe;EAU5B,KAAA,EAAO,MAAA;EATL;EAWF,kBAAA,EAAoB,mBAAA;AAAA;;;;;KAOV,2BAAA,GAA8B,oBAAA;AAAA,KAE9B,oBAAA,aAAiC,OAAA,oBACzC,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,aAAkC,OAAA,oBAC1C,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,sBAAA,CAAuB,OAAA,IAAW,iBAAA,GAClC,OAAA,SAAgB,gBAAA,GACd,oBAAA,CAAqB,OAAA,IAAW,iBAAA,GAChC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,iBACM,mBAAA,8BACd,oBAAA,CAAqB,OAAA;AAAA,KAEb,sBAAA,iBACM,mBAAA,8BACd,qBAAA,CAAsB,OAAA;;;;KAKd,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EAnEzC;;;EAyEA,QAAA,EAAU,UAAA;EAvEN;;;;;;;;;;;;;;;;;;AAqBN;;EAuEE,IAAA,EAAM,UAAA,GAAa,UAAA;EAvEqB;;AAE1C;EAyEE,KAAA,EAAO,MAAA,GAAS,iBAAA;EAzEc;;;EA6E9B,OAAA,EAAS,SAAA,CAAQ,QAAA;AAAA;;;;;KAOP,eAAA,iBACM,MAAA,oBAA0B,iBAAA,yBAG1C,OAAA,EAAS,eAAA,CAAgB,OAAA,EAAS,QAAA,MAC/B,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;KAMtB,gBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACzD,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,MACtD,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;;;;KAStB,oBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,IAAA,CACP,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA;;;;;EAKzC,YAAA;EAAuB,aAAA,GAAgB,aAAA;AAAA,MAC1C,cAAA,CAAe,SAAA;;;;;;;;;;;;;;KAeR,iBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACtD,OAAA,EAAS,oBAAA,CAAqB,OAAA,EAAS,QAAA,MACpC,cAAA,CAAe,SAAA,GAAY,OAAA;;;;;;;;;KAU3B,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;AAjLlB;KA6LK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;AA1MlB;;;;;KAqNK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;;;cAQL,gBAAA;;;;AA7Nb;;KAoOY,kBAAA,GAAqB,eAAA,CAC/B,mBAAA,cACE,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,8BAGX,UAAA,GAAa,UAAA,KACvB,aAAA,OAAoB,iBAAA;;;;;;;;;;;;;;;;;;;;UAsBL,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA,iCAEwB,aAAA,OAAoB,iBAAA,SAC9C,aAAA,OAAoB,iBAAA;EA/OtB;;;;;EAAA,SA4PS,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA,EACA,mBAAA;EA7PuB;;AAO3B;EA4PE,IAAA;EA5PyB;;;;;;;EAqQzB,WAAA,GAAc,OAAA;EAhQX;;;;;;;EAyQH,aAAA,GAAgB,cAAA;EA1QS;;;EA+QzB,KAAA,GAAQ,MAAA;EA9QU;;;;AAMpB;EA+QE,kBAAA,GAAqB,mBAAA;EA/QK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAe5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgUE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EAtS/B;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoUV,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EA9TzB;;;;AAAqB;;;;EAwUvC,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA5TtB;;;;;;;;EAsUjB,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAxUjB;;;;;;;;EAkVtB,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EA/UnB;;;;;;AAOpB;;EAkVE,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;;KAMlC,kBAAA,oBACS,CAAA,IAAK,CAAA,gCAAiC,CAAA,GAAI,CAAA,CAAE,CAAA;;;;;;KAQ9C,2BAAA,MAAiC,CAAA;EAC3C,kBAAA;AAAA,IAEE,KAAA,SAAc,oBAAA,GACZ,KAAA,WAEF,CAAA,SAAU,oBAAA,GACR,CAAA;;;;;;;KASM,mBAAA,oBAEM,oBAAA,IACd,2BAAA,CAA4B,CAAA,EAAG,CAAA;;;AA5W7B;KAiXM,qBAAA,MAA2B,mBAAA,CAAoB,CAAA;;;;KAK/C,4BAAA,MAAkC,mBAAA,CAC5C,CAAA;;;;KAOU,0BAAA,MAAgC,mBAAA,CAC1C,CAAA;;;;KAOU,8BAAA,MAAoC,mBAAA,CAAoB,CAAA;;;;KAKxD,2CAAA,MACV,mBAAA,CAAoB,CAAA;AAAA,KAEV,gBAAA,WAA2B,iBAAA,GAAoB,gBAAA,IACzD,CAAA,SAAU,iBAAA,GACN,gBAAA,CAAiB,CAAA,aACjB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA;;;;;;KAQjB,oBAAA,WAA+B,eAAA,IACzC,CAAA,SAAU,eAAA,sCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;AA7YhD;KAsZY,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,sCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,sBAAA,CAAuB,OAAA,KAC1C,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOpC,qBAAA,oBAAyC,kBAAA,MACnD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,oBAAA,CAAqB,KAAA,IAAS,qBAAA,CAAsB,IAAA,IACpD,oBAAA,CAAqB,KAAA;;;;KAOrB,0BAAA,oBACS,kBAAA,MACjB,CAAA,4BAEA,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,yBAAA,CAA0B,KAAA,IAAS,0BAAA,CAA2B,IAAA,IAC9D,yBAAA,CAA0B,KAAA;;;;KAOxB,gBAAA,oBAAoC,kBAAA,MAC9C,qBAAA,CAAsB,CAAA,IAAK,iBAAA;;;;KAKjB,qBAAA,oBAAyC,kBAAA,MACnD,0BAAA,CAA2B,CAAA,IAAK,iBAAA;;;;KAKtB,sBAAA,WAAiC,eAAA,IAC3C,CAAA,SAAU,eAAA,6CACN,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOjB,2BAAA,WAAsC,eAAA,IAChD,CAAA,SAAU,eAAA,6CACN,cAAA,SAAuB,kBAAA,gBACrB,oBAAA,CAAqB,KAAA,gBACrB,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;AApdzB;;KA2dM,uBAAA,oBAA2C,kBAAA,MACrD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,sBAAA,CAAuB,KAAA,IAAS,uBAAA,CAAwB,IAAA,IACxD,sBAAA,CAAuB,KAAA;;;;KAO9B,iBAAA,UAA2B,CAAA,yBAC3B,CAAA,oCAEC,CAAA,gBACD,CAAA,wBACC,CAAA,gBACC,CAAA,WAAY,CAAA,IACX,CAAA,IACC,CAAA,WAAY,CAAA,IACX,CAAA,GACA,CAAA,GAAI,CAAA;;;;KAKF,4BAAA,oBACS,kBAAA,MACjB,CAAA,4BAEA,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,kBAAA,KACpB,iBAAA,CACE,2BAAA,CAA4B,KAAA,GAC5B,4BAAA,CAA6B,IAAA,KAE/B,2BAAA,CAA4B,KAAA;;;;KAO1B,iBAAA,uBACY,iBAAA,GAAoB,gBAAA,IACxC,aAAA,SAAsB,gBAAA,GACtB,oBAAA,CAAqB,aAAA,IACrB,aAAA,SAAsB,iBAAA,GACpB,gBAAA,CAAiB,aAAA;AAAA,KAGX,gBAAA,WAA2B,mBAAA,IACrC,CAAA,SAAU,iBAAA,GACN,CAAA,GACA,CAAA,SAAU,gBAAA,GACR,2BAAA,CAA4B,CAAA;AAAA,KAGxB,gBAAA,WAA2B,mBAAA,iBACrC,CAAA,6BAGE,CAAA,SAAU,WAAA,kBACR,qBAAA,CAAsB,OAAA,IACtB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA;AAAA,KAGE,gBAAA,WAA2B,mBAAA,iBACrC,CAAA,6BAGE,CAAA,SAAU,WAAA,kBACR,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA,IACrB,CAAA,SAAU,iBAAA,GACR,CAAA"}