langchain 1.0.0-alpha.7 → 1.0.0-alpha.8

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

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.ts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","AgentMiddleware","Runtime","MiddlewareResult","AgentBuiltInState","ModelRequest","JumpToTarget","ClientTool","ServerTool","createMiddleware","TSchema","TContextSchema","Partial","Promise"],"sources":["../../../src/agents/middlewareAgent/middleware.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { AgentMiddleware, Runtime, MiddlewareResult, AgentBuiltInState, ModelRequest, JumpToTarget } from \"./types.js\";\nimport type { ClientTool, ServerTool } from \"../types.js\";\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.modifyModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport declare function createMiddleware<TSchema extends InteropZodObject | undefined = undefined, TContextSchema extends InteropZodObject | InteropZodOptional<InteropZodObject> | InteropZodDefault<InteropZodObject> | undefined = undefined>(config: {\n    /**\n     * The name of the middleware\n     */\n    name: string;\n    /**\n     * The schema of the middleware state. Middleware state is 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    stateSchema?: TSchema;\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     * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n     */\n    beforeModelJumpTo?: JumpToTarget[];\n    /**\n     * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n     */\n    afterModelJumpTo?: JumpToTarget[];\n    /**\n     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n     * It allows to modify the model request before it is passed to the model.\n     *\n     * @param request - The model request\n     * @param request.model - The model to use for this step.\n     * @param request.messages - The messages to send to the model.\n     * @param request.systemPrompt - The system message for this step.\n     * @param request.toolChoice - The tool choice configuration for this step.\n     * @param request.tools - The tools to make available for this step.\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified model request or undefined to pass through\n     */\n    modifyModelRequest?: (options: ModelRequest, state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<(TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<ModelRequest | void> | ModelRequest | void;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<(TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>;\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?: (state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<(TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>;\n}): AgentMiddleware<TSchema, TContextSchema, any>;\n"],"mappings":";;;;;;;AAkCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CqVe,iBA/C7TF,gBA+C6TE,CAAAA,gBA/C5Rf,gBA+C4Re,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,uBA/C3Nf,gBA+C2Ne,GA/CxMb,kBA+CwMa,CA/CrLf,gBA+CqLe,CAAAA,GA/CjKd,iBA+CiKc,CA/C/If,gBA+C+Ie,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,CAAAA,CAAAA,MAAAA,EAAAA;EAAc;;;EAAuD,IAAmBA,EAAAA,MAAAA;EAAc;;;;;;;EAAsG,WAAwBN,CAAAA,EAnCviBK,OAmCuiBL;EAAY;;;;;;;EAS5Z,aAAwBK,CAAAA,EApC7KC,cAoC6KD;EAAO;;;EAA2C,iBAASd,CAAAA,EAhCpOU,YAgCoOV,EAAAA;EAAgB;;;EAAyD,gBAASC,CAAAA,EA5BvTS,YA4BuTT,EAAAA;EAAiB;;;EAA8D,KAASC,CAAAA,EAAAA,CAxBzZS,UAwByZT,GAxB5YU,UAwB4YV,CAAAA,EAAAA;EAAkB;;;;;;;;;;;;;;EAAwQ,kBAA5BC,CAAAA,EAAAA,CAAAA,OAAAA,EATjoBM,YASioBN,EAAAA,KAAAA,EAAAA,CAT3mBW,OAS2mBX,SAT3lBH,gBAS2lBG,GATxkBA,oBASwkBA,CATnjBW,OASmjBX,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GATliBK,iBASkiBL,EAAAA,OAAAA,EATtgBG,OASsgBH,CAAAA,CAT7fW,OAS6fX,SAT7eH,gBAS6eG,GAT1dA,oBAS0dA,CATrcW,OASqcX,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GATpbK,iBASobL,EATjaY,cASiaZ,SAT1YH,gBAS0YG,GATvXC,qBASuXD,CATjWY,cASiWZ,CAAAA,GAT/UY,cAS+UZ,SATxTF,iBASwTE,CAAAA,GAAAA,CAAAA,GAT/RC,qBAS+RD,CATzQY,cASyQZ,CAAAA,GATvPY,cASuPZ,SAThOD,kBASgOC,CAAAA,GAAAA,CAAAA,GATtMa,OASsMb,CAT9LC,qBAS8LD,CATxKY,cASwKZ,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GAT1Ic,OAS0Id,CATlIM,YASkIN,GAAAA,IAAAA,CAAAA,GAT3GM,YAS2GN,GAAAA,IAAAA;EAAoB;;;;;;;;EASziB,WAASH,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAT7Hc,OAS6Hd,SAT7GA,gBAS6GA,GAT1FG,oBAS0FH,CATrEc,OASqEd,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GATpDQ,iBASoDR,EAAAA,OAAAA,EATxBM,OASwBN,CAAAA,CATfc,OASed,SATCA,gBASDA,GAToBG,oBASpBH,CATyCc,OASzCd,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAT0DQ,iBAS1DR,EAT6Ee,cAS7Ef,SAToGA,gBASpGA,GATuHI,qBASvHJ,CAT6Ie,cAS7If,CAAAA,GAT+Je,cAS/Jf,SATsLC,iBAStLD,CAAAA,GAAAA,CAAAA,GAT+MI,qBAS/MJ,CATqOe,cASrOf,CAAAA,GATuPe,cASvPf,SAT8QE,kBAS9QF,CAAAA,GAAAA,CAAAA,GATwSgB,OASxShB,CATgTI,qBAShTJ,CATsUe,cAStUf,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GAToWiB,OASpWjB,CAT4WO,gBAS5WP,CAT6XgB,OAS7XhB,CATqYc,OASrYd,SATqZA,gBASrZA,GATwaG,oBASxaH,CAT6bc,OAS7bd,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GATgdO,gBAShdP,CATiegB,OASjehB,CATyec,OASzed,SATyfA,gBASzfA,GAT4gBG,oBAS5gBH,CATiiBc,OASjiBd,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;EAAgB;;;;;;;;EAA4J,UAASC,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAAnTa,OAAmTb,SAAnSD,gBAAmSC,GAAhRE,oBAAgRF,CAA3Pa,OAA2Pb,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAA1OO,iBAA0OP,EAAAA,OAAAA,EAA9MK,OAA8ML,CAAAA,CAArMa,OAAqMb,SAArLD,gBAAqLC,GAAlKE,oBAAkKF,CAA7Ia,OAA6Ib,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAA5HO,iBAA4HP,EAAzGc,cAAyGd,SAAlFD,gBAAkFC,GAA/DG,qBAA+DH,CAAzCc,cAAyCd,CAAAA,GAAvBc,cAAuBd,SAAAA,iBAAAA,CAAAA,GAAAA,CAAAA,GAAyBG,qBAAzBH,CAA+Cc,cAA/Cd,CAAAA,GAAiEc,cAAjEd,SAAwFC,kBAAxFD,CAAAA,GAAAA,CAAAA,GAAkHe,OAAlHf,CAA0HG,qBAA1HH,CAAgJc,cAAhJd,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GAA8KgB,OAA9KhB,CAAsLM,gBAAtLN,CAAuMe,OAAvMf,CAA+Ma,OAA/Mb,SAA+ND,gBAA/NC,GAAkPE,oBAAlPF,CAAuQa,OAAvQb,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAA0RM,gBAA1RN,CAA2Se,OAA3Sf,CAAmTa,OAAnTb,SAAmUD,gBAAnUC,GAAsVE,oBAAtVF,CAA2Wa,OAA3Wb,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;CAAiB,CAAA,EAC1VI,eADwXU,CACxWD,OADwWC,EAC/VA,cAD+VA,EAAAA,GAAAA,CAAAA"}
+{"version":3,"file":"middleware.d.ts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","AgentMiddleware","Runtime","MiddlewareResult","AgentBuiltInState","ModelRequest","JumpToTarget","ClientTool","ServerTool","createMiddleware","TSchema","TContextSchema","Partial","Promise","Error"],"sources":["../../../src/agents/middlewareAgent/middleware.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { AgentMiddleware, Runtime, MiddlewareResult, AgentBuiltInState, ModelRequest, JumpToTarget } from \"./types.js\";\nimport type { ClientTool, ServerTool } from \"../types.js\";\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.modifyModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport declare function createMiddleware<TSchema extends InteropZodObject | undefined = undefined, TContextSchema extends InteropZodObject | InteropZodOptional<InteropZodObject> | InteropZodDefault<InteropZodObject> | undefined = undefined>(config: {\n    /**\n     * The name of the middleware\n     */\n    name: string;\n    /**\n     * The schema of the middleware state. Middleware state is 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    stateSchema?: TSchema;\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     * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n     */\n    beforeModelJumpTo?: JumpToTarget[];\n    /**\n     * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n     */\n    afterModelJumpTo?: JumpToTarget[];\n    /**\n     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n     * It allows to modify the model request before it is passed to the model.\n     *\n     * @param request - The model request\n     * @param request.model - The model to use for this step.\n     * @param request.messages - The messages to send to the model.\n     * @param request.systemPrompt - The system message for this step.\n     * @param request.toolChoice - The tool choice configuration for this step.\n     * @param request.tools - The tools to make available for this step.\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified model request or undefined to pass through\n     */\n    modifyModelRequest?: (options: ModelRequest, state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<ModelRequest | void> | ModelRequest | void;\n    /**\n     * The function to handle model invocation errors and optionally retry.\n     *\n     * @param error - The exception that occurred during model invocation\n     * @param request - The original model request that failed\n     * @param state - The current agent state\n     * @param runtime - The runtime context\n     * @param attempt - The current attempt number (1-indexed)\n     * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise)\n     */\n    retryModelRequest?: (error: Error, request: ModelRequest, state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>, attempt: number) => Promise<ModelRequest | void> | ModelRequest | void;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>;\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?: (state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends InteropZodObject ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodDefault<any> ? InferInteropZodOutput<TContextSchema> : TContextSchema extends InteropZodOptional<any> ? Partial<InferInteropZodOutput<TContextSchema>> : never>) => Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>;\n}): AgentMiddleware<TSchema, TContextSchema, any>;\n"],"mappings":";;;;;;;AAkCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+CuWE,iBA/C/UW,gBA+C+UX,CAAAA,gBA/C9SF,gBA+C8SE,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,uBA/C7OF,gBA+C6OE,GA/C1NA,kBA+C0NA,CA/CvMF,gBA+CuME,CAAAA,GA/CnLD,iBA+CmLC,CA/CjKF,gBA+CiKE,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,CAAAA,CAAAA,MAAAA,EAAAA;EAAkB;;;EAAe,IAA1OI,EAAAA,MAAAA;EAAO;;;;;;;EAW/D,WAAwBQ,CAAAA,EA9C5GA,OA8C4GA;EAAO;;;;;;;EAA8I,aAASb,CAAAA,EAtCxQc,cAsCwQd;EAAiB;;;EAA8D,iBAASC,CAAAA,EAlC5VQ,YAkC4VR,EAAAA;EAAkB;;;EAAe,gBAA1OI,CAAAA,EA9BpJI,YA8BoJJ,EAAAA;EAAO;;;EAAoV,KAS3eQ,CAAAA,EAAAA,CAnCdH,UAmCcG,GAnCDF,UAmCCE,CAAAA,EAAAA;EAAO;;;;;;;;;;;;;;EAAyT,kBAAsCC,CAAAA,EAAAA,CAAAA,OAAAA,EApB9VN,YAoB8VM,EAAAA,KAAAA,EAAAA,CApBxUD,OAoBwUC,SApBxTf,gBAoBwTe,GApBrSZ,oBAoBqSY,CApBhRD,OAoBgRC,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GApB/PP,iBAoB+PO,EAAAA,OAAAA,EApBnOT,OAoBmOS,CApB3NA,cAoB2NA,SApBpMf,gBAoBoMe,GApBjLX,qBAoBiLW,CApB3JA,cAoB2JA,CAAAA,GApBzIA,cAoByIA,SApBlHd,iBAoBkHc,CAAAA,GAAAA,CAAAA,GApBzFX,qBAoByFW,CApBnEA,cAoBmEA,CAAAA,GApBjDA,cAoBiDA,SApB1Bb,kBAoB0Ba,CAAAA,GAAAA,CAAAA,GApBAC,OAoBAD,CApBQX,qBAoBRW,CApB8BA,cAoB9BA,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GApB4DE,OAoB5DF,CApBoEN,YAoBpEM,GAAAA,IAAAA,CAAAA,GApB2FN,YAoB3FM,GAAAA,IAAAA;EAAc;;;;;;;;;;EAAuB,iBAA8HD,CAAAA,EAAAA,CAAAA,KAAAA,EATpgBI,KASogBJ,EAAAA,OAAAA,EATpfL,YASofK,EAAAA,KAAAA,EAAAA,CAT9dA,OAS8dA,SAT9cd,gBAS8cc,GAT3bX,oBAS2bW,CATtaA,OASsaA,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GATrZN,iBASqZM,EAAAA,OAAAA,EATzXR,OASyXQ,CATjXC,cASiXD,SAT1Vd,gBAS0Vc,GATvUV,qBASuUU,CATjTC,cASiTD,CAAAA,GAT/RC,cAS+RD,SATxQb,iBASwQa,CAAAA,GAAAA,CAAAA,GAT/OV,qBAS+OU,CATzNC,cASyND,CAAAA,GATvMC,cASuMD,SAThLZ,kBASgLY,CAAAA,GAAAA,CAAAA,GATtJE,OASsJF,CAT9IV,qBAS8IU,CATxHC,cASwHD,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,OAAAA,EAAAA,MAAAA,EAAAA,GATzEG,OASyEH,CATjEL,YASiEK,GAAAA,IAAAA,CAAAA,GAT1CL,YAS0CK,GAAAA,IAAAA;EAAO;;;;;;;;EASld,WAA5BX,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CATlCW,OASkCX,SATlBH,gBASkBG,GATCA,oBASDA,CATsBW,OAStBX,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GATuCK,iBASvCL,EAAAA,OAAAA,EATmEG,OASnEH,CAT2EY,cAS3EZ,SATkGH,gBASlGG,GATqHC,qBASrHD,CAT2IY,cAS3IZ,CAAAA,GAT6JY,cAS7JZ,SAToLF,iBASpLE,CAAAA,GAAAA,CAAAA,GAT6MC,qBAS7MD,CATmOY,cASnOZ,CAAAA,GATqPY,cASrPZ,SAT4QD,kBAS5QC,CAAAA,GAAAA,CAAAA,GATsSa,OAStSb,CAT8SC,qBAS9SD,CAToUY,cASpUZ,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GATkWc,OASlWd,CAT0WI,gBAS1WJ,CAT2Xa,OAS3Xb,CATmYW,OASnYX,SATmZH,gBASnZG,GATsaA,oBAStaA,CAT2bW,OAS3bX,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAT8cI,gBAS9cJ,CAT+da,OAS/db,CATueW,OASveX,SATufH,gBASvfG,GAT0gBA,oBAS1gBA,CAT+hBW,OAS/hBX,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;EAAoB;;;;;;;;EAA4N,UAApCC,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAA/OU,OAA+OV,SAA/NJ,gBAA+NI,GAA5MD,oBAA4MC,CAAvLU,OAAuLV,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAAtKI,iBAAsKJ,EAAAA,OAAAA,EAA1IE,OAA0IF,CAAlIW,cAAkIX,SAA3GJ,gBAA2GI,GAAxFA,qBAAwFA,CAAlEW,cAAkEX,CAAAA,GAAhDW,cAAgDX,SAAzBH,iBAAyBG,CAAAA,GAAAA,CAAAA,GAAAA,qBAAAA,CAAsBW,cAAtBX,CAAAA,GAAwCW,cAAxCX,SAA+DF,kBAA/DE,CAAAA,GAAAA,CAAAA,GAAyFY,OAAzFZ,CAAiGA,qBAAjGA,CAAuHW,cAAvHX,CAAAA,CAAAA,GAAAA,KAAAA,CAAAA,EAAAA,GAAqJa,OAArJb,CAA6JG,gBAA7JH,CAA8KY,OAA9KZ,CAAsLU,OAAtLV,SAAsMJ,gBAAtMI,GAAyND,oBAAzNC,CAA8OU,OAA9OV,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAAiQG,gBAAjQH,CAAkRY,OAAlRZ,CAA0RU,OAA1RV,SAA0SJ,gBAA1SI,GAA6TD,oBAA7TC,CAAkVU,OAAlVV,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;CAAqB,CAAA,EAC1RC,eAD6SU,CAC7RD,OAD6RC,EACpRA,cADoRA,EAAAA,GAAAA,CAAAA"}

package/dist/agents/middlewareAgent/middleware.js

@@ -39,6 +39,7 @@ function createMiddleware(config) {
 		tools: config.tools ?? []
 	};
 	if (config.modifyModelRequest) middleware.modifyModelRequest = async (options, state, runtime) => Promise.resolve(config.modifyModelRequest(options, state, runtime));
+	if (config.retryModelRequest) middleware.retryModelRequest = async (error, request, state, runtime, attempt) => Promise.resolve(config.retryModelRequest(error, request, state, runtime, attempt));
 	if (config.beforeModel) middleware.beforeModel = async (state, runtime) => Promise.resolve(config.beforeModel(state, runtime));
 	if (config.afterModel) middleware.afterModel = async (state, runtime) => Promise.resolve(config.afterModel(state, runtime));
 	return middleware;

package/dist/agents/middlewareAgent/middleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.js","names":["config: {\n  /**\n   * The name of the middleware\n   */\n  name: string;\n  /**\n   * The schema of the middleware state. Middleware state is 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  stateSchema?: TSchema;\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   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n   */\n  beforeModelJumpTo?: JumpToTarget[];\n  /**\n   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n   */\n  afterModelJumpTo?: JumpToTarget[];\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: (ClientTool | ServerTool)[];\n  /**\n   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n   * It allows to modify the model request before it is passed to the model.\n   *\n   * @param request - The model request\n   * @param request.model - The model to use for this step.\n   * @param request.messages - The messages to send to the model.\n   * @param request.systemPrompt - The system message for this step.\n   * @param request.toolChoice - The tool choice configuration for this step.\n   * @param request.tools - The tools to make available for this step.\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified model request or undefined to pass through\n   */\n  modifyModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\n      >;\n}","middleware: AgentMiddleware<TSchema, TContextSchema, any>"],"sources":["../../../src/agents/middlewareAgent/middleware.ts"],"sourcesContent":["/* eslint-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 {\n  AgentMiddleware,\n  Runtime,\n  MiddlewareResult,\n  AgentBuiltInState,\n  ModelRequest,\n  JumpToTarget,\n} from \"./types.js\";\nimport type { ClientTool, ServerTool } from \"../types.js\";\n\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.modifyModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport function createMiddleware<\n  TSchema extends InteropZodObject | undefined = undefined,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodOptional<InteropZodObject>\n    | InteropZodDefault<InteropZodObject>\n    | undefined = undefined\n>(config: {\n  /**\n   * The name of the middleware\n   */\n  name: string;\n  /**\n   * The schema of the middleware state. Middleware state is 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  stateSchema?: TSchema;\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   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n   */\n  beforeModelJumpTo?: JumpToTarget[];\n  /**\n   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n   */\n  afterModelJumpTo?: JumpToTarget[];\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: (ClientTool | ServerTool)[];\n  /**\n   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n   * It allows to modify the model request before it is passed to the model.\n   *\n   * @param request - The model request\n   * @param request.model - The model to use for this step.\n   * @param request.messages - The messages to send to the model.\n   * @param request.systemPrompt - The system message for this step.\n   * @param request.toolChoice - The tool choice configuration for this step.\n   * @param request.tools - The tools to make available for this step.\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified model request or undefined to pass through\n   */\n  modifyModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) &\n        AgentBuiltInState,\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\n      >;\n}): AgentMiddleware<TSchema, TContextSchema, any> {\n  const middleware: AgentMiddleware<TSchema, TContextSchema, any> = {\n    name: config.name,\n    stateSchema: config.stateSchema,\n    contextSchema: config.contextSchema,\n    beforeModelJumpTo: config.beforeModelJumpTo,\n    afterModelJumpTo: config.afterModelJumpTo,\n    tools: config.tools ?? [],\n  };\n\n  if (config.modifyModelRequest) {\n    middleware.modifyModelRequest = async (options, state, runtime) =>\n      Promise.resolve(\n        config.modifyModelRequest!(\n          options,\n          state,\n          runtime as Runtime<\n            (TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}) &\n              AgentBuiltInState,\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  if (config.beforeModel) {\n    middleware.beforeModel = async (state, runtime) =>\n      Promise.resolve(\n        config.beforeModel!(\n          state,\n          runtime as Runtime<\n            (TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}) &\n              AgentBuiltInState,\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  if (config.afterModel) {\n    middleware.afterModel = async (state, runtime) =>\n      Promise.resolve(\n        config.afterModel!(\n          state,\n          runtime as Runtime<\n            (TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}) &\n              AgentBuiltInState,\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  return middleware;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDA,SAAgB,iBAOdA,QA+IgD;CAChD,MAAMC,aAA4D;EAChE,MAAM,OAAO;EACb,aAAa,OAAO;EACpB,eAAe,OAAO;EACtB,mBAAmB,OAAO;EAC1B,kBAAkB,OAAO;EACzB,OAAO,OAAO,SAAS,CAAE;CAC1B;AAED,KAAI,OAAO,oBACT,WAAW,qBAAqB,OAAO,SAAS,OAAO,YACrD,QAAQ,QACN,OAAO,mBACL,SACA,OACA,QAaD,CACF;AAGL,KAAI,OAAO,aACT,WAAW,cAAc,OAAO,OAAO,YACrC,QAAQ,QACN,OAAO,YACL,OACA,QAaD,CACF;AAGL,KAAI,OAAO,YACT,WAAW,aAAa,OAAO,OAAO,YACpC,QAAQ,QACN,OAAO,WACL,OACA,QAaD,CACF;AAGL,QAAO;AACR"}
+{"version":3,"file":"middleware.js","names":["config: {\n  /**\n   * The name of the middleware\n   */\n  name: string;\n  /**\n   * The schema of the middleware state. Middleware state is 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  stateSchema?: TSchema;\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   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n   */\n  beforeModelJumpTo?: JumpToTarget[];\n  /**\n   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n   */\n  afterModelJumpTo?: JumpToTarget[];\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: (ClientTool | ServerTool)[];\n  /**\n   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n   * It allows to modify the model request before it is passed to the model.\n   *\n   * @param request - The model request\n   * @param request.model - The model to use for this step.\n   * @param request.messages - The messages to send to the model.\n   * @param request.systemPrompt - The system message for this step.\n   * @param request.toolChoice - The tool choice configuration for this step.\n   * @param request.tools - The tools to make available for this step.\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified model request or undefined to pass through\n   */\n  modifyModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to handle model invocation errors and optionally retry.\n   *\n   * @param error - The exception that occurred during model invocation\n   * @param request - The original model request that failed\n   * @param state - The current agent state\n   * @param runtime - The runtime context\n   * @param attempt - The current attempt number (1-indexed)\n   * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise)\n   */\n  retryModelRequest?: (\n    error: Error,\n    request: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >,\n    attempt: number\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\n      >;\n}","middleware: AgentMiddleware<TSchema, TContextSchema, any>"],"sources":["../../../src/agents/middlewareAgent/middleware.ts"],"sourcesContent":["/* eslint-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 {\n  AgentMiddleware,\n  Runtime,\n  MiddlewareResult,\n  AgentBuiltInState,\n  ModelRequest,\n  JumpToTarget,\n} from \"./types.js\";\nimport type { ClientTool, ServerTool } from \"../types.js\";\n\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.modifyModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport function createMiddleware<\n  TSchema extends InteropZodObject | undefined = undefined,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodOptional<InteropZodObject>\n    | InteropZodDefault<InteropZodObject>\n    | undefined = undefined\n>(config: {\n  /**\n   * The name of the middleware\n   */\n  name: string;\n  /**\n   * The schema of the middleware state. Middleware state is 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  stateSchema?: TSchema;\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   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.\n   */\n  beforeModelJumpTo?: JumpToTarget[];\n  /**\n   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.\n   */\n  afterModelJumpTo?: JumpToTarget[];\n  /**\n   * Additional tools registered by the middleware.\n   */\n  tools?: (ClientTool | ServerTool)[];\n  /**\n   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.\n   * It allows to modify the model request before it is passed to the model.\n   *\n   * @param request - The model request\n   * @param request.model - The model to use for this step.\n   * @param request.messages - The messages to send to the model.\n   * @param request.systemPrompt - The system message for this step.\n   * @param request.toolChoice - The tool choice configuration for this step.\n   * @param request.tools - The tools to make available for this step.\n   * @param state - The middleware state\n   * @param runtime - The middleware runtime\n   * @returns The modified model request or undefined to pass through\n   */\n  modifyModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to handle model invocation errors and optionally retry.\n   *\n   * @param error - The exception that occurred during model invocation\n   * @param request - The original model request that failed\n   * @param state - The current agent state\n   * @param runtime - The runtime context\n   * @param attempt - The current attempt number (1-indexed)\n   * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise)\n   */\n  retryModelRequest?: (\n    error: Error,\n    request: ModelRequest,\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >,\n    attempt: number\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  /**\n   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` 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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\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?: (\n    state: (TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema>\n      : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends InteropZodObject\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodDefault<any>\n        ? InferInteropZodOutput<TContextSchema>\n        : TContextSchema extends InteropZodOptional<any>\n        ? Partial<InferInteropZodOutput<TContextSchema>>\n        : never\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<\n            TSchema extends InteropZodObject\n              ? InferInteropZodInput<TSchema>\n              : {}\n          >\n        >\n      >\n    | MiddlewareResult<\n        Partial<\n          TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}\n        >\n      >;\n}): AgentMiddleware<TSchema, TContextSchema, any> {\n  const middleware: AgentMiddleware<TSchema, TContextSchema, any> = {\n    name: config.name,\n    stateSchema: config.stateSchema,\n    contextSchema: config.contextSchema,\n    beforeModelJumpTo: config.beforeModelJumpTo,\n    afterModelJumpTo: config.afterModelJumpTo,\n    tools: config.tools ?? [],\n  };\n\n  if (config.modifyModelRequest) {\n    middleware.modifyModelRequest = async (options, state, runtime) =>\n      Promise.resolve(\n        config.modifyModelRequest!(\n          options,\n          state,\n          runtime as Runtime<\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  if (config.retryModelRequest) {\n    middleware.retryModelRequest = async (\n      error,\n      request,\n      state,\n      runtime,\n      attempt\n    ) =>\n      Promise.resolve(\n        config.retryModelRequest!(\n          error,\n          request,\n          state,\n          runtime as Runtime<\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >,\n          attempt\n        )\n      );\n  }\n\n  if (config.beforeModel) {\n    middleware.beforeModel = async (state, runtime) =>\n      Promise.resolve(\n        config.beforeModel!(\n          state,\n          runtime as Runtime<\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  if (config.afterModel) {\n    middleware.afterModel = async (state, runtime) =>\n      Promise.resolve(\n        config.afterModel!(\n          state,\n          runtime as Runtime<\n            TContextSchema extends InteropZodObject\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodDefault<any>\n              ? InferInteropZodOutput<TContextSchema>\n              : TContextSchema extends InteropZodOptional<any>\n              ? Partial<InferInteropZodOutput<TContextSchema>>\n              : never\n          >\n        )\n      );\n  }\n\n  return middleware;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDA,SAAgB,iBAOdA,QAqKgD;CAChD,MAAMC,aAA4D;EAChE,MAAM,OAAO;EACb,aAAa,OAAO;EACpB,eAAe,OAAO;EACtB,mBAAmB,OAAO;EAC1B,kBAAkB,OAAO;EACzB,OAAO,OAAO,SAAS,CAAE;CAC1B;AAED,KAAI,OAAO,oBACT,WAAW,qBAAqB,OAAO,SAAS,OAAO,YACrD,QAAQ,QACN,OAAO,mBACL,SACA,OACA,QASD,CACF;AAGL,KAAI,OAAO,mBACT,WAAW,oBAAoB,OAC7B,OACA,SACA,OACA,SACA,YAEA,QAAQ,QACN,OAAO,kBACL,OACA,SACA,OACA,SASA,QACD,CACF;AAGL,KAAI,OAAO,aACT,WAAW,cAAc,OAAO,OAAO,YACrC,QAAQ,QACN,OAAO,YACL,OACA,QASD,CACF;AAGL,KAAI,OAAO,YACT,WAAW,aAAa,OAAO,OAAO,YACpC,QAAQ,QACN,OAAO,WACL,OACA,QASD,CACF;AAGL,QAAO;AACR"}

package/dist/agents/middlewareAgent/nodes/AfterModalNode.cjs

@@ -11,11 +11,11 @@ var AfterModelNode = class extends require_middleware.MiddlewareNode {
 		"afterModalNodes"
 	];
 	name;
-	constructor(middleware) {
+	constructor(middleware, options) {
 		super({
 			name: `AfterModelNode_${middleware.name}`,
 			func: async (state, config) => this.invokeMiddleware(state, config)
-		});
+		}, options);
 		this.middleware = middleware;
 		this.name = `AfterModelNode_${middleware.name}`;
 	}

package/dist/agents/middlewareAgent/nodes/AfterModalNode.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"AfterModalNode.cjs","names":["MiddlewareNode","middleware: AgentMiddleware<any, any, any>","state: TStateSchema","config?: LangGraphRunnableConfig","runtime: Runtime<TStateSchema, TContextSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode } from \"./middleware.js\";\nimport type {\n  AgentBuiltInState,\n  AgentMiddleware,\n  MiddlewareResult,\n  Runtime,\n} from \"../types.js\";\n\n/**\n * Node for executing a single middleware's afterModel hook.\n */\nexport class AfterModelNode<\n  TStateSchema extends Record<string, any> = Record<string, any>,\n  TContextSchema extends Record<string, any> = Record<string, any>\n> extends MiddlewareNode<TStateSchema, TContextSchema> {\n  lc_namespace = [\"langchain\", \"agents\", \"afterModalNodes\"];\n\n  name: string;\n\n  constructor(public middleware: AgentMiddleware<any, any, any>) {\n    super({\n      name: `AfterModelNode_${middleware.name}`,\n      func: async (state: TStateSchema, config?: LangGraphRunnableConfig) =>\n        this.invokeMiddleware(state, config),\n    });\n    this.name = `AfterModelNode_${middleware.name}`;\n  }\n\n  runHook(state: TStateSchema, runtime: Runtime<TStateSchema, TContextSchema>) {\n    return this.middleware.afterModel!(\n      state as Record<string, any> & AgentBuiltInState,\n      runtime as Runtime<TStateSchema, unknown>\n    ) as Promise<MiddlewareResult<TStateSchema>>;\n  }\n}\n"],"mappings":";;;;;;AAaA,IAAa,iBAAb,cAGUA,kCAA6C;CACrD,eAAe;EAAC;EAAa;EAAU;CAAkB;CAEzD;CAEA,YAAmBC,YAA4C;EAC7D,MAAM;GACJ,MAAM,CAAC,eAAe,EAAE,WAAW,MAAM;GACzC,MAAM,OAAOC,OAAqBC,WAChC,KAAK,iBAAiB,OAAO,OAAO;EACvC,EAAC;EALe;EAMjB,KAAK,OAAO,CAAC,eAAe,EAAE,WAAW,MAAM;CAChD;CAED,QAAQD,OAAqBE,SAAgD;AAC3E,SAAO,KAAK,WAAW,WACrB,OACA,QACD;CACF;AACF"}
+{"version":3,"file":"AfterModalNode.cjs","names":["MiddlewareNode","middleware: AgentMiddleware<any, any, any>","options: MiddlewareNodeOptions","state: TStateSchema","config?: LangGraphRunnableConfig","runtime: Runtime<TContextSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode, MiddlewareNodeOptions } from \"./middleware.js\";\nimport type {\n  AgentBuiltInState,\n  AgentMiddleware,\n  MiddlewareResult,\n  Runtime,\n} from \"../types.js\";\n\n/**\n * Node for executing a single middleware's afterModel hook.\n */\nexport class AfterModelNode<\n  TStateSchema extends Record<string, any> = Record<string, any>,\n  TContextSchema extends Record<string, any> = Record<string, any>\n> extends MiddlewareNode<TStateSchema, TContextSchema> {\n  lc_namespace = [\"langchain\", \"agents\", \"afterModalNodes\"];\n\n  name: string;\n\n  constructor(\n    public middleware: AgentMiddleware<any, any, any>,\n    options: MiddlewareNodeOptions\n  ) {\n    super(\n      {\n        name: `AfterModelNode_${middleware.name}`,\n        func: async (state: TStateSchema, config?: LangGraphRunnableConfig) =>\n          this.invokeMiddleware(state, config),\n      },\n      options\n    );\n    this.name = `AfterModelNode_${middleware.name}`;\n  }\n\n  runHook(state: TStateSchema, runtime: Runtime<TContextSchema>) {\n    return this.middleware.afterModel!(\n      state as Record<string, any> & AgentBuiltInState,\n      runtime as Runtime<unknown>\n    ) as Promise<MiddlewareResult<TStateSchema>>;\n  }\n}\n"],"mappings":";;;;;;AAaA,IAAa,iBAAb,cAGUA,kCAA6C;CACrD,eAAe;EAAC;EAAa;EAAU;CAAkB;CAEzD;CAEA,YACSC,YACPC,SACA;EACA,MACE;GACE,MAAM,CAAC,eAAe,EAAE,WAAW,MAAM;GACzC,MAAM,OAAOC,OAAqBC,WAChC,KAAK,iBAAiB,OAAO,OAAO;EACvC,GACD,QACD;EAVM;EAWP,KAAK,OAAO,CAAC,eAAe,EAAE,WAAW,MAAM;CAChD;CAED,QAAQD,OAAqBE,SAAkC;AAC7D,SAAO,KAAK,WAAW,WACrB,OACA,QACD;CACF;AACF"}

package/dist/agents/middlewareAgent/nodes/AfterModalNode.js

@@ -11,11 +11,11 @@ var AfterModelNode = class extends MiddlewareNode {
 		"afterModalNodes"
 	];
 	name;
-	constructor(middleware) {
+	constructor(middleware, options) {
 		super({
 			name: `AfterModelNode_${middleware.name}`,
 			func: async (state, config) => this.invokeMiddleware(state, config)
-		});
+		}, options);
 		this.middleware = middleware;
 		this.name = `AfterModelNode_${middleware.name}`;
 	}

package/dist/agents/middlewareAgent/nodes/AfterModalNode.js.map

@@ -1 +1 @@
-{"version":3,"file":"AfterModalNode.js","names":["middleware: AgentMiddleware<any, any, any>","state: TStateSchema","config?: LangGraphRunnableConfig","runtime: Runtime<TStateSchema, TContextSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode } from \"./middleware.js\";\nimport type {\n  AgentBuiltInState,\n  AgentMiddleware,\n  MiddlewareResult,\n  Runtime,\n} from \"../types.js\";\n\n/**\n * Node for executing a single middleware's afterModel hook.\n */\nexport class AfterModelNode<\n  TStateSchema extends Record<string, any> = Record<string, any>,\n  TContextSchema extends Record<string, any> = Record<string, any>\n> extends MiddlewareNode<TStateSchema, TContextSchema> {\n  lc_namespace = [\"langchain\", \"agents\", \"afterModalNodes\"];\n\n  name: string;\n\n  constructor(public middleware: AgentMiddleware<any, any, any>) {\n    super({\n      name: `AfterModelNode_${middleware.name}`,\n      func: async (state: TStateSchema, config?: LangGraphRunnableConfig) =>\n        this.invokeMiddleware(state, config),\n    });\n    this.name = `AfterModelNode_${middleware.name}`;\n  }\n\n  runHook(state: TStateSchema, runtime: Runtime<TStateSchema, TContextSchema>) {\n    return this.middleware.afterModel!(\n      state as Record<string, any> & AgentBuiltInState,\n      runtime as Runtime<TStateSchema, unknown>\n    ) as Promise<MiddlewareResult<TStateSchema>>;\n  }\n}\n"],"mappings":";;;;;;AAaA,IAAa,iBAAb,cAGU,eAA6C;CACrD,eAAe;EAAC;EAAa;EAAU;CAAkB;CAEzD;CAEA,YAAmBA,YAA4C;EAC7D,MAAM;GACJ,MAAM,CAAC,eAAe,EAAE,WAAW,MAAM;GACzC,MAAM,OAAOC,OAAqBC,WAChC,KAAK,iBAAiB,OAAO,OAAO;EACvC,EAAC;EALe;EAMjB,KAAK,OAAO,CAAC,eAAe,EAAE,WAAW,MAAM;CAChD;CAED,QAAQD,OAAqBE,SAAgD;AAC3E,SAAO,KAAK,WAAW,WACrB,OACA,QACD;CACF;AACF"}
+{"version":3,"file":"AfterModalNode.js","names":["middleware: AgentMiddleware<any, any, any>","options: MiddlewareNodeOptions","state: TStateSchema","config?: LangGraphRunnableConfig","runtime: Runtime<TContextSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode, MiddlewareNodeOptions } from \"./middleware.js\";\nimport type {\n  AgentBuiltInState,\n  AgentMiddleware,\n  MiddlewareResult,\n  Runtime,\n} from \"../types.js\";\n\n/**\n * Node for executing a single middleware's afterModel hook.\n */\nexport class AfterModelNode<\n  TStateSchema extends Record<string, any> = Record<string, any>,\n  TContextSchema extends Record<string, any> = Record<string, any>\n> extends MiddlewareNode<TStateSchema, TContextSchema> {\n  lc_namespace = [\"langchain\", \"agents\", \"afterModalNodes\"];\n\n  name: string;\n\n  constructor(\n    public middleware: AgentMiddleware<any, any, any>,\n    options: MiddlewareNodeOptions\n  ) {\n    super(\n      {\n        name: `AfterModelNode_${middleware.name}`,\n        func: async (state: TStateSchema, config?: LangGraphRunnableConfig) =>\n          this.invokeMiddleware(state, config),\n      },\n      options\n    );\n    this.name = `AfterModelNode_${middleware.name}`;\n  }\n\n  runHook(state: TStateSchema, runtime: Runtime<TContextSchema>) {\n    return this.middleware.afterModel!(\n      state as Record<string, any> & AgentBuiltInState,\n      runtime as Runtime<unknown>\n    ) as Promise<MiddlewareResult<TStateSchema>>;\n  }\n}\n"],"mappings":";;;;;;AAaA,IAAa,iBAAb,cAGU,eAA6C;CACrD,eAAe;EAAC;EAAa;EAAU;CAAkB;CAEzD;CAEA,YACSA,YACPC,SACA;EACA,MACE;GACE,MAAM,CAAC,eAAe,EAAE,WAAW,MAAM;GACzC,MAAM,OAAOC,OAAqBC,WAChC,KAAK,iBAAiB,OAAO,OAAO;EACvC,GACD,QACD;EAVM;EAWP,KAAK,OAAO,CAAC,eAAe,EAAE,WAAW,MAAM;CAChD;CAED,QAAQD,OAAqBE,SAAkC;AAC7D,SAAO,KAAK,WAAW,WACrB,OACA,QACD;CACF;AACF"}

package/dist/agents/middlewareAgent/nodes/AgentNode.cjs

@@ -6,7 +6,6 @@ const require_RunnableCallable = require('../../RunnableCallable.cjs');
 const require_utils$1 = require('../../nodes/utils.cjs');
 const require_withAgentName = require('../../withAgentName.cjs');
 const require_responses = require('../../responses.cjs');
-const require_utils$2 = require('./utils.cjs');
 const __langchain_core_messages = require_rolldown_runtime.__toESM(require("@langchain/core/messages"));
 const __langchain_langgraph = require_rolldown_runtime.__toESM(require("@langchain/langgraph"));
 const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
@@ -15,6 +14,7 @@ const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
 //#region src/agents/middlewareAgent/nodes/AgentNode.ts
 var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 	#options;
+	#runState = { runModelCallCount: 0 };
 	constructor(options) {
 		super({
 			name: options.name ?? "model",
@@ -67,13 +67,20 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 		* return directly without invoking the model again
 		*/
 		return { messages: [] };
+		const privateState = this.getState()._privateState;
 		const response = await this.#invokeModel(state, config);
+		this.#runState.runModelCallCount++;
+		const _privateState = {
+			...privateState,
+			threadLevelCallCount: privateState.threadLevelCallCount + 1
+		};
 		/**
 		* if we were able to generate a structured response, return it
 		*/
 		if ("structuredResponse" in response) return {
 			messages: [...state.messages, ...response.messages || []],
-			structuredResponse: response.structuredResponse
+			structuredResponse: response.structuredResponse,
+			_privateState
 		};
 		/**
 		* if we need to direct the agent to the model, return the update
@@ -81,12 +88,18 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 		if (response instanceof __langchain_langgraph.Command) return response;
 		response.name = this.name;
 		response.lc_kwargs.name = this.name;
-		if (this.#areMoreStepsNeeded(state, response)) return { messages: [new __langchain_core_messages.AIMessage({
-			content: "Sorry, need more steps to process this request.",
-			name: this.name,
-			id: response.id
-		})] };
-		return { messages: [response] };
+		if (this.#areMoreStepsNeeded(state, response)) return {
+			messages: [new __langchain_core_messages.AIMessage({
+				content: "Sorry, need more steps to process this request.",
+				name: this.name,
+				id: response.id
+			})],
+			_privateState
+		};
+		return {
+			messages: [response],
+			_privateState
+		};
 	}
 	/**
 	* Derive the model from the options.
@@ -104,58 +117,117 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 		/**
 		* Execute modifyModelRequest hooks from beforeModelNodes
 		*/
-		const preparedOptions = await this.#executePrepareModelRequestHooks(model, state, config);
+		let preparedOptions = await this.#executePrepareModelRequestHooks(model, state, config);
 		/**
-		* If user provides a model in the preparedOptions, use it,
-		* otherwise use the model from the options
+		* Retry loop for model invocation with error handling
+		* Hard limit of 100 attempts to prevent infinite loops from buggy middleware
 		*/
-		const finalModel = preparedOptions?.model ?? model;
-		/**
-		* Check if the LLM already has bound tools and throw if it does.
-		*/
-		require_utils.validateLLMHasNoBoundTools(finalModel);
-		const structuredResponseFormat = this.#getResponseFormat(finalModel);
-		const modelWithTools = await this.#bindTools(finalModel, preparedOptions, structuredResponseFormat);
-		let modelInput = this.#getModelInputState(state);
-		/**
-		* Use messages from preparedOptions if provided
-		*/
-		if (preparedOptions?.messages) modelInput = {
-			...modelInput,
-			messages: preparedOptions.messages
-		};
-		const signal = require_utils$1.mergeAbortSignals(this.#options.signal, config.signal);
-		const invokeConfig = {
-			...config,
-			signal
-		};
-		const response = await modelWithTools.invoke(modelInput, invokeConfig);
-		/**
-		* if the user requests a native schema output, try to parse the response
-		* and return the structured response if it is valid
-		*/
-		if (structuredResponseFormat?.type === "native") {
-			const structuredResponse = structuredResponseFormat.strategy.parse(response);
-			if (structuredResponse) return {
-				structuredResponse,
-				messages: [response]
+		const maxAttempts = 100;
+		for (let attempt = 1; attempt <= maxAttempts; attempt++) try {
+			/**
+			* If user provides a model in the preparedOptions, use it,
+			* otherwise use the model from the options
+			*/
+			const finalModel = preparedOptions?.model ?? model;
+			/**
+			* Check if the LLM already has bound tools and throw if it does.
+			*/
+			require_utils.validateLLMHasNoBoundTools(finalModel);
+			const structuredResponseFormat = this.#getResponseFormat(finalModel);
+			const modelWithTools = await this.#bindTools(finalModel, preparedOptions, structuredResponseFormat);
+			let modelInput = this.#getModelInputState(state);
+			/**
+			* Use messages from preparedOptions if provided
+			*/
+			if (preparedOptions?.messages) modelInput = {
+				...modelInput,
+				messages: preparedOptions.messages
+			};
+			const signal = require_utils$1.mergeAbortSignals(this.#options.signal, config.signal);
+			const invokeConfig = {
+				...config,
+				signal
 			};
-			return response;
+			const response = await modelWithTools.invoke(modelInput, invokeConfig);
+			/**
+			* if the user requests a native schema output, try to parse the response
+			* and return the structured response if it is valid
+			*/
+			if (structuredResponseFormat?.type === "native") {
+				const structuredResponse = structuredResponseFormat.strategy.parse(response);
+				if (structuredResponse) return {
+					structuredResponse,
+					messages: [response]
+				};
+				return response;
+			}
+			if (!structuredResponseFormat || !response.tool_calls) return response;
+			const toolCalls = response.tool_calls.filter((call) => call.name in structuredResponseFormat.tools);
+			/**
+			* if there were not structured tool calls, we can return the response
+			*/
+			if (toolCalls.length === 0) return response;
+			/**
+			* if there were multiple structured tool calls, we should throw an error as this
+			* scenario is not defined/supported.
+			*/
+			if (toolCalls.length > 1) return this.#handleMultipleStructuredOutputs(response, toolCalls, structuredResponseFormat);
+			const toolStrategy = structuredResponseFormat.tools[toolCalls[0].name];
+			const toolMessageContent = toolStrategy?.options?.toolMessageContent;
+			return this.#handleSingleStructuredOutput(response, toolCalls[0], structuredResponseFormat, toolMessageContent ?? options.lastMessage);
+		} catch (error) {
+			const retryMiddleware = this.#options.retryModelRequestHookMiddleware ?? [];
+			let shouldRetry = false;
+			for (const [middleware, getMiddlewareState] of retryMiddleware) if (middleware.retryModelRequest) {
+				/**
+				* Cast config to LangGraphRunnableConfig to access LangGraph-specific properties
+				*/
+				const lgConfig = config;
+				/**
+				* Merge context with default context of middleware
+				*/
+				const context = middleware.contextSchema ? (0, __langchain_core_utils_types.interopParse)(middleware.contextSchema, lgConfig?.context || {}) : lgConfig?.context;
+				/**
+				* Create runtime
+				*/
+				const privateState = this.getState()._privateState;
+				const runtime = {
+					...privateState,
+					context,
+					writer: lgConfig.writer,
+					interrupt: lgConfig.interrupt,
+					signal: lgConfig.signal
+				};
+				const retryRequest = await middleware.retryModelRequest(
+					error,
+					{
+						model: preparedOptions?.model ?? model,
+						systemPrompt: preparedOptions?.systemPrompt,
+						messages: preparedOptions?.messages ?? state.messages,
+						tools: this.#options.toolClasses
+					},
+					{
+						...getMiddlewareState(),
+						messages: state.messages
+					},
+					/**
+					* ensure runtime is frozen to prevent modifications
+					*/
+					Object.freeze({
+						...runtime,
+						context
+					}),
+					attempt
+				);
+				if (retryRequest) {
+					preparedOptions = retryRequest;
+					shouldRetry = true;
+					break;
+				}
+			}
+			if (!shouldRetry) throw error;
 		}
-		if (!structuredResponseFormat || !response.tool_calls) return response;
-		const toolCalls = response.tool_calls.filter((call) => call.name in structuredResponseFormat.tools);
-		/**
-		* if there were not structured tool calls, we can return the response
-		*/
-		if (toolCalls.length === 0) return response;
-		/**
-		* if there were multiple structured tool calls, we should throw an error as this
-		* scenario is not defined/supported.
-		*/
-		if (toolCalls.length > 1) return this.#handleMultipleStructuredOutputs(response, toolCalls, structuredResponseFormat);
-		const toolStrategy = structuredResponseFormat.tools[toolCalls[0].name];
-		const toolMessageContent = toolStrategy?.options?.toolMessageContent;
-		return this.#handleSingleStructuredOutput(response, toolCalls[0], structuredResponseFormat, toolMessageContent ?? options.lastMessage);
+		throw new Error(`Maximum retry attempts (${maxAttempts}) exceeded`);
 	}
 	/**
 	* If the model returns multiple structured outputs, we need to handle it.
@@ -263,7 +335,7 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 			model,
 			systemPrompt,
 			messages: state.messages,
-			tools: this.#options.toolClasses.map((tool) => tool.name)
+			tools: this.#options.toolClasses
 		};
 		/**
 		* Execute modifyModelRequest hooks from all middleware
@@ -277,23 +349,19 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 			/**
 			* Create runtime
 			*/
+			const privateState = this.getState()._privateState;
 			const runtime = {
-				toolCalls: require_utils$2.parseToolCalls(state.messages),
-				tools: this.#options.toolClasses,
+				...privateState,
 				context,
 				writer: config.writer,
 				interrupt: config.interrupt,
-				signal: config.signal,
-				terminate: (result$1) => ({
-					type: "terminate",
-					result: result$1
-				})
+				signal: config.signal
 			};
 			const result = await middleware.modifyModelRequest(
 				currentOptions,
 				{
-					messages: state.messages,
-					...getMiddlewareState()
+					...getMiddlewareState(),
+					messages: state.messages
 				},
 				/**
 				* ensure runtime is frozen to prevent modifications
@@ -303,23 +371,32 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 					context
 				})
 			);
-			/**
-			* raise meaningful error if unknown tools were selected
-			*/
-			const unknownTools = result?.tools?.filter((tool) => !this.#options.toolClasses.some((t) => t.name === tool)) ?? [];
-			if (unknownTools.length > 0) throw new Error(`Unknown tools selected in middleware "${middleware.name}": ${unknownTools.join(", ")}, available tools: ${this.#options.toolClasses.map((t) => t.name).join(", ")}!`);
-			if (result) currentOptions = {
-				...currentOptions,
-				...result
-			};
+			if (result) {
+				const modifiedTools = result.tools ?? [];
+				/**
+				* Verify that the user didn't add any new tools.
+				* We can't allow this as the ToolNode is already initiated with given tools.
+				*/
+				const newTools = modifiedTools.filter((tool) => require_utils.isClientTool(tool) && !this.#options.toolClasses.some((t) => t.name === tool.name));
+				if (newTools.length > 0) throw new Error(`You have added a new tool in "modifyModelRequest" hook of middleware "${middleware.name}": ${newTools.map((tool) => tool.name).join(", ")}. This is not supported.`);
+				/**
+				* Verify that user has not added or modified a tool with the same name.
+				* We can't allow this as the ToolNode is already initiated with given tools.
+				*/
+				const invalidTools = modifiedTools.filter((tool) => require_utils.isClientTool(tool) && this.#options.toolClasses.every((t) => t !== tool));
+				if (invalidTools.length > 0) throw new Error(`You have modified a tool in "modifyModelRequest" hook of middleware "${middleware.name}": ${invalidTools.map((tool) => tool.name).join(", ")}. This is not supported.`);
+				currentOptions = {
+					...currentOptions,
+					...result
+				};
+			}
 		}
 		return currentOptions;
 	}
 	async #bindTools(model, preparedOptions, structuredResponseFormat) {
 		const options = {};
 		const structuredTools = Object.values(structuredResponseFormat && "tools" in structuredResponseFormat ? structuredResponseFormat.tools : {});
-		const preparedTools = preparedOptions?.tools ?? [];
-		const allTools = [...preparedTools.length > 0 ? this.#options.toolClasses.filter((tool) => typeof tool.name === "string" && preparedTools.includes(tool.name)) : this.#options.toolClasses, ...structuredTools.map((toolStrategy) => toolStrategy.tool)];
+		const allTools = [...preparedOptions?.tools ?? [], ...structuredTools.map((toolStrategy) => toolStrategy.tool)];
 		/**
 		* If there are structured tools, we need to set the tool choice to "any"
 		* so that the model can choose to use a structured tool or not.
@@ -365,7 +442,24 @@ var AgentNode = class extends require_RunnableCallable.RunnableCallable {
 		return modelRunnable;
 	}
 	static get nodeOptions() {
-		return { input: zod_v3.z.object({ messages: zod_v3.z.array(zod_v3.z.custom()) }) };
+		return { input: zod_v3.z.object({
+			messages: zod_v3.z.array(zod_v3.z.custom()),
+			_privateState: zod_v3.z.object({ threadLevelCallCount: zod_v3.z.number() })
+		}) };
+	}
+	getState() {
+		const origState = super.getState() ?? { _privateState: {
+			threadLevelCallCount: 0,
+			runModelCallCount: 0
+		} };
+		return {
+			messages: [],
+			...origState,
+			_privateState: {
+				...origState._privateState ?? {},
+				...this.#runState
+			}
+		};
 	}
 };