langchain 1.0.0-alpha.5 → 1.0.0-alpha.7

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

@@ -1,5 +1,6 @@
-import { AgentBuiltInState, AgentMiddleware, Controls, MiddlewareResult, ModelRequest, Runtime } from "./types.cjs";
-import { z } from "zod";
+import { ClientTool, ServerTool } from "../types.cjs";
+import { AgentBuiltInState, AgentMiddleware, JumpToTarget, MiddlewareResult, ModelRequest, Runtime } from "./types.cjs";
+import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
 
 //#region src/agents/middlewareAgent/middleware.d.ts
 
@@ -10,7 +11,7 @@ import { z } from "zod";
  * @param config.name - The name of the middleware
  * @param config.stateSchema - The schema of the middleware state
  * @param config.contextSchema - The schema of the middleware context
- * @param config.prepareModelRequest - The function to prepare the model request
+ * @param config.modifyModelRequest - The function to prepare the model request
  * @param config.beforeModel - The function to run before the model call
  * @param config.afterModel - The function to run after the model call
  * @returns A middleware instance
@@ -33,13 +34,72 @@ import { z } from "zod";
  * });
  * ```
  */
-declare function createMiddleware<TSchema extends z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {
+declare function createMiddleware<TSchema extends InteropZodObject | undefined = undefined, TContextSchema extends InteropZodObject | InteropZodOptional<InteropZodObject> | InteropZodDefault<InteropZodObject> | undefined = undefined>(config: {
+  /**
+   * The name of the middleware
+   */
   name: string;
+  /**
+   * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:
+   * - A Zod object
+   * - A Zod optional object
+   * - A Zod default object
+   * - Undefined
+   */
   stateSchema?: TSchema;
+  /**
+   * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:
+   * - A Zod object
+   * - A Zod optional object
+   * - A Zod default object
+   * - Undefined
+   */
   contextSchema?: TContextSchema;
-  prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;
-  beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
-  afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
+  /**
+   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.
+   */
+  beforeModelJumpTo?: JumpToTarget[];
+  /**
+   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.
+   */
+  afterModelJumpTo?: JumpToTarget[];
+  /**
+   * Additional tools registered by the middleware.
+   */
+  tools?: (ClientTool | ServerTool)[];
+  /**
+   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.
+   * It allows to modify the model request before it is passed to the model.
+   *
+   * @param request - The model request
+   * @param request.model - The model to use for this step.
+   * @param request.messages - The messages to send to the model.
+   * @param request.systemPrompt - The system message for this step.
+   * @param request.toolChoice - The tool choice configuration for this step.
+   * @param request.tools - The tools to make available for this step.
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified model request or undefined to pass through
+   */
+  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;
+  /**
+   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` hook.
+   * It allows to modify the state of the agent.
+   *
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified middleware state or undefined to pass through
+   */
+  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> : {}>>;
+  /**
+   * The function to run after the model call. This function is called after the model is invoked and before any tools are called.
+   * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.
+   *
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified middleware state or undefined to pass through
+   */
+  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> : {}>>;
 }): AgentMiddleware<TSchema, TContextSchema, any>;
 //#endregion
 export { createMiddleware };

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

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.cts","names":["z","AgentMiddleware","Runtime","Controls","MiddlewareResult","AgentBuiltInState","ModelRequest","createMiddleware","ZodObject","TSchema","TContextSchema","infer","Promise","Partial"],"sources":["../../../src/agents/middlewareAgent/middleware.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\nimport { z } from \"zod\";\nimport type { AgentMiddleware, Runtime, Controls, MiddlewareResult, AgentBuiltInState, ModelRequest } 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.prepareModelRequest - 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 z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {\n    name: string;\n    stateSchema?: TSchema;\n    contextSchema?: TContextSchema;\n    prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;\n    beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n    afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n}): AgentMiddleware<TSchema, TContextSchema, any>;\n"],"mappings":";;;;;;AAiCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKmHE,iBAL3FK,gBAK2FL,CAAAA,gBAL1DF,CAAAA,CAAEQ,SAKwDN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,uBALOF,CAAAA,CAAEQ,SAKTN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,CAAAA,CAAAA,MAAAA,EAAAA;EAAO,IAA8FO,EAAAA,MAAAA;EAAO,WAAWD,CAAAA,EAHxNC,OAGwND;EAAS,aAAgBC,CAAAA,EAF/OC,cAE+OD;EAAO,mBAAbE,CAAAA,EAAAA,CAAAA,OAAAA,EADzNL,YACyNK,EAAAA,KAAAA,EAAAA,CADnMF,OACmME,SADnLX,CAAAA,CAAEQ,SACiLG,CAAAA,GAAAA,CAAAA,GADhKX,CAAAA,CAAEW,KAC8JA,CADxJF,OACwJE,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GADvIN,iBACuIM,EAAAA,OAAAA,EAD3GT,OAC2GS,CADnGD,cACmGC,SAD5EX,CAAAA,CAAEQ,SAC0EG,CAAAA,GAAAA,CAAAA,GADzDX,CAAAA,CAAEW,KACuDA,CADjDD,cACiDC,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,GADvBC,OACuBD,CADfL,YACeK,GAAAA,IAAAA,CAAAA,GADQL,YACRK,GAAAA,IAAAA;EAAK,WAAkBN,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAAzPI,OAAyPJ,SAAzOL,CAAAA,CAAEQ,SAAuOH,CAAAA,GAAAA,CAAAA,GAAtNL,CAAAA,CAAEW,KAAoNN,CAA9MI,OAA8MJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAA7LA,iBAA6LA,EAAAA,OAAAA,EAAjKH,OAAiKG,CAAzJK,cAAyJL,SAAlIL,CAAAA,CAAEQ,SAAgIH,CAAAA,GAAAA,CAAAA,GAA/GL,CAAAA,CAAEW,KAA6GN,CAAvGK,cAAuGL,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EAAtEF,QAAsEE,CAAAA,CAA5DI,OAA4DJ,SAA5CL,CAAAA,CAAEQ,SAA0CH,CAAAA,GAAAA,CAAAA,GAAzBL,CAAAA,CAAEW,KAAuBN,CAAjBI,OAAiBJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAAAA,iBAAAA,CAAAA,EAAAA,GAAuBO,OAAvBP,CAA+BD,gBAA/BC,CAAgDQ,OAAhDR,CAAwDI,OAAxDJ,SAAwEL,CAAAA,CAAEQ,SAA1EH,CAAAA,GAAAA,CAAAA,GAA2FL,CAAAA,CAAEW,KAA7FN,CAAmGI,OAAnGJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAAsHD,gBAAtHC,CAAuIQ,OAAvIR,CAA+II,OAA/IJ,SAA+JL,CAAAA,CAAEQ,SAAjKH,CAAAA,GAAAA,CAAAA,GAAkLL,CAAAA,CAAEW,KAApLN,CAA0LI,OAA1LJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;EAAiB,UAAvFF,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CACpLM,OADoLN,SACpKH,CAAAA,CAAEQ,SADkKL,CAAAA,GAAAA,CAAAA,GACjJH,CAAAA,CAAEW,KAD+IR,CACzIM,OADyIN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACxHE,iBADwHF,EAAAA,OAAAA,EAC5FD,OAD4FC,CACpFO,cADoFP,SAC7DH,CAAAA,CAAEQ,SAD2DL,CAAAA,GAAAA,CAAAA,GAC1CH,CAAAA,CAAEW,KADwCR,CAClCO,cADkCP,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EACDA,QADCA,CAAAA,CACSM,OADTN,SACyBH,CAAAA,CAAEQ,SAD3BL,CAAAA,GAAAA,CAAAA,GAC4CH,CAAAA,CAAEW,KAD9CR,CACoDM,OADpDN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACqEE,iBADrEF,CAAAA,EAAAA,GAC4FS,OAD5FT,CACoGC,gBADpGD,CACqHU,OADrHV,CAC6HM,OAD7HN,SAC6IH,CAAAA,CAAEQ,SAD/IL,CAAAA,GAAAA,CAAAA,GACgKH,CAAAA,CAAEW,KADlKR,CACwKM,OADxKN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAC2LC,gBAD3LD,CAC4MU,OAD5MV,CACoNM,OADpNN,SACoOH,CAAAA,CAAEQ,SADtOL,CAAAA,GAAAA,CAAAA,GACuPH,CAAAA,CAAEW,KADzPR,CAC+PM,OAD/PN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;CAAQ,CAAA,EAElNF,eAFwUQ,CAExTA,OAFwTA,EAE/SC,cAF+SD,EAAAA,GAAAA,CAAAA"}
+{"version":3,"file":"middleware.d.cts","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"}

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

@@ -1,5 +1,6 @@
-import { AgentBuiltInState, AgentMiddleware, Controls, MiddlewareResult, ModelRequest, Runtime } from "./types.js";
-import { z } from "zod";
+import { ClientTool, ServerTool } from "../types.js";
+import { AgentBuiltInState, AgentMiddleware, JumpToTarget, MiddlewareResult, ModelRequest, Runtime } from "./types.js";
+import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
 
 //#region src/agents/middlewareAgent/middleware.d.ts
 
@@ -10,7 +11,7 @@ import { z } from "zod";
  * @param config.name - The name of the middleware
  * @param config.stateSchema - The schema of the middleware state
  * @param config.contextSchema - The schema of the middleware context
- * @param config.prepareModelRequest - The function to prepare the model request
+ * @param config.modifyModelRequest - The function to prepare the model request
  * @param config.beforeModel - The function to run before the model call
  * @param config.afterModel - The function to run after the model call
  * @returns A middleware instance
@@ -33,13 +34,72 @@ import { z } from "zod";
  * });
  * ```
  */
-declare function createMiddleware<TSchema extends z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {
+declare function createMiddleware<TSchema extends InteropZodObject | undefined = undefined, TContextSchema extends InteropZodObject | InteropZodOptional<InteropZodObject> | InteropZodDefault<InteropZodObject> | undefined = undefined>(config: {
+  /**
+   * The name of the middleware
+   */
   name: string;
+  /**
+   * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:
+   * - A Zod object
+   * - A Zod optional object
+   * - A Zod default object
+   * - Undefined
+   */
   stateSchema?: TSchema;
+  /**
+   * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:
+   * - A Zod object
+   * - A Zod optional object
+   * - A Zod default object
+   * - Undefined
+   */
   contextSchema?: TContextSchema;
-  prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;
-  beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
-  afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
+  /**
+   * Explitictly defines which targets are allowed to be jumped to from the `beforeModel` hook.
+   */
+  beforeModelJumpTo?: JumpToTarget[];
+  /**
+   * Explitictly defines which targets are allowed to be jumped to from the `afterModel` hook.
+   */
+  afterModelJumpTo?: JumpToTarget[];
+  /**
+   * Additional tools registered by the middleware.
+   */
+  tools?: (ClientTool | ServerTool)[];
+  /**
+   * The function to modify the model request. This function is called after the `beforeModel` hook of this middleware and before the model is invoked.
+   * It allows to modify the model request before it is passed to the model.
+   *
+   * @param request - The model request
+   * @param request.model - The model to use for this step.
+   * @param request.messages - The messages to send to the model.
+   * @param request.systemPrompt - The system message for this step.
+   * @param request.toolChoice - The tool choice configuration for this step.
+   * @param request.tools - The tools to make available for this step.
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified model request or undefined to pass through
+   */
+  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;
+  /**
+   * The function to run before the model call. This function is called before the model is invoked and before the `modifyModelRequest` hook.
+   * It allows to modify the state of the agent.
+   *
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified middleware state or undefined to pass through
+   */
+  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> : {}>>;
+  /**
+   * The function to run after the model call. This function is called after the model is invoked and before any tools are called.
+   * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.
+   *
+   * @param state - The middleware state
+   * @param runtime - The middleware runtime
+   * @returns The modified middleware state or undefined to pass through
+   */
+  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> : {}>>;
 }): AgentMiddleware<TSchema, TContextSchema, any>;
 //#endregion
 export { createMiddleware };

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

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.ts","names":["z","AgentMiddleware","Runtime","Controls","MiddlewareResult","AgentBuiltInState","ModelRequest","createMiddleware","ZodObject","TSchema","TContextSchema","infer","Promise","Partial"],"sources":["../../../src/agents/middlewareAgent/middleware.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\nimport { z } from \"zod\";\nimport type { AgentMiddleware, Runtime, Controls, MiddlewareResult, AgentBuiltInState, ModelRequest } 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.prepareModelRequest - 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 z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {\n    name: string;\n    stateSchema?: TSchema;\n    contextSchema?: TContextSchema;\n    prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;\n    beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n    afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n}): AgentMiddleware<TSchema, TContextSchema, any>;\n"],"mappings":";;;;;;AAiCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKmHE,iBAL3FK,gBAK2FL,CAAAA,gBAL1DF,CAAAA,CAAEQ,SAKwDN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,uBALOF,CAAAA,CAAEQ,SAKTN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,CAAAA,CAAAA,MAAAA,EAAAA;EAAO,IAA8FO,EAAAA,MAAAA;EAAO,WAAWD,CAAAA,EAHxNC,OAGwND;EAAS,aAAgBC,CAAAA,EAF/OC,cAE+OD;EAAO,mBAAbE,CAAAA,EAAAA,CAAAA,OAAAA,EADzNL,YACyNK,EAAAA,KAAAA,EAAAA,CADnMF,OACmME,SADnLX,CAAAA,CAAEQ,SACiLG,CAAAA,GAAAA,CAAAA,GADhKX,CAAAA,CAAEW,KAC8JA,CADxJF,OACwJE,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GADvIN,iBACuIM,EAAAA,OAAAA,EAD3GT,OAC2GS,CADnGD,cACmGC,SAD5EX,CAAAA,CAAEQ,SAC0EG,CAAAA,GAAAA,CAAAA,GADzDX,CAAAA,CAAEW,KACuDA,CADjDD,cACiDC,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,GADvBC,OACuBD,CADfL,YACeK,GAAAA,IAAAA,CAAAA,GADQL,YACRK,GAAAA,IAAAA;EAAK,WAAkBN,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAAzPI,OAAyPJ,SAAzOL,CAAAA,CAAEQ,SAAuOH,CAAAA,GAAAA,CAAAA,GAAtNL,CAAAA,CAAEW,KAAoNN,CAA9MI,OAA8MJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAA7LA,iBAA6LA,EAAAA,OAAAA,EAAjKH,OAAiKG,CAAzJK,cAAyJL,SAAlIL,CAAAA,CAAEQ,SAAgIH,CAAAA,GAAAA,CAAAA,GAA/GL,CAAAA,CAAEW,KAA6GN,CAAvGK,cAAuGL,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EAAtEF,QAAsEE,CAAAA,CAA5DI,OAA4DJ,SAA5CL,CAAAA,CAAEQ,SAA0CH,CAAAA,GAAAA,CAAAA,GAAzBL,CAAAA,CAAEW,KAAuBN,CAAjBI,OAAiBJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAAAA,iBAAAA,CAAAA,EAAAA,GAAuBO,OAAvBP,CAA+BD,gBAA/BC,CAAgDQ,OAAhDR,CAAwDI,OAAxDJ,SAAwEL,CAAAA,CAAEQ,SAA1EH,CAAAA,GAAAA,CAAAA,GAA2FL,CAAAA,CAAEW,KAA7FN,CAAmGI,OAAnGJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAAsHD,gBAAtHC,CAAuIQ,OAAvIR,CAA+II,OAA/IJ,SAA+JL,CAAAA,CAAEQ,SAAjKH,CAAAA,GAAAA,CAAAA,GAAkLL,CAAAA,CAAEW,KAApLN,CAA0LI,OAA1LJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;EAAiB,UAAvFF,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CACpLM,OADoLN,SACpKH,CAAAA,CAAEQ,SADkKL,CAAAA,GAAAA,CAAAA,GACjJH,CAAAA,CAAEW,KAD+IR,CACzIM,OADyIN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACxHE,iBADwHF,EAAAA,OAAAA,EAC5FD,OAD4FC,CACpFO,cADoFP,SAC7DH,CAAAA,CAAEQ,SAD2DL,CAAAA,GAAAA,CAAAA,GAC1CH,CAAAA,CAAEW,KADwCR,CAClCO,cADkCP,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EACDA,QADCA,CAAAA,CACSM,OADTN,SACyBH,CAAAA,CAAEQ,SAD3BL,CAAAA,GAAAA,CAAAA,GAC4CH,CAAAA,CAAEW,KAD9CR,CACoDM,OADpDN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACqEE,iBADrEF,CAAAA,EAAAA,GAC4FS,OAD5FT,CACoGC,gBADpGD,CACqHU,OADrHV,CAC6HM,OAD7HN,SAC6IH,CAAAA,CAAEQ,SAD/IL,CAAAA,GAAAA,CAAAA,GACgKH,CAAAA,CAAEW,KADlKR,CACwKM,OADxKN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAC2LC,gBAD3LD,CAC4MU,OAD5MV,CACoNM,OADpNN,SACoOH,CAAAA,CAAEQ,SADtOL,CAAAA,GAAAA,CAAAA,GACuPH,CAAAA,CAAEW,KADzPR,CAC+PM,OAD/PN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;CAAQ,CAAA,EAElNF,eAFwUQ,CAExTA,OAFwTA,EAE/SC,cAF+SD,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"],"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"}

package/dist/agents/middlewareAgent/middleware.js

@@ -6,7 +6,7 @@
 * @param config.name - The name of the middleware
 * @param config.stateSchema - The schema of the middleware state
 * @param config.contextSchema - The schema of the middleware context
-* @param config.prepareModelRequest - The function to prepare the model request
+* @param config.modifyModelRequest - The function to prepare the model request
 * @param config.beforeModel - The function to run before the model call
 * @param config.afterModel - The function to run after the model call
 * @returns A middleware instance
@@ -33,11 +33,14 @@ function createMiddleware(config) {
 	const middleware = {
 		name: config.name,
 		stateSchema: config.stateSchema,
-		contextSchema: config.contextSchema
+		contextSchema: config.contextSchema,
+		beforeModelJumpTo: config.beforeModelJumpTo,
+		afterModelJumpTo: config.afterModelJumpTo,
+		tools: config.tools ?? []
 	};
-	if (config.prepareModelRequest) middleware.prepareModelRequest = async (options, state, runtime) => Promise.resolve(config.prepareModelRequest(options, state, runtime));
-	if (config.beforeModel) middleware.beforeModel = async (state, runtime, controls) => Promise.resolve(config.beforeModel(state, runtime, controls));
-	if (config.afterModel) middleware.afterModel = async (state, runtime, controls) => Promise.resolve(config.afterModel(state, runtime, controls));
+	if (config.modifyModelRequest) middleware.modifyModelRequest = async (options, state, runtime) => Promise.resolve(config.modifyModelRequest(options, state, runtime));
+	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  name: string;\n  stateSchema?: TSchema;\n  contextSchema?: TContextSchema;\n  prepareModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  beforeModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n  afterModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n}","middleware: AgentMiddleware<TSchema, TContextSchema, any>"],"sources":["../../../src/agents/middlewareAgent/middleware.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\nimport { z } from \"zod\";\nimport type {\n  AgentMiddleware,\n  Runtime,\n  Controls,\n  MiddlewareResult,\n  AgentBuiltInState,\n  ModelRequest,\n} 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.prepareModelRequest - 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 z.ZodObject<any> | undefined = undefined,\n  TContextSchema extends z.ZodObject<any> | undefined = undefined\n>(config: {\n  name: string;\n  stateSchema?: TSchema;\n  contextSchema?: TContextSchema;\n  prepareModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  beforeModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n  afterModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\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  };\n\n  if (config.prepareModelRequest) {\n    middleware.prepareModelRequest = async (options, state, runtime) =>\n      Promise.resolve(config.prepareModelRequest!(options, state, runtime));\n  }\n\n  if (config.beforeModel) {\n    middleware.beforeModel = async (state, runtime, controls) =>\n      Promise.resolve(config.beforeModel!(state, runtime, controls));\n  }\n\n  if (config.afterModel) {\n    middleware.afterModel = async (state, runtime, controls) =>\n      Promise.resolve(config.afterModel!(state, runtime, controls));\n  }\n\n  return middleware;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCA,SAAgB,iBAGdA,QAkDgD;CAChD,MAAMC,aAA4D;EAChE,MAAM,OAAO;EACb,aAAa,OAAO;EACpB,eAAe,OAAO;CACvB;AAED,KAAI,OAAO,qBACT,WAAW,sBAAsB,OAAO,SAAS,OAAO,YACtD,QAAQ,QAAQ,OAAO,oBAAqB,SAAS,OAAO,QAAQ,CAAC;AAGzE,KAAI,OAAO,aACT,WAAW,cAAc,OAAO,OAAO,SAAS,aAC9C,QAAQ,QAAQ,OAAO,YAAa,OAAO,SAAS,SAAS,CAAC;AAGlE,KAAI,OAAO,YACT,WAAW,aAAa,OAAO,OAAO,SAAS,aAC7C,QAAQ,QAAQ,OAAO,WAAY,OAAO,SAAS,SAAS,CAAC;AAGjE,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      (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"}

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

@@ -19,8 +19,8 @@ var AfterModelNode = class extends require_middleware.MiddlewareNode {
 		this.middleware = middleware;
 		this.name = `AfterModelNode_${middleware.name}`;
 	}
-	runHook(state, runtime, controls) {
-		return this.middleware.afterModel(state, runtime, controls);
+	runHook(state, runtime) {
+		return this.middleware.afterModel(state, runtime);
 	}
 };
 

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<TContextSchema>","controls: Controls<TStateSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["import { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode } from \"./middleware.js\";\nimport type {\n  Controls,\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(\n    state: TStateSchema,\n    runtime: Runtime<TContextSchema>,\n    controls: Controls<TStateSchema>\n  ) {\n    return this.middleware.afterModel!(state, runtime, controls) as Promise<\n      MiddlewareResult<TStateSchema>\n    >;\n  }\n}\n"],"mappings":";;;;;;AAYA,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,QACED,OACAE,SACAC,UACA;AACA,SAAO,KAAK,WAAW,WAAY,OAAO,SAAS,SAAS;CAG7D;AACF"}
+{"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"}

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

@@ -19,8 +19,8 @@ var AfterModelNode = class extends MiddlewareNode {
 		this.middleware = middleware;
 		this.name = `AfterModelNode_${middleware.name}`;
 	}
-	runHook(state, runtime, controls) {
-		return this.middleware.afterModel(state, runtime, controls);
+	runHook(state, runtime) {
+		return this.middleware.afterModel(state, runtime);
 	}
 };
 

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<TContextSchema>","controls: Controls<TStateSchema>"],"sources":["../../../../src/agents/middlewareAgent/nodes/AfterModalNode.ts"],"sourcesContent":["import { LangGraphRunnableConfig } from \"@langchain/langgraph\";\nimport { MiddlewareNode } from \"./middleware.js\";\nimport type {\n  Controls,\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(\n    state: TStateSchema,\n    runtime: Runtime<TContextSchema>,\n    controls: Controls<TStateSchema>\n  ) {\n    return this.middleware.afterModel!(state, runtime, controls) as Promise<\n      MiddlewareResult<TStateSchema>\n    >;\n  }\n}\n"],"mappings":";;;;;;AAYA,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,QACED,OACAE,SACAC,UACA;AACA,SAAO,KAAK,WAAW,WAAY,OAAO,SAAS,SAAS;CAG7D;AACF"}
+{"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"}