langchain 1.0.0-alpha.6 → 1.0.0-alpha.8

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.cts","names":["BaseMessage","countTokensApproximately"],"sources":["../../../../src/agents/middlewareAgent/middleware/utils.d.ts"],"sourcesContent":["import type { BaseMessage } from \"@langchain/core/messages\";\n/**\n * Default token counter that approximates based on character count\n * @param messages Messages to count tokens for\n * @returns Approximate token count\n */\nexport declare function countTokensApproximately(messages: BaseMessage[]): number;\n"],"mappings":";;;;;;AAMA;;;iBAAwBC,wBAAAA,WAAmCD"}

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

@@ -0,0 +1,13 @@
+import { BaseMessage } from "@langchain/core/messages";
+
+//#region src/agents/middlewareAgent/middleware/utils.d.ts
+
+/**
+ * Default token counter that approximates based on character count
+ * @param messages Messages to count tokens for
+ * @returns Approximate token count
+ */
+declare function countTokensApproximately(messages: BaseMessage[]): number;
+//#endregion
+export { countTokensApproximately };
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","names":["BaseMessage","countTokensApproximately"],"sources":["../../../../src/agents/middlewareAgent/middleware/utils.d.ts"],"sourcesContent":["import type { BaseMessage } from \"@langchain/core/messages\";\n/**\n * Default token counter that approximates based on character count\n * @param messages Messages to count tokens for\n * @returns Approximate token count\n */\nexport declare function countTokensApproximately(messages: BaseMessage[]): number;\n"],"mappings":";;;;;;AAMA;;;iBAAwBC,wBAAAA,WAAmCD"}

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

@@ -0,0 +1,25 @@
+//#region src/agents/middlewareAgent/middleware/utils.ts
+/**
+* Default token counter that approximates based on character count
+* @param messages Messages to count tokens for
+* @returns Approximate token count
+*/
+function countTokensApproximately(messages) {
+	let totalChars = 0;
+	for (const msg of messages) {
+		let textContent;
+		if (typeof msg.content === "string") textContent = msg.content;
+		else if (Array.isArray(msg.content)) textContent = msg.content.map((item) => {
+			if (typeof item === "string") return item;
+			if (item.type === "text" && "text" in item) return item.text;
+			return "";
+		}).join("");
+		else textContent = "";
+		totalChars += textContent.length;
+	}
+	return Math.ceil(totalChars / 4);
+}
+
+//#endregion
+export { countTokensApproximately };
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","names":["messages: BaseMessage[]","textContent: string"],"sources":["../../../../src/agents/middlewareAgent/middleware/utils.ts"],"sourcesContent":["import type { BaseMessage } from \"@langchain/core/messages\";\n\n/**\n * Default token counter that approximates based on character count\n * @param messages Messages to count tokens for\n * @returns Approximate token count\n */\nexport function countTokensApproximately(messages: BaseMessage[]): number {\n  let totalChars = 0;\n  for (const msg of messages) {\n    let textContent: string;\n    if (typeof msg.content === \"string\") {\n      textContent = msg.content;\n    } else if (Array.isArray(msg.content)) {\n      textContent = msg.content\n        .map((item) => {\n          if (typeof item === \"string\") return item;\n          if (item.type === \"text\" && \"text\" in item) return item.text;\n          return \"\";\n        })\n        .join(\"\");\n    } else {\n      textContent = \"\";\n    }\n    totalChars += textContent.length;\n  }\n  // Approximate 1 token = 4 characters\n  return Math.ceil(totalChars / 4);\n}\n"],"mappings":";;;;;;AAOA,SAAgB,yBAAyBA,UAAiC;CACxE,IAAI,aAAa;AACjB,MAAK,MAAM,OAAO,UAAU;EAC1B,IAAIC;AACJ,MAAI,OAAO,IAAI,YAAY,UACzB,cAAc,IAAI;WACT,MAAM,QAAQ,IAAI,QAAQ,EACnC,cAAc,IAAI,QACf,IAAI,CAAC,SAAS;AACb,OAAI,OAAO,SAAS,SAAU,QAAO;AACrC,OAAI,KAAK,SAAS,UAAU,UAAU,KAAM,QAAO,KAAK;AACxD,UAAO;EACR,EAAC,CACD,KAAK,GAAG;OAEX,cAAc;EAEhB,cAAc,YAAY;CAC3B;AAED,QAAO,KAAK,KAAK,aAAa,EAAE;AACjC"}

package/dist/agents/middlewareAgent/middleware.cjs

@@ -7,7 +7,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
@@ -34,11 +34,15 @@ 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.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.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.cjs","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/no-explicit-any */\nimport { z } from \"zod/v3\";\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.cjs","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/middleware.d.cts

@@ -1,5 +1,6 @@
-import { AgentBuiltInState, AgentMiddleware, Controls, MiddlewareResult, ModelRequest, Runtime } from "./types.cjs";
-import { z } from "zod/v3";
+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/v3";
  * @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,83 @@ import { z } from "zod/v3";
  * });
  * ```
  */
-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<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 handle model invocation errors and optionally retry.
+   *
+   * @param error - The exception that occurred during model invocation
+   * @param request - The original model request that failed
+   * @param state - The current agent state
+   * @param runtime - The runtime context
+   * @param attempt - The current attempt number (1-indexed)
+   * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise)
+   */
+  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;
+  /**
+   * 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<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<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/no-explicit-any */\nimport { z } from \"zod/v3\";\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","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.d.ts

@@ -1,5 +1,6 @@
-import { AgentBuiltInState, AgentMiddleware, Controls, MiddlewareResult, ModelRequest, Runtime } from "./types.js";
-import { z } from "zod/v3";
+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/v3";
  * @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,83 @@ import { z } from "zod/v3";
  * });
  * ```
  */
-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<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 handle model invocation errors and optionally retry.
+   *
+   * @param error - The exception that occurred during model invocation
+   * @param request - The original model request that failed
+   * @param state - The current agent state
+   * @param runtime - The runtime context
+   * @param attempt - The current attempt number (1-indexed)
+   * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise)
+   */
+  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;
+  /**
+   * 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<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<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/no-explicit-any */\nimport { z } from \"zod/v3\";\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","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

@@ -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,15 @@ 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.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  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/no-explicit-any */\nimport { z } from \"zod/v3\";\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      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,16 +11,16 @@ 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}`;
 	}
-	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":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { 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":";;;;;;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,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>","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,16 +11,16 @@ 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}`;
 	}
-	runHook(state, runtime, controls) {
-		return this.middleware.afterModel(state, runtime, controls);
+	runHook(state, runtime) {
+		return this.middleware.afterModel(state, runtime);
 	}
 };