langchain 1.0.2 → 1.0.3

package/dist/agents/middleware/toolRetry.js

@@ -0,0 +1,223 @@
+import { sleep } from "./utils.js";
+import { createMiddleware } from "../middleware.js";
+import { ToolMessage } from "@langchain/core/messages";
+import { z } from "zod/v3";
+
+//#region src/agents/middleware/toolRetry.ts
+/**
+* Calculate delay for a retry attempt with exponential backoff and jitter.
+*
+* @param retryNumber - The retry attempt number (0-indexed)
+* @param config - Configuration for backoff calculation
+* @returns Delay in milliseconds before next retry
+*
+* @internal Exported for testing purposes
+*/
+function calculateRetryDelay(config, retryNumber) {
+	const { backoffFactor, initialDelayMs, maxDelayMs, jitter } = config;
+	let delay;
+	if (backoffFactor === 0) delay = initialDelayMs;
+	else delay = initialDelayMs * backoffFactor ** retryNumber;
+	delay = Math.min(delay, maxDelayMs);
+	if (jitter && delay > 0) {
+		const jitterAmount = delay * .25;
+		delay = delay + (Math.random() * 2 - 1) * jitterAmount;
+		delay = Math.max(0, delay);
+	}
+	return delay;
+}
+/**
+* Configuration options for the Tool Retry Middleware.
+*/
+const ToolRetryMiddlewareOptionsSchema = z.object({
+	maxRetries: z.number().min(0).default(2),
+	tools: z.array(z.union([
+		z.custom(),
+		z.custom(),
+		z.string()
+	])).optional(),
+	retryOn: z.union([z.function().args(z.instanceof(Error)).returns(z.boolean()), z.array(z.custom())]).default(() => () => true),
+	onFailure: z.union([
+		z.literal("raise"),
+		z.literal("return_message"),
+		z.function().args(z.instanceof(Error)).returns(z.string())
+	]).default("return_message"),
+	backoffFactor: z.number().min(0).default(2),
+	initialDelayMs: z.number().min(0).default(1e3),
+	maxDelayMs: z.number().min(0).default(6e4),
+	jitter: z.boolean().default(true)
+});
+/**
+* Middleware that automatically retries failed tool calls with configurable backoff.
+*
+* Supports retrying on specific exceptions and exponential backoff.
+*
+* @example Basic usage with default settings (2 retries, exponential backoff)
+* ```ts
+* import { createAgent, toolRetryMiddleware } from "langchain";
+*
+* const agent = createAgent({
+*   model: "openai:gpt-4o",
+*   tools: [searchTool],
+*   middleware: [toolRetryMiddleware()],
+* });
+* ```
+*
+* @example Retry specific exceptions only
+* ```ts
+* import { toolRetryMiddleware } from "langchain";
+*
+* const retry = toolRetryMiddleware({
+*   maxRetries: 4,
+*   retryOn: [TimeoutError, NetworkError],
+*   backoffFactor: 1.5,
+* });
+* ```
+*
+* @example Custom exception filtering
+* ```ts
+* function shouldRetry(error: Error): boolean {
+*   // Only retry on 5xx errors
+*   if (error.name === "HTTPError" && "statusCode" in error) {
+*     const statusCode = (error as any).statusCode;
+*     return 500 <= statusCode && statusCode < 600;
+*   }
+*   return false;
+* }
+*
+* const retry = toolRetryMiddleware({
+*   maxRetries: 3,
+*   retryOn: shouldRetry,
+* });
+* ```
+*
+* @example Apply to specific tools with custom error handling
+* ```ts
+* const formatError = (error: Error) =>
+*   "Database temporarily unavailable. Please try again later.";
+*
+* const retry = toolRetryMiddleware({
+*   maxRetries: 4,
+*   tools: ["search_database"],
+*   onFailure: formatError,
+* });
+* ```
+*
+* @example Apply to specific tools using BaseTool instances
+* ```ts
+* import { tool } from "@langchain/core/tools";
+* import { z } from "zod";
+*
+* const searchDatabase = tool(
+*   async ({ query }) => {
+*     // Search implementation
+*     return results;
+*   },
+*   {
+*     name: "search_database",
+*     description: "Search the database",
+*     schema: z.object({ query: z.string() }),
+*   }
+* );
+*
+* const retry = toolRetryMiddleware({
+*   maxRetries: 4,
+*   tools: [searchDatabase], // Pass BaseTool instance
+* });
+* ```
+*
+* @example Constant backoff (no exponential growth)
+* ```ts
+* const retry = toolRetryMiddleware({
+*   maxRetries: 5,
+*   backoffFactor: 0.0, // No exponential growth
+*   initialDelayMs: 2000, // Always wait 2 seconds
+* });
+* ```
+*
+* @example Raise exception on failure
+* ```ts
+* const retry = toolRetryMiddleware({
+*   maxRetries: 2,
+*   onFailure: "raise", // Re-raise exception instead of returning message
+* });
+* ```
+*
+* @param config - Configuration options for the retry middleware
+* @returns A middleware instance that handles tool failures with retries
+*/
+function toolRetryMiddleware(config = {}) {
+	const { maxRetries, tools, retryOn, onFailure, backoffFactor, initialDelayMs, maxDelayMs, jitter } = ToolRetryMiddlewareOptionsSchema.parse(config);
+	const toolFilter = [];
+	for (const tool of tools ?? []) if (typeof tool === "string") toolFilter.push(tool);
+	else if ("name" in tool && typeof tool.name === "string") toolFilter.push(tool.name);
+	else throw new TypeError("Expected a tool name string or tool instance to be passed to toolRetryMiddleware");
+	/**
+	* Check if retry logic should apply to this tool.
+	*/
+	const shouldRetryTool = (toolName) => {
+		if (toolFilter.length === 0) return true;
+		return toolFilter.includes(toolName);
+	};
+	/**
+	* Check if the exception should trigger a retry.
+	*/
+	const shouldRetryException = (error) => {
+		if (typeof retryOn === "function") return retryOn(error);
+		return retryOn.some((ErrorConstructor) => error.constructor === ErrorConstructor);
+	};
+	const delayConfig = {
+		backoffFactor,
+		initialDelayMs,
+		maxDelayMs,
+		jitter
+	};
+	/**
+	* Format the failure message when retries are exhausted.
+	*/
+	const formatFailureMessage = (toolName, error, attemptsMade) => {
+		const errorType = error.constructor.name;
+		const attemptWord = attemptsMade === 1 ? "attempt" : "attempts";
+		return `Tool '${toolName}' failed after ${attemptsMade} ${attemptWord} with ${errorType}`;
+	};
+	/**
+	* Handle failure when all retries are exhausted.
+	*/
+	const handleFailure = (toolName, toolCallId, error, attemptsMade) => {
+		if (onFailure === "raise") throw error;
+		let content;
+		if (typeof onFailure === "function") content = onFailure(error);
+		else content = formatFailureMessage(toolName, error, attemptsMade);
+		return new ToolMessage({
+			content,
+			tool_call_id: toolCallId,
+			name: toolName,
+			status: "error"
+		});
+	};
+	return createMiddleware({
+		name: "toolRetryMiddleware",
+		contextSchema: ToolRetryMiddlewareOptionsSchema,
+		wrapToolCall: async (request, handler) => {
+			const toolName = request.tool.name;
+			if (!shouldRetryTool(toolName)) return handler(request);
+			const toolCallId = request.toolCall.id ?? "";
+			for (let attempt = 0; attempt <= maxRetries; attempt++) try {
+				return await handler(request);
+			} catch (error) {
+				const attemptsMade = attempt + 1;
+				const err = error && typeof error === "object" && "message" in error ? error : new Error(String(error));
+				if (!shouldRetryException(err)) return handleFailure(toolName, toolCallId, err, attemptsMade);
+				if (attempt < maxRetries) {
+					const delay = calculateRetryDelay(delayConfig, attempt);
+					if (delay > 0) await sleep(delay);
+				} else return handleFailure(toolName, toolCallId, err, attemptsMade);
+			}
+			throw new Error("Unexpected: retry loop completed without returning");
+		}
+	});
+}
+
+//#endregion
+export { toolRetryMiddleware };
+//# sourceMappingURL=toolRetry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"toolRetry.js","names":["config: {\n    backoffFactor: number;\n    initialDelayMs: number;\n    maxDelayMs: number;\n    jitter: boolean;\n  }","retryNumber: number","delay: number","config: ToolRetryMiddlewareConfig","toolFilter: string[]","toolName: string","error: Error","attemptsMade: number","toolCallId: string","content: string"],"sources":["../../../src/agents/middleware/toolRetry.ts"],"sourcesContent":["/**\n * Tool retry middleware for agents.\n */\nimport { z } from \"zod/v3\";\nimport { ToolMessage } from \"@langchain/core/messages\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\n\nimport { createMiddleware } from \"../middleware.js\";\nimport type { AgentMiddleware } from \"./types.js\";\nimport { sleep } from \"./utils.js\";\n\n/**\n * Calculate delay for a retry attempt with exponential backoff and jitter.\n *\n * @param retryNumber - The retry attempt number (0-indexed)\n * @param config - Configuration for backoff calculation\n * @returns Delay in milliseconds before next retry\n *\n * @internal Exported for testing purposes\n */\nexport function calculateRetryDelay(\n  config: {\n    backoffFactor: number;\n    initialDelayMs: number;\n    maxDelayMs: number;\n    jitter: boolean;\n  },\n  retryNumber: number\n): number {\n  const { backoffFactor, initialDelayMs, maxDelayMs, jitter } = config;\n\n  let delay: number;\n  if (backoffFactor === 0.0) {\n    delay = initialDelayMs;\n  } else {\n    delay = initialDelayMs * backoffFactor ** retryNumber;\n  }\n\n  // Cap at maxDelayMs\n  delay = Math.min(delay, maxDelayMs);\n\n  if (jitter && delay > 0) {\n    const jitterAmount = delay * 0.25;\n    delay = delay + (Math.random() * 2 - 1) * jitterAmount;\n    // Ensure delay is not negative after jitter\n    delay = Math.max(0, delay);\n  }\n\n  return delay;\n}\n\n/**\n * Configuration options for the Tool Retry Middleware.\n */\nexport const ToolRetryMiddlewareOptionsSchema = z.object({\n  /**\n   * Maximum number of retry attempts after the initial call.\n   * Default is 2 retries (3 total attempts). Must be >= 0.\n   */\n  maxRetries: z.number().min(0).default(2),\n\n  /**\n   * Optional list of tools or tool names to apply retry logic to.\n   * Can be a list of `BaseTool` instances or tool name strings.\n   * If `undefined`, applies to all tools. Default is `undefined`.\n   */\n  tools: z\n    .array(\n      z.union([z.custom<ClientTool>(), z.custom<ServerTool>(), z.string()])\n    )\n    .optional(),\n\n  /**\n   * Either an array of error constructors to retry on, or a function\n   * that takes an error and returns `true` if it should be retried.\n   * Default is to retry on all errors.\n   */\n  retryOn: z\n    .union([\n      z.function().args(z.instanceof(Error)).returns(z.boolean()),\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      z.array(z.custom<new (...args: any[]) => Error>()),\n    ])\n    .default(() => () => true),\n\n  /**\n   * Behavior when all retries are exhausted. Options:\n   * - `\"return_message\"` (default): Return a ToolMessage with error details,\n   *   allowing the LLM to handle the failure and potentially recover.\n   * - `\"raise\"`: Re-raise the exception, stopping agent execution.\n   * - Custom function: Function that takes the exception and returns a string\n   *   for the ToolMessage content, allowing custom error formatting.\n   */\n  onFailure: z\n    .union([\n      z.literal(\"raise\"),\n      z.literal(\"return_message\"),\n      z.function().args(z.instanceof(Error)).returns(z.string()),\n    ])\n    .default(\"return_message\"),\n\n  /**\n   * Multiplier for exponential backoff. Each retry waits\n   * `initialDelayMs * (backoffFactor ** retryNumber)` milliseconds.\n   * Set to 0.0 for constant delay. Default is 2.0.\n   */\n  backoffFactor: z.number().min(0).default(2.0),\n\n  /**\n   * Initial delay in milliseconds before first retry. Default is 1000 (1 second).\n   */\n  initialDelayMs: z.number().min(0).default(1000),\n\n  /**\n   * Maximum delay in milliseconds between retries. Caps exponential\n   * backoff growth. Default is 60000 (60 seconds).\n   */\n  maxDelayMs: z.number().min(0).default(60000),\n\n  /**\n   * Whether to add random jitter (±25%) to delay to avoid thundering herd.\n   * Default is `true`.\n   */\n  jitter: z.boolean().default(true),\n});\n\nexport type ToolRetryMiddlewareConfig = z.input<\n  typeof ToolRetryMiddlewareOptionsSchema\n>;\n\n/**\n * Middleware that automatically retries failed tool calls with configurable backoff.\n *\n * Supports retrying on specific exceptions and exponential backoff.\n *\n * @example Basic usage with default settings (2 retries, exponential backoff)\n * ```ts\n * import { createAgent, toolRetryMiddleware } from \"langchain\";\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   tools: [searchTool],\n *   middleware: [toolRetryMiddleware()],\n * });\n * ```\n *\n * @example Retry specific exceptions only\n * ```ts\n * import { toolRetryMiddleware } from \"langchain\";\n *\n * const retry = toolRetryMiddleware({\n *   maxRetries: 4,\n *   retryOn: [TimeoutError, NetworkError],\n *   backoffFactor: 1.5,\n * });\n * ```\n *\n * @example Custom exception filtering\n * ```ts\n * function shouldRetry(error: Error): boolean {\n *   // Only retry on 5xx errors\n *   if (error.name === \"HTTPError\" && \"statusCode\" in error) {\n *     const statusCode = (error as any).statusCode;\n *     return 500 <= statusCode && statusCode < 600;\n *   }\n *   return false;\n * }\n *\n * const retry = toolRetryMiddleware({\n *   maxRetries: 3,\n *   retryOn: shouldRetry,\n * });\n * ```\n *\n * @example Apply to specific tools with custom error handling\n * ```ts\n * const formatError = (error: Error) =>\n *   \"Database temporarily unavailable. Please try again later.\";\n *\n * const retry = toolRetryMiddleware({\n *   maxRetries: 4,\n *   tools: [\"search_database\"],\n *   onFailure: formatError,\n * });\n * ```\n *\n * @example Apply to specific tools using BaseTool instances\n * ```ts\n * import { tool } from \"@langchain/core/tools\";\n * import { z } from \"zod\";\n *\n * const searchDatabase = tool(\n *   async ({ query }) => {\n *     // Search implementation\n *     return results;\n *   },\n *   {\n *     name: \"search_database\",\n *     description: \"Search the database\",\n *     schema: z.object({ query: z.string() }),\n *   }\n * );\n *\n * const retry = toolRetryMiddleware({\n *   maxRetries: 4,\n *   tools: [searchDatabase], // Pass BaseTool instance\n * });\n * ```\n *\n * @example Constant backoff (no exponential growth)\n * ```ts\n * const retry = toolRetryMiddleware({\n *   maxRetries: 5,\n *   backoffFactor: 0.0, // No exponential growth\n *   initialDelayMs: 2000, // Always wait 2 seconds\n * });\n * ```\n *\n * @example Raise exception on failure\n * ```ts\n * const retry = toolRetryMiddleware({\n *   maxRetries: 2,\n *   onFailure: \"raise\", // Re-raise exception instead of returning message\n * });\n * ```\n *\n * @param config - Configuration options for the retry middleware\n * @returns A middleware instance that handles tool failures with retries\n */\nexport function toolRetryMiddleware(\n  config: ToolRetryMiddlewareConfig = {}\n): AgentMiddleware {\n  const {\n    maxRetries,\n    tools,\n    retryOn,\n    onFailure,\n    backoffFactor,\n    initialDelayMs,\n    maxDelayMs,\n    jitter,\n  } = ToolRetryMiddlewareOptionsSchema.parse(config);\n\n  // Extract tool names from BaseTool instances or strings\n  const toolFilter: string[] = [];\n  for (const tool of tools ?? []) {\n    if (typeof tool === \"string\") {\n      toolFilter.push(tool);\n    } else if (\"name\" in tool && typeof tool.name === \"string\") {\n      toolFilter.push(tool.name);\n    } else {\n      throw new TypeError(\n        \"Expected a tool name string or tool instance to be passed to toolRetryMiddleware\"\n      );\n    }\n  }\n\n  /**\n   * Check if retry logic should apply to this tool.\n   */\n  const shouldRetryTool = (toolName: string): boolean => {\n    if (toolFilter.length === 0) {\n      return true;\n    }\n    return toolFilter.includes(toolName);\n  };\n\n  /**\n   * Check if the exception should trigger a retry.\n   */\n  const shouldRetryException = (error: Error): boolean => {\n    if (typeof retryOn === \"function\") {\n      return retryOn(error);\n    }\n    // retryOn is an array of error constructors\n    return retryOn.some(\n      (ErrorConstructor) => error.constructor === ErrorConstructor\n    );\n  };\n\n  // Use the exported calculateRetryDelay function with our config\n  const delayConfig = { backoffFactor, initialDelayMs, maxDelayMs, jitter };\n\n  /**\n   * Format the failure message when retries are exhausted.\n   */\n  const formatFailureMessage = (\n    toolName: string,\n    error: Error,\n    attemptsMade: number\n  ): string => {\n    const errorType = error.constructor.name;\n    const attemptWord = attemptsMade === 1 ? \"attempt\" : \"attempts\";\n    return `Tool '${toolName}' failed after ${attemptsMade} ${attemptWord} with ${errorType}`;\n  };\n\n  /**\n   * Handle failure when all retries are exhausted.\n   */\n  const handleFailure = (\n    toolName: string,\n    toolCallId: string,\n    error: Error,\n    attemptsMade: number\n  ): ToolMessage => {\n    if (onFailure === \"raise\") {\n      throw error;\n    }\n\n    let content: string;\n    if (typeof onFailure === \"function\") {\n      content = onFailure(error);\n    } else {\n      content = formatFailureMessage(toolName, error, attemptsMade);\n    }\n\n    return new ToolMessage({\n      content,\n      tool_call_id: toolCallId,\n      name: toolName,\n      status: \"error\",\n    });\n  };\n\n  return createMiddleware({\n    name: \"toolRetryMiddleware\",\n    contextSchema: ToolRetryMiddlewareOptionsSchema,\n    wrapToolCall: async (request, handler) => {\n      const toolName = request.tool.name as string;\n\n      // Check if retry should apply to this tool\n      if (!shouldRetryTool(toolName)) {\n        return handler(request);\n      }\n\n      const toolCallId = request.toolCall.id ?? \"\";\n\n      // Initial attempt + retries\n      for (let attempt = 0; attempt <= maxRetries; attempt++) {\n        try {\n          return await handler(request);\n        } catch (error) {\n          const attemptsMade = attempt + 1; // attempt is 0-indexed\n\n          // Ensure error is an Error instance\n          const err =\n            error && typeof error === \"object\" && \"message\" in error\n              ? (error as Error)\n              : new Error(String(error));\n\n          // Check if we should retry this exception\n          if (!shouldRetryException(err)) {\n            // Exception is not retryable, handle failure immediately\n            return handleFailure(toolName, toolCallId, err, attemptsMade);\n          }\n\n          // Check if we have more retries left\n          if (attempt < maxRetries) {\n            // Calculate and apply backoff delay\n            const delay = calculateRetryDelay(delayConfig, attempt);\n            if (delay > 0) {\n              await sleep(delay);\n            }\n            // Continue to next retry\n          } else {\n            // No more retries, handle failure\n            return handleFailure(toolName, toolCallId, err, attemptsMade);\n          }\n        }\n      }\n\n      // Unreachable: loop always returns via handler success or handleFailure\n      throw new Error(\"Unexpected: retry loop completed without returning\");\n    },\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;AAoBA,SAAgB,oBACdA,QAMAC,aACQ;CACR,MAAM,EAAE,eAAe,gBAAgB,YAAY,QAAQ,GAAG;CAE9D,IAAIC;AACJ,KAAI,kBAAkB,GACpB,QAAQ;MAER,QAAQ,iBAAiB,iBAAiB;CAI5C,QAAQ,KAAK,IAAI,OAAO,WAAW;AAEnC,KAAI,UAAU,QAAQ,GAAG;EACvB,MAAM,eAAe,QAAQ;EAC7B,QAAQ,SAAS,KAAK,QAAQ,GAAG,IAAI,KAAK;EAE1C,QAAQ,KAAK,IAAI,GAAG,MAAM;CAC3B;AAED,QAAO;AACR;;;;AAKD,MAAa,mCAAmC,EAAE,OAAO;CAKvD,YAAY,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE;CAOxC,OAAO,EACJ,MACC,EAAE,MAAM;EAAC,EAAE,QAAoB;EAAE,EAAE,QAAoB;EAAE,EAAE,QAAQ;CAAC,EAAC,CACtE,CACA,UAAU;CAOb,SAAS,EACN,MAAM,CACL,EAAE,UAAU,CAAC,KAAK,EAAE,WAAW,MAAM,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,EAE3D,EAAE,MAAM,EAAE,QAAuC,CAAC,AACnD,EAAC,CACD,QAAQ,MAAM,MAAM,KAAK;CAU5B,WAAW,EACR,MAAM;EACL,EAAE,QAAQ,QAAQ;EAClB,EAAE,QAAQ,iBAAiB;EAC3B,EAAE,UAAU,CAAC,KAAK,EAAE,WAAW,MAAM,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC;CAC3D,EAAC,CACD,QAAQ,iBAAiB;CAO5B,eAAe,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAI;CAK7C,gBAAgB,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,IAAK;CAM/C,YAAY,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,IAAM;CAM5C,QAAQ,EAAE,SAAS,CAAC,QAAQ,KAAK;AAClC,EAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyGF,SAAgB,oBACdC,SAAoC,CAAE,GACrB;CACjB,MAAM,EACJ,YACA,OACA,SACA,WACA,eACA,gBACA,YACA,QACD,GAAG,iCAAiC,MAAM,OAAO;CAGlD,MAAMC,aAAuB,CAAE;AAC/B,MAAK,MAAM,QAAQ,SAAS,CAAE,EAC5B,KAAI,OAAO,SAAS,UAClB,WAAW,KAAK,KAAK;UACZ,UAAU,QAAQ,OAAO,KAAK,SAAS,UAChD,WAAW,KAAK,KAAK,KAAK;KAE1B,OAAM,IAAI,UACR;;;;CAQN,MAAM,kBAAkB,CAACC,aAA8B;AACrD,MAAI,WAAW,WAAW,EACxB,QAAO;AAET,SAAO,WAAW,SAAS,SAAS;CACrC;;;;CAKD,MAAM,uBAAuB,CAACC,UAA0B;AACtD,MAAI,OAAO,YAAY,WACrB,QAAO,QAAQ,MAAM;AAGvB,SAAO,QAAQ,KACb,CAAC,qBAAqB,MAAM,gBAAgB,iBAC7C;CACF;CAGD,MAAM,cAAc;EAAE;EAAe;EAAgB;EAAY;CAAQ;;;;CAKzE,MAAM,uBAAuB,CAC3BD,UACAC,OACAC,iBACW;EACX,MAAM,YAAY,MAAM,YAAY;EACpC,MAAM,cAAc,iBAAiB,IAAI,YAAY;AACrD,SAAO,CAAC,MAAM,EAAE,SAAS,eAAe,EAAE,aAAa,CAAC,EAAE,YAAY,MAAM,EAAE,WAAW;CAC1F;;;;CAKD,MAAM,gBAAgB,CACpBF,UACAG,YACAF,OACAC,iBACgB;AAChB,MAAI,cAAc,QAChB,OAAM;EAGR,IAAIE;AACJ,MAAI,OAAO,cAAc,YACvB,UAAU,UAAU,MAAM;OAE1B,UAAU,qBAAqB,UAAU,OAAO,aAAa;AAG/D,SAAO,IAAI,YAAY;GACrB;GACA,cAAc;GACd,MAAM;GACN,QAAQ;EACT;CACF;AAED,QAAO,iBAAiB;EACtB,MAAM;EACN,eAAe;EACf,cAAc,OAAO,SAAS,YAAY;GACxC,MAAM,WAAW,QAAQ,KAAK;AAG9B,OAAI,CAAC,gBAAgB,SAAS,CAC5B,QAAO,QAAQ,QAAQ;GAGzB,MAAM,aAAa,QAAQ,SAAS,MAAM;AAG1C,QAAK,IAAI,UAAU,GAAG,WAAW,YAAY,UAC3C,KAAI;AACF,WAAO,MAAM,QAAQ,QAAQ;GAC9B,SAAQ,OAAO;IACd,MAAM,eAAe,UAAU;IAG/B,MAAM,MACJ,SAAS,OAAO,UAAU,YAAY,aAAa,QAC9C,QACD,IAAI,MAAM,OAAO,MAAM;AAG7B,QAAI,CAAC,qBAAqB,IAAI,CAE5B,QAAO,cAAc,UAAU,YAAY,KAAK,aAAa;AAI/D,QAAI,UAAU,YAAY;KAExB,MAAM,QAAQ,oBAAoB,aAAa,QAAQ;AACvD,SAAI,QAAQ,GACV,MAAM,MAAM,MAAM;IAGrB,MAEC,QAAO,cAAc,UAAU,YAAY,KAAK,aAAa;GAEhE;AAIH,SAAM,IAAI,MAAM;EACjB;CACF,EAAC;AACH"}

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

@@ -1,8 +1,8 @@
 import { JumpToTarget } from "../constants.cjs";
-import { ClientTool, ServerTool } from "../tools.cjs";
 import { AgentBuiltInState, Runtime } from "../runtime.cjs";
 import { ModelRequest } from "../nodes/types.cjs";
 import { AIMessage, ToolMessage } from "@langchain/core/messages";
+import { ClientTool, ServerTool } from "@langchain/core/tools";
 import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
 import { ToolCall as ToolCall$1 } from "@langchain/core/messages/tool";
 import { AnnotationRoot, Command } from "@langchain/langgraph";
@@ -11,11 +11,13 @@ import { InteropZodToStateDefinition } from "@langchain/langgraph/zod";
 //#region src/agents/middleware/types.d.ts
 type PromiseOrValue<T> = T | Promise<T>;
 type AnyAnnotationRoot = AnnotationRoot<any>;
-type NormalizedSchemaInput<TSchema extends InteropZodObject | undefined = any> = TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {};
+type NormalizedSchemaInput<TSchema extends InteropZodObject | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends Record<string, unknown> ? TSchema & AgentBuiltInState : AgentBuiltInState;
 /**
  * Result type for middleware functions.
  */
-type MiddlewareResult<TState> = TState | void;
+type MiddlewareResult<TState> = (TState & {
+  jumpTo?: JumpToTarget;
+}) | void;
 /**
  * Represents a tool call request for the wrapToolCall hook.
  * Contains the tool call information along with the agent's current state and runtime.
@@ -43,12 +45,12 @@ interface ToolCallRequest<TState extends Record<string, unknown> = Record<string
  * Handler function type for wrapping tool calls.
  * Takes a tool call request and returns the tool result or a command.
  */
-type ToolCallHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>) => PromiseOrValue<ToolMessage | Command>;
+type ToolCallHandler<TSchema extends Record<string, unknown> = AgentBuiltInState, TContext = unknown> = (request: ToolCallRequest<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;
 /**
  * Wrapper function type for the wrapToolCall hook.
  * Allows middleware to intercept and modify tool execution.
  */
-type WrapToolCallHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>, handler: ToolCallHandler<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;
+type WrapToolCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<ToolMessage | Command>;
 /**
  * Handler function type for wrapping model calls.
  * Takes a model request and returns the AI message response.
@@ -56,7 +58,7 @@ type WrapToolCallHook<TSchema extends InteropZodObject | undefined = any, TConte
  * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime
  * @returns The AI message response from the model
  */
-type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>) => PromiseOrValue<AIMessage>;
+type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<AIMessage>;
 /**
  * Wrapper function type for the wrapModelCall hook.
  * Allows middleware to intercept and modify model execution.
@@ -70,7 +72,7 @@ type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = any, TC
  * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response
  * @returns The AI message response from the model (or a modified version)
  */
-type WrapModelCallHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;
+type WrapModelCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;
 /**
  * Handler function type for the beforeAgent hook.
  * Called once at the start of agent invocation before any model calls or tool executions.
@@ -79,14 +81,14 @@ type WrapModelCallHook<TSchema extends InteropZodObject | undefined = any, TCont
  * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.
  * @returns A middleware result containing partial state updates or undefined to pass through
  */
-type BeforeAgentHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;
+type BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
 /**
  * Hook type for the beforeAgent lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called once at the start of the agent invocation.
  */
-type BeforeAgentHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = BeforeAgentHandler<TSchema, TContext> | {
-  hook: BeforeAgentHandler<TSchema, TContext>;
+type BeforeAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -97,14 +99,14 @@ type BeforeAgentHook<TSchema extends InteropZodObject | undefined = any, TContex
  * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.
  * @returns A middleware result containing partial state updates or undefined to pass through
  */
-type BeforeModelHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;
+type BeforeModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
 /**
  * Hook type for the beforeModel lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called before each model invocation.
  */
-type BeforeModelHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = BeforeModelHandler<TSchema, TContext> | {
-  hook: BeforeModelHandler<TSchema, TContext>;
+type BeforeModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -116,14 +118,14 @@ type BeforeModelHook<TSchema extends InteropZodObject | undefined = any, TContex
  * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.
  * @returns A middleware result containing partial state updates or undefined to pass through
  */
-type AfterModelHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;
+type AfterModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
 /**
  * Hook type for the afterModel lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called after each model invocation.
  */
-type AfterModelHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = AfterModelHandler<TSchema, TContext> | {
-  hook: AfterModelHandler<TSchema, TContext>;
+type AfterModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -134,14 +136,14 @@ type AfterModelHook<TSchema extends InteropZodObject | undefined = any, TContext
  * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.
  * @returns A middleware result containing partial state updates or undefined to pass through
  */
-type AfterAgentHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;
+type AfterAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
 /**
  * Hook type for the afterAgent lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called once at the end of the agent invocation.
  */
-type AfterAgentHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = AfterAgentHandler<TSchema, TContext> | {
-  hook: AfterAgentHandler<TSchema, TContext>;
+type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","InteropZodToStateDefinition","AnnotationRoot","AIMessage","ToolMessage","ToolCall","Command","JumpToTarget","ClientTool","ServerTool","Runtime","AgentBuiltInState","ModelRequest","PromiseOrValue","T","Promise","AnyAnnotationRoot","NormalizedSchemaInput","TSchema","MiddlewareResult","TState","ToolCallRequest","Record","TContext","ToolCallHandler","WrapToolCallHook","WrapModelCallHandler","WrapModelCallHook","BeforeAgentHandler","Partial","BeforeAgentHook","BeforeModelHandler","BeforeModelHook","AfterModelHandler","AfterModelHook","AfterAgentHandler","AfterAgentHook","AgentMiddleware","TContextSchema","TFullContext","FilterPrivateProps","K","InferChannelType","ToAnnotationRoot","InferMiddlewareState","S","InferMiddlewareInputState","InferMiddlewareStates","First","Rest","InferMiddlewareInputStates","InferMergedState","InferMergedInputState","InferMiddlewareContext","C","InferMiddlewareContextInput","Inner","InferMiddlewareContexts","MergeContextTypes","A","B","InferMiddlewareContextInputs","InferContextInput","ContextSchema","InferSchemaInput"],"sources":["../../../src/agents/middleware/types.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot } from \"@langchain/langgraph\";\nimport type { AIMessage, ToolMessage } from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command } from \"@langchain/langgraph\";\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { ClientTool, ServerTool } from \"../tools.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\ntype PromiseOrValue<T> = T | Promise<T>;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\ntype NormalizedSchemaInput<TSchema extends InteropZodObject | undefined = any> = TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {};\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> = TState | void;\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<TState extends Record<string, unknown> = Record<string, unknown>, TContext = unknown> {\n    /**\n     * The tool call to be executed\n     */\n    toolCall: ToolCall;\n    /**\n     * The BaseTool instance being invoked.\n     * Provides access to tool metadata like name, description, schema, etc.\n     */\n    tool: ClientTool | ServerTool;\n    /**\n     * The current agent state (includes both middleware state and built-in state).\n     */\n    state: TState & AgentBuiltInState;\n    /**\n     * The runtime context containing metadata, signal, writer, interrupt, etc.\n     */\n    runtime: Runtime<TContext>;\n}\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>, handler: ToolCallHandler<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema> & AgentBuiltInState, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\nexport type BeforeAgentHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = BeforeAgentHandler<TSchema, TContext> | {\n    hook: BeforeAgentHandler<TSchema, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\nexport type BeforeModelHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = BeforeModelHandler<TSchema, TContext> | {\n    hook: BeforeModelHandler<TSchema, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\nexport type AfterModelHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = AfterModelHandler<TSchema, TContext> | {\n    hook: AfterModelHandler<TSchema, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\nexport type AfterAgentHandler<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = (state: NormalizedSchemaInput<TSchema> & AgentBuiltInState, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<NormalizedSchemaInput<TSchema>>>>;\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<TSchema extends InteropZodObject | undefined = any, TContext = unknown> = AfterAgentHandler<TSchema, TContext> | {\n    hook: AfterAgentHandler<TSchema, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Base middleware interface.\n */\nexport interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {\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     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * Wraps tool execution with custom logic. This allows you to:\n     * - Modify tool call parameters before execution\n     * - Handle errors and retry with different parameters\n     * - Post-process tool results\n     * - Implement caching, logging, authentication, or other cross-cutting concerns\n     * - Return Command objects for advanced control flow\n     *\n     * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n     * along with a handler function to execute the actual tool.\n     *\n     * @param request - The tool call request containing toolCall, state, and runtime.\n     * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n     * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n     *\n     * @example\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   console.log(`Calling tool: ${request.tool.name}`);\n     *   console.log(`Tool description: ${request.tool.description}`);\n     *\n     *   try {\n     *     // Execute the tool\n     *     const result = await handler(request);\n     *     console.log(`Tool ${request.tool.name} succeeded`);\n     *     return result;\n     *   } catch (error) {\n     *     console.error(`Tool ${request.tool.name} failed:`, error);\n     *     // Could return a custom error message or retry\n     *     throw error;\n     *   }\n     * }\n     * ```\n     *\n     * @example Authentication\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   // Check if user is authorized for this tool\n     *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n     *     return new ToolMessage({\n     *       content: \"Unauthorized to call this tool\",\n     *       tool_call_id: request.toolCall.id,\n     *     });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     *\n     * @example Caching\n     * ```ts\n     * const cache = new Map();\n     * wrapToolCall: async (request, handler) => {\n     *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n     *   if (cache.has(cacheKey)) {\n     *     return cache.get(cacheKey);\n     *   }\n     *   const result = await handler(request);\n     *   cache.set(cacheKey, result);\n     *   return result;\n     * }\n     * ```\n     */\n    wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n    /**\n     * Wraps the model invocation with custom logic. This allows you to:\n     * - Modify the request before calling the model\n     * - Handle errors and retry with different parameters\n     * - Post-process the response\n     * - Implement custom caching, logging, or other cross-cutting concerns\n     *\n     * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n     * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n     * @returns The response from the model (or a modified version).\n     *\n     * @example\n     * ```ts\n     * wrapModelCall: async (request, handler) => {\n     *   // Modify request before calling\n     *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n     *\n     *   try {\n     *     // Call the model\n     *     return await handler(modifiedRequest);\n     *   } catch (error) {\n     *     // Handle errors and retry with fallback\n     *     const fallbackRequest = { ...request, model: fallbackModel };\n     *     return await handler(fallbackRequest);\n     *   }\n     * }\n     * ```\n     */\n    wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n     * It allows to modify the state of the agent before any model calls or tool executions.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n     * It allows to modify the state of the agent.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n     * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterModel?: AfterModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n     * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n    [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>[\"State\"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<S>> : {} : {};\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<S>> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T = AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined] ? undefined : B | undefined : [B] extends [undefined] ? A | undefined : [A] extends [B] ? A : [B] extends [A] ? B : A & B;\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>[\"State\"] : {};\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>[\"State\"] : {};\nexport {};\n"],"mappings":";;;;;;;;;;;AAUsD,KACjDiB,cAAAA,CAAAA,CAAc,CAAA,GAAMC,CAAN,GAAUC,OAAV,CAAkBD,CAAlB,CAAA;AAAA,KACPE,iBAAAA,GAAoBd,cADb,CAAA,GAAA,CAAA;KAEde,qBAFoBH,CAAAA,gBAEkBlB,gBAFlBkB,GAAAA,SAAAA,GAAAA,GAAAA,CAAAA,GAEwDI,OAFxDJ,SAEwElB,gBAFxEkB,GAE2Ff,oBAF3Fe,CAEgHI,OAFhHJ,CAAAA,GAAAA,CAAAA,CAAAA;;;AAAW;AACxBE,KAKAG,gBALiB,CAAA,MAAGjB,CAAAA,GAKOkB,MALPlB,GAAAA,IAAc;AAAM;;;;AAC6CN,UAShFyB,eATgFzB,CAAAA,eASjD0B,MATiD1B,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GASvB0B,MATuB1B,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,WAAAA,OAAAA,CAAAA,CAAAA;EAAgB;;AAAuB;EAI5HuB,QAAAA,EASEd,UATFc;EAKKE;;;;EAA+D,IAIlEhB,EAKJG,UALIH,GAKSI,UALTJ;EAAQ;;;EASL,KAAGM,EAATS,MAAST,GAAAA,iBAAAA;EAAiB;;AAIjB;EAMRa,OAAAA,EANCd,OAMDc,CANSD,QAMM,CAAA;;;;;;AAAyJA,KAAxKC,eAAwKD,CAAAA,gBAAxI3B,gBAAwI2B,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAApEF,eAAoEE,CAApDN,qBAAoDM,CAA9BL,OAA8BK,CAAAA,GAAnBZ,iBAAmBY,EAAAA,QAAAA,CAAAA,EAAAA,GAAcV,cAAdU,CAA6BnB,WAA7BmB,GAA2CjB,OAA3CiB,CAAAA;;;;;AAA4B,KAKpME,gBALoM,CAAA,gBAKnK7B,gBALmK,GAAA,SAAA,GAAA,GAAA,EAAA,WAAA,OAAA,CAAA,GAAA,CAAA,OAAA,EAK/FyB,eAL+F,CAK/EJ,qBAL+E,CAKzDC,OALyD,CAAA,GAK9CP,iBAL8C,EAK3BY,QAL2B,CAAA,EAAA,OAAA,EAKPC,eALO,CAKSN,OALT,EAKkBK,QALlB,CAAA,EAAA,GAKgCV,cALhC,CAK+CT,WAL/C,GAK6DE,OAL7D,CAAA;AAKhN;;;;;;;AAAiHe,KAQrGK,oBARqGL,CAAAA,gBAQhEzB,gBARgEyB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAQIT,YARJS,CAQiBJ,qBARjBI,CAQuCH,OARvCG,CAAAA,GAQkDV,iBARlDU,EAQqEE,QARrEF,CAAAA,EAAAA,GAQmFR,cARnFQ,CAQkGlB,SARlGkB,CAAAA;;;;;;;AAA6I;AAQ9P;;;;;;AAAsLE,KAc1KI,iBAd0KJ,CAAAA,gBAcxI3B,gBAdwI2B,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAcpEX,YAdoEW,CAcvDN,qBAduDM,CAcjCL,OAdiCK,CAAAA,GActBZ,iBAdsBY,EAcHA,QAdGA,CAAAA,EAAAA,OAAAA,EAciBG,oBAdjBH,CAcsCL,OAdtCK,EAc+CA,QAd/CA,CAAAA,EAAAA,GAc6DV,cAd7DU,CAc4EpB,SAd5EoB,CAAAA;;;;AAA4B;AAclN;;;;AAA+HN,KASnHW,kBATmHX,CAAAA,gBAShFrB,gBATgFqB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EASdA,qBATcA,CASQC,OATRD,CAAAA,GASmBN,iBATnBM,EAAAA,OAAAA,EAS+CP,OAT/CO,CASuDM,QATvDN,CAAAA,EAAAA,GASqEJ,cATrEI,CASoFE,gBATpFF,CASqGY,OATrGZ,CAS6GA,qBAT7GA,CASmIC,OATnID,CAAAA,CAAAA,CAAAA,CAAAA;;;;;;AAAwES,KAe3LI,eAf2LJ,CAAAA,gBAe3J9B,gBAf2J8B,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAejGE,kBAfiGF,CAe9ER,OAf8EQ,EAerEH,QAfqEG,CAAAA,GAAAA;EAAoB,IAAuCvB,EAgBxPyB,kBAhBwPzB,CAgBrOe,OAhBqOf,EAgB5NoB,QAhB4NpB,CAAAA;EAAS,SAAxBU,CAAAA,EAiBnON,YAjBmOM,EAAAA;AAAc,CAAA;AASjQ;;;;;;;;AAAkQK,KAkBtPa,kBAlBsPb,CAAAA,gBAkBnNtB,gBAlBmNsB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAkBjJD,qBAlBiJC,CAkB3HA,OAlB2HA,CAAAA,GAkBhHP,iBAlBgHO,EAAAA,OAAAA,EAkBpFR,OAlBoFQ,CAkB5EK,QAlB4EL,CAAAA,EAAAA,GAkB9DL,cAlB8DK,CAkB/CC,gBAlB+CD,CAkB9BW,OAlB8BX,CAkBtBD,qBAlBsBC,CAkBAA,OAlBAA,CAAAA,CAAAA,CAAAA,CAAAA;;;;;AAAhD;AAMtMY,KAkBAE,eAlBe,CAAA,gBAkBiBpC,gBAlBjB,GAAA,SAAA,GAAA,GAAA,EAAA,WAAA,OAAA,CAAA,GAkB2EmC,kBAlB3E,CAkB8Fb,OAlB9F,EAkBuGK,QAlBvG,CAAA,GAAA;EAAA,IAAA,EAmBjBQ,kBAnBiB,CAmBEb,OAnBF,EAmBWK,QAnBX,CAAA;EAAA,SAAiB3B,CAAAA,EAoB5BW,YApB4BX,EAAAA;CAAgB;;;;;;;AAEhC;AAU5B;;AAA+CA,KAmBnCqC,iBAnBmCrC,CAAAA,gBAmBDA,gBAnBCA,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAmBiEqB,qBAnBjErB,CAmBuFsB,OAnBvFtB,CAAAA,GAmBkGe,iBAnBlGf,EAAAA,OAAAA,EAmB8Hc,OAnB9Hd,CAmBsI2B,QAnBtI3B,CAAAA,EAAAA,GAmBoJiB,cAnBpJjB,CAmBmKuB,gBAnBnKvB,CAmBoLiC,OAnBpLjC,CAmB4LqB,qBAnB5LrB,CAmBkNsB,OAnBlNtB,CAAAA,CAAAA,CAAAA,CAAAA;;;;;;AAAmNsB,KAyBtPgB,cAzBsPhB,CAAAA,gBAyBvNtB,gBAzBuNsB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAyB7Je,iBAzB6Jf,CAyB3IA,OAzB2IA,EAyBlIK,QAzBkIL,CAAAA,GAAAA;EAAO,IAA7BD,EA0BlOgB,iBA1BkOhB,CA0BhNC,OA1BgND,EA0BvMM,QA1BuMN,CAAAA;EAAqB,SAA7BY,CAAAA,EA2BpNtB,YA3BoNsB,EAAAA;CAAO;;AAAzB;AAMlN;;;;;;AAC6BX,KA8BjBiB,iBA9BiBjB,CAAAA,gBA8BiBtB,gBA9BjBsB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EA8BmFD,qBA9BnFC,CA8ByGA,OA9BzGA,CAAAA,GA8BoHP,iBA9BpHO,EAAAA,OAAAA,EA8BgJR,OA9BhJQ,CA8BwJK,QA9BxJL,CAAAA,EAAAA,GA8BsKL,cA9BtKK,CA8BqLC,gBA9BrLD,CA8BsMW,OA9BtMX,CA8B8MD,qBA9B9MC,CA8BoOA,OA9BpOA,CAAAA,CAAAA,CAAAA,CAAAA;;;;AACD;AAW5B;AAA6B,KAwBjBkB,cAxBiB,CAAA,gBAwBcxC,gBAxBd,GAAA,SAAA,GAAA,GAAA,EAAA,WAAA,OAAA,CAAA,GAwBwEuC,iBAxBxE,CAwB0FjB,OAxB1F,EAwBmGK,QAxBnG,CAAA,GAAA;EAAA,IAAiB3B,EAyBpCuC,iBAzBoCvC,CAyBlBsB,OAzBkBtB,EAyBT2B,QAzBS3B,CAAAA;EAAgB,SAAwEsB,CAAAA,EA0BtHX,YA1BsHW,EAAAA;CAAO;;;;AAAoHA,UA+BhPmB,eA/BgPnB,CAAAA,gBA+BhNtB,gBA/BgNsB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,uBA+BrJtB,gBA/BqJsB,GA+BlIrB,iBA/BkIqB,CA+BhHtB,gBA/BgHsB,CAAAA,GA+B5FpB,kBA/B4FoB,CA+BzEtB,gBA/ByEsB,CAAAA,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,eAAAA,GAAAA,CAAAA,CAAAA;EAAO;;;EAAtC,IAA/BL,EAAAA,MAAAA;EAAc;AAMjN;;;;;;EAAsH,WAC1FK,CAAAA,EAoCVA,OApCUA;EAAO;;;AACP;AAU5B;;;EAA8D,aAAwEA,CAAAA,EAiClHoB,cAjCkHpB;EAAO;;;EAAgD,KAAhBR,CAAAA,EAAAA,CAqChKF,UArCgKE,GAqCnJD,UArCmJC,CAAAA,EAAAA;EAAO;;;;;AAA6B;AAMjN;;;;;;;;;;AAE4B;AAK5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwJ+B;AAC9B;;;;;;;AAK4D;AAO7D;;;;;;EAA8G,YAASd,CAAAA,EA9EpG6B,gBA8EoG7B,CA9EnFsB,OA8EmFtB,EA9E1E2C,YA8E0E3C,CAAAA;EAAgB;;;AAAqB;AAK5J;;;;;;;;;;AAAiK;AAIjK;;;;;;;;;;;;;EAAuQ,aAA8BoD,CAAAA,EA1DjRrB,iBA0DiRqB,CA1D/P9B,OA0D+P8B,EA1DtPT,YA0DsPS,CAAAA;EAAK;AAAN;AAIpS;;;;;;EAAmK,WAASX,CAAAA,EArD1JP,eAqD0JO,CArD1InB,OAqD0ImB,EArDjIE,YAqDiIF,CAAAA;EAAe;;;;;;;;EAA4I,WAAA,CAAA,EA5CrTL,eA4CqT,CA5CrSd,OA4CqS,EA5C5RqB,YA4C4R,CAAA;EAgB3TgB;;;;;;;;EAA8K,UAA1BxD,CAAAA,EAnD/ImC,cAmD+InC,CAnDhImB,OAmDgInB,EAnDvHwC,YAmDuHxC,CAAAA;EAAoB;;;;AAAuE;AAIkF;;;EAI5S,UAAyB6D,CAAAA,EAlDzCxB,cAkDyCwB,CAlD1B1C,OAkD0B0C,EAlDjBrB,YAkDiBqB,CAAAA;;;;;KA7CrDpB,kBA6CiKoB,CAAAA,CAAAA,CAAAA,GAAAA,QAAKD,MA5C3J7C,CA4C2J6C,IA5CtJlB,CA4CsJkB,SAAAA,IAAAA,MAAAA,EAAAA,GAAAA,KAAAA,GA5CrHlB,CA4CqHkB,GA5CjH7C,CA4CiH6C,CA5C/GlB,CA4C+GkB,CAAAA,EAAC;;;;AAA8B;AAI9LE,KAzCAjB,oBAyCAiB,CAAAA,UAzC+BxB,eAyCH,CAAA,GAzCsBvB,CAyCtB,SAzCgCuB,eAyChC,CAAA,KAAA,EAAA,EAAA,GAAA,EAAA,GAAA,CAAA,GAzCqEQ,CAyCrE,SAzC+EjD,gBAyC/E,GAzCkG4C,kBAyClG,CAzCqHxC,qBAyCrH,CAzC2I6C,CAyC3I,CAAA,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;;;;;AAAwHG,KApCpJF,yBAoCoJE,CAAAA,UApChHX,eAoCgHW,CAAAA,GApC7FlC,CAoC6FkC,SApCnFX,eAoCmFW,CAAAA,KAAAA,EAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GApC9CH,CAoC8CG,SApCpCpD,gBAoCoCoD,GApCjBR,kBAoCiBQ,CApCEjD,oBAoCFiD,CApCuBH,CAoCvBG,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAAwHA,KAhC5QD,qBAgC4QC,CAAAA,IAhClPX,eAgCkPW,EAAAA,CAAAA,GAhC7NlC,CAgC6NkC,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAhChMlC,CAgCgMkC,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAhC9IA,KAgC8IA,SAhChIX,eAgCgIW,GAhC9GC,IAgC8GD,SAAAA,SAhCxFX,eAgCwFW,EAAAA,GAhCpEJ,oBAgCoEI,CAhC/CA,KAgC+CA,CAAAA,GAhCtCD,qBAgCsCC,CAhChBC,IAgCgBD,CAAAA,GAhCRJ,oBAgCQI,CAhCaA,KAgCbA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA9CU,KA5B9NR,0BA4B8NQ,CAAAA,UAAAA,SA5BhLrB,eA4BgLqB,EAAAA,CAAAA,GA5B3J5C,CA4B2J4C,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GA5B9H5C,CA4B8H4C,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GA5B5EV,KA4B4EU,SA5B9DrB,eA4B8DqB,GA5B5CT,IA4B4CS,SAAAA,SA5BtBrB,eA4BsBqB,EAAAA,GA5BFZ,yBA4BEY,CA5BwBV,KA4BxBU,CAAAA,GA5BiCR,0BA4BjCQ,CA5B4DT,IA4B5DS,CAAAA,GA5BoEZ,yBA4BpEY,CA5B8FV,KA4B9FU,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;AAAuH;;;;;AAI5H,KAhBzNH,2BAgByN,CAAA,UAhBnLlB,eAgBmL,CAAA,GAhBhKvB,CAgBgK,SAhBtJuB,eAgBsJ,CAAA,GAAA,EAAA,KAAA,EAAA,EAAA,GAAA,CAAA,GAhBjHiB,CAgBiH,SAhBvGxD,kBAgBuG,CAAA,KAAA,MAAA,CAAA,GAhBrEC,oBAgBqE,CAhBhDyD,KAgBgD,CAAA,GAAA,SAAA,GAhB3BF,CAgB2B,SAhBjB1D,gBAgBiB,GAhBEG,oBAgBF,CAhBuBuD,CAgBvB,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;AACrO;;;;;;;KATKI,iBAS4GC,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CATjFA,CASiFA,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,CATvDC,CASuDD,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,SAAAA,GATlBC,CASkBD,GAAAA,SAAAA,GAAAA,CATDC,CASCD,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GATwBA,CASxBA,GAAAA,SAAAA,GAAAA,CATyCA,CASzCA,CAAAA,SAAAA,CATqDC,CASrDD,CAAAA,GAT0DA,CAS1DA,GAAAA,CAT+DC,CAS/DD,CAAAA,SAAAA,CAT2EA,CAS3EA,CAAAA,GATgFC,CAShFD,GAToFA,CASpFA,GATwFC,CASxFD;;;;AAA6BzD,KALlI2D,4BAKkI3D,CAAAA,UAAAA,SALlFmC,eAKkFnC,EAAAA,CAAAA,GAL7DY,CAK6DZ,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GALhCY,CAKgCZ,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GALkB8C,KAKlB9C,SALgCmC,eAKhCnC,GALkD+C,IAKlD/C,SAAAA,SALwEmC,eAKxEnC,EAAAA,GAL4FwD,iBAK5FxD,CAL8GqD,2BAK9GrD,CAL0I8C,KAK1I9C,CAAAA,EALkJ2D,4BAKlJ3D,CAL+K+C,IAK/K/C,CAAAA,CAAAA,GALwLqD,2BAKxLrD,CALoN8C,KAKpN9C,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAc;AAC5J;;AAAuCc,KAF3B8C,iBAE2B9C,CAAAA,sBAFaA,iBAEbA,GAFiCpB,gBAEjCoB,CAAAA,GAFqD+C,aAErD/C,SAF2EpB,gBAE3EoB,GAF8FjB,oBAE9FiB,CAFmH+C,aAEnH/C,CAAAA,GAFoI+C,aAEpI/C,SAF0JA,iBAE1JA,GAF8K2B,gBAE9K3B,CAF+L+C,aAE/L/C,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAoBpB,KAD/C+C,gBAC+C/C,CAAAA,UADpBoB,iBACoBpB,GADAA,gBACAA,CAAAA,GADoB+D,CACpB/D,SAD8BoB,iBAC9BpB,GADkD+D,CAClD/D,GADsD+D,CACtD/D,SADgEA,gBAChEA,GADmFM,cACnFN,CADkGK,2BAClGL,CAD8H+D,CAC9H/D,CAAAA,CAAAA,GAAAA,KAAAA;AAAgC+D,KAA/EK,gBAA+EL,CAAAA,UAApD3C,iBAAoD2C,GAAhC/D,gBAAgC+D,GAAAA,SAAAA,CAAAA,GAAAA,CAAAA,SAAU3C,iBAAV2C,GAA8B/D,gBAA9B+D,GAAiDhB,gBAAjDgB,CAAkEA,CAAlEA,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA"}
+{"version":3,"file":"types.d.cts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","InteropZodToStateDefinition","AnnotationRoot","AIMessage","ToolMessage","ToolCall","Command","ClientTool","ServerTool","JumpToTarget","Runtime","AgentBuiltInState","ModelRequest","PromiseOrValue","T","Promise","AnyAnnotationRoot","NormalizedSchemaInput","TSchema","Record","MiddlewareResult","TState","ToolCallRequest","TContext","ToolCallHandler","WrapToolCallHook","WrapModelCallHandler","WrapModelCallHook","BeforeAgentHandler","Partial","BeforeAgentHook","BeforeModelHandler","BeforeModelHook","AfterModelHandler","AfterModelHook","AfterAgentHandler","AfterAgentHook","AgentMiddleware","TContextSchema","TFullContext","FilterPrivateProps","K","InferChannelType","ToAnnotationRoot","InferMiddlewareState","S","InferMiddlewareInputState","InferMiddlewareStates","First","Rest","InferMiddlewareInputStates","InferMergedState","InferMergedInputState","InferMiddlewareContext","C","InferMiddlewareContextInput","Inner","InferMiddlewareContexts","MergeContextTypes","A","B","InferMiddlewareContextInputs","InferContextInput","ContextSchema","InferSchemaInput"],"sources":["../../../src/agents/middleware/types.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot } from \"@langchain/langgraph\";\nimport type { AIMessage, ToolMessage } from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command } from \"@langchain/langgraph\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\ntype PromiseOrValue<T> = T | Promise<T>;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\nexport type NormalizedSchemaInput<TSchema extends InteropZodObject | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends Record<string, unknown> ? TSchema & AgentBuiltInState : AgentBuiltInState;\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> = (TState & {\n    jumpTo?: JumpToTarget;\n}) | void;\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<TState extends Record<string, unknown> = Record<string, unknown>, TContext = unknown> {\n    /**\n     * The tool call to be executed\n     */\n    toolCall: ToolCall;\n    /**\n     * The BaseTool instance being invoked.\n     * Provides access to tool metadata like name, description, schema, etc.\n     */\n    tool: ClientTool | ServerTool;\n    /**\n     * The current agent state (includes both middleware state and built-in state).\n     */\n    state: TState & AgentBuiltInState;\n    /**\n     * The runtime context containing metadata, signal, writer, interrupt, etc.\n     */\n    runtime: Runtime<TContext>;\n}\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<TSchema extends Record<string, unknown> = AgentBuiltInState, TContext = unknown> = (request: ToolCallRequest<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Base middleware interface.\n */\nexport interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {\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     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * Wraps tool execution with custom logic. This allows you to:\n     * - Modify tool call parameters before execution\n     * - Handle errors and retry with different parameters\n     * - Post-process tool results\n     * - Implement caching, logging, authentication, or other cross-cutting concerns\n     * - Return Command objects for advanced control flow\n     *\n     * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n     * along with a handler function to execute the actual tool.\n     *\n     * @param request - The tool call request containing toolCall, state, and runtime.\n     * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n     * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n     *\n     * @example\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   console.log(`Calling tool: ${request.tool.name}`);\n     *   console.log(`Tool description: ${request.tool.description}`);\n     *\n     *   try {\n     *     // Execute the tool\n     *     const result = await handler(request);\n     *     console.log(`Tool ${request.tool.name} succeeded`);\n     *     return result;\n     *   } catch (error) {\n     *     console.error(`Tool ${request.tool.name} failed:`, error);\n     *     // Could return a custom error message or retry\n     *     throw error;\n     *   }\n     * }\n     * ```\n     *\n     * @example Authentication\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   // Check if user is authorized for this tool\n     *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n     *     return new ToolMessage({\n     *       content: \"Unauthorized to call this tool\",\n     *       tool_call_id: request.toolCall.id,\n     *     });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     *\n     * @example Caching\n     * ```ts\n     * const cache = new Map();\n     * wrapToolCall: async (request, handler) => {\n     *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n     *   if (cache.has(cacheKey)) {\n     *     return cache.get(cacheKey);\n     *   }\n     *   const result = await handler(request);\n     *   cache.set(cacheKey, result);\n     *   return result;\n     * }\n     * ```\n     */\n    wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n    /**\n     * Wraps the model invocation with custom logic. This allows you to:\n     * - Modify the request before calling the model\n     * - Handle errors and retry with different parameters\n     * - Post-process the response\n     * - Implement custom caching, logging, or other cross-cutting concerns\n     *\n     * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n     * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n     * @returns The response from the model (or a modified version).\n     *\n     * @example\n     * ```ts\n     * wrapModelCall: async (request, handler) => {\n     *   // Modify request before calling\n     *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n     *\n     *   try {\n     *     // Call the model\n     *     return await handler(modifiedRequest);\n     *   } catch (error) {\n     *     // Handle errors and retry with fallback\n     *     const fallbackRequest = { ...request, model: fallbackModel };\n     *     return await handler(fallbackRequest);\n     *   }\n     * }\n     * ```\n     */\n    wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n     * It allows to modify the state of the agent before any model calls or tool executions.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n     * It allows to modify the state of the agent.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n     * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterModel?: AfterModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n     * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n    [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>[\"State\"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<S>> : {} : {};\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<S>> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T = AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined] ? undefined : B | undefined : [B] extends [undefined] ? A | undefined : [A] extends [B] ? A : [B] extends [A] ? B : A & B;\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>[\"State\"] : {};\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>[\"State\"] : {};\nexport {};\n"],"mappings":";;;;;;;;;;;AAUsD,KACjDiB,cAAAA,CAAAA,CAAc,CAAA,GAAMC,CAAN,GAAUC,OAAV,CAAkBD,CAAlB,CAAA;AAAA,KACPE,iBAAAA,GAAoBd,cADb,CAAA,GAAA,CAAA;AAAMY,KAEbG,qBAFaH,CAAAA,gBAEyBlB,gBAFzBkB,GAAAA,SAAAA,GAAAA,KAAAA,GAAAA,GAAAA,CAAAA,GAAAA,CAEwEI,OAFxEJ,CAAAA,SAAAA,CAAAA,KAAAA,CAAAA,GAEmGH,iBAFnGG,GAEuHI,OAFvHJ,SAEuIlB,gBAFvIkB,GAE0Jd,qBAF1Jc,CAEgLI,OAFhLJ,CAAAA,GAE2LH,iBAF3LG,GAE+MI,OAF/MJ,SAE+NK,MAF/NL,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAEyPI,OAFzPJ,GAEmQH,iBAFnQG,GAEuRH,iBAFvRG;;;AAAW;AACxBE,KAKAI,gBALiB,CAAA,MAAA,CAAGlB,GAAAA,CAKQmB,MALRnB,GAAAA;EACpBe,MAAAA,CAAAA,EAKCR,YALDQ;CAAqB,CAAA,GAAA,IAAA;;;;;AAA+HrB,UAW/I0B,eAX+I1B,CAAAA,eAWhHuB,MAXgHvB,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAWtFuB,MAXsFvB,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,WAAAA,OAAAA,CAAAA,CAAAA;EAAgB;;;EAAqD,QAAGsB,EAe1Nb,UAf0Na;EAAO;;;;EAAkF,IAAA,EAoBvTX,UApBuT,GAoB1SC,UApB0S;EAIrTY;;;EAAkC,KACjCX,EAmBFY,MAnBEZ,GAmBOE,iBAnBPF;EAAY;AAMzB;;EAAgC,OAAgBU,EAiBnCT,OAjBmCS,CAiB3BI,QAjB2BJ,CAAAA;;;;;;AAa5BR,KAURa,eAVQb,CAAAA,gBAUwBQ,MAVxBR,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAUkDA,iBAVlDA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAUqGW,eAVrGX,CAUqHO,OAVrHP,EAU8HY,QAV9HZ,CAAAA,EAAAA,GAU4IE,cAV5IF,CAU2JP,WAV3JO,GAUyKL,OAVzKK,CAAAA;;;AAIA;AAMpB;AAA2B,KAKfc,gBALe,CAAA,gBAKkB7B,gBALlB,GAAA,SAAA,GAAA,SAAA,EAAA,WAAA,OAAA,CAAA,GAAA,CAAA,OAAA,EAK4F0B,eAL5F,CAK4GL,qBAL5G,CAKkIC,OALlI,CAAA,EAK4IK,QAL5I,CAAA,EAAA,OAAA,EAKgKC,eALhK,CAKgLP,qBALhL,CAKsMC,OALtM,CAAA,EAKgNK,QALhN,CAAA,EAAA,GAK8NV,cAL9N,CAK6OT,WAL7O,GAK2PE,OAL3P,CAAA;;;;;;;;AAAqIO,KAapJa,oBAboJb,CAAAA,gBAa/GjB,gBAb+GiB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAarCD,YAbqCC,CAaxBI,qBAbwBJ,CAaFK,OAbEL,CAAAA,EAaQU,QAbRV,CAAAA,EAAAA,GAasBA,cAbtBA,CAaqCV,SAbrCU,CAAAA;AAAc;AAK9K;;;;;;;;;;;;AAAsRP,KAsB1QqB,iBAtB0QrB,CAAAA,gBAsBxOV,gBAtBwOU,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAsB9JM,YAtB8JN,CAsBjJW,qBAtBiJX,CAsB3HY,OAtB2HZ,CAAAA,EAsBjHiB,QAtBiHjB,CAAAA,EAAAA,OAAAA,EAsB7FoB,oBAtB6FpB,CAsBxEY,OAtBwEZ,EAsB/DiB,QAtB+DjB,CAAAA,EAAAA,GAsBjDO,cAtBiDP,CAsBlCH,SAtBkCG,CAAAA;;AAAf;AAQvQ;;;;;;KAuBKsB,kBAvBsHhB,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAuBtEM,OAvBsEN,EAAAA,OAAAA,EAuBpDF,OAvBoDE,CAuB5CW,QAvB4CX,CAAAA,EAAAA,GAuB9BC,cAvB8BD,CAuBfQ,gBAvBeR,CAuBEiB,OAvBFjB,CAuBUM,OAvBVN,CAAAA,CAAAA,CAAAA;;;AAAyE;AAcpM;;AAA8ChB,KAelCkC,eAfkClC,CAAAA,gBAeFA,gBAfEA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAe8DgC,kBAf9DhC,CAeiFqB,qBAfjFrB,CAeuGsB,OAfvGtB,CAAAA,EAeiH2B,QAfjH3B,CAAAA,GAAAA;EAAgB,IAA6FsB,EAgBjJU,kBAhBiJV,CAgB9HD,qBAhB8HC,CAgBxGA,OAhBwGA,CAAAA,EAgB9FK,QAhB8FL,CAAAA;EAAO,SAA7BD,CAAAA,EAiBrHR,YAjBqHQ,EAAAA;CAAqB;;;;;;;AAAyF;AAAY;KA2B1Pc,kBAlBkB,CAAA,OAAA,EAAA,QAAA,CAAA,GAAA,CAAA,KAAA,EAkB8Bb,OAlB9B,EAAA,OAAA,EAkBgDR,OAlBhD,CAkBwDa,QAlBxD,CAAA,EAAA,GAkBsEV,cAlBtE,CAkBqFO,gBAlBrF,CAkBsGS,OAlBtG,CAkB8GX,OAlB9G,CAAA,CAAA,CAAA;;;;;;AAAqFE,KAwBhGY,eAxBgGZ,CAAAA,gBAwBhExB,gBAxBgEwB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAwBAW,kBAxBAX,CAwBmBH,qBAxBnBG,CAwByCF,OAxBzCE,CAAAA,EAwBmDG,QAxBnDH,CAAAA,GAAAA;EAAgB,IAA/BP,EAyBnFkB,kBAzBmFlB,CAyBhEI,qBAzBgEJ,CAyB1CK,OAzB0CL,CAAAA,EAyBhCU,QAzBgCV,CAAAA;EAAc,SAAA,CAAA,EA0B3FJ,YA1B2F,EAAA;AAM3G,CAAA;;;;;;;;;;KA+BKwB,iBA9BKL,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EA8B0CV,OA9B1CU,EAAAA,OAAAA,EA8B4DlB,OA9B5DkB,CA8BoEL,QA9BpEK,CAAAA,EAAAA,GA8BkFf,cA9BlFe,CA8BiGR,gBA9BjGQ,CA8BkHC,OA9BlHD,CA8B0HV,OA9B1HU,CAAAA,CAAAA,CAAAA;;AACkB;AAC1B;;;AAS6EL,KAyBnEW,cAzBmEX,CAAAA,gBAyBpC3B,gBAzBoC2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAyB4BU,iBAzB5BV,CAyB8CN,qBAzB9CM,CAyBoEL,OAzBpEK,CAAAA,EAyB8EA,QAzB9EA,CAAAA,GAAAA;EAAQ,IAAhBb,EA0B7DuB,iBA1B6DvB,CA0B3CO,qBA1B2CP,CA0BrBQ,OA1BqBR,CAAAA,EA0BXa,QA1BWb,CAAAA;EAAO,SAAuDQ,CAAAA,EA2BrHT,YA3BqHS,EAAAA;CAAO;;;AAAjC;AAM3G;;;;;KA+BKiB,iBA/B0JZ,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EA+B3GL,OA/B2GK,EAAAA,OAAAA,EA+BzFb,OA/ByFa,CA+BjFA,QA/BiFA,CAAAA,EAAAA,GA+BnEV,cA/BmEU,CA+BpDH,gBA/BoDG,CA+BnCM,OA/BmCN,CA+B3BL,OA/B2BK,CAAAA,CAAAA,CAAAA;;;;;;AAE/Id,KAmCJ2B,cAnCI3B,CAAAA,gBAmC2Bb,gBAnC3Ba,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAmC2F0B,iBAnC3F1B,CAmC6GQ,qBAnC7GR,CAmCmIS,OAnCnIT,CAAAA,EAmC6Ic,QAnC7Id,CAAAA,GAAAA;EAAY,IAAA,EAoClB0B,iBApCkB,CAoCAlB,qBApCA,CAoCsBC,OApCtB,CAAA,EAoCgCK,QApChC,CAAA;EAWvBU,SAAAA,CAAAA,EA0BWxB,YA1BM,EAAA;CAAA;;;;AAA8GS,UA+BnHmB,eA/BmHnB,CAAAA,gBA+BnFtB,gBA/BmFsB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,uBA+BxBtB,gBA/BwBsB,GA+BLrB,iBA/BKqB,CA+BatB,gBA/BbsB,CAAAA,GA+BiCpB,kBA/BjCoB,CA+BoDtB,gBA/BpDsB,CAAAA,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,eAAAA,GAAAA,CAAAA,CAAAA;EAAO;;;EAAjC,IAAA,EAAA,MAAA;EAM9FgB;;;;;;;EAAgH,WAC1EhB,CAAAA,EAoChCA,OApCgCA;EAAO;;;;AAC7B;AAC1B;;EASoB,aAA8BA,CAAAA,EAiChCoB,cAjCgCpB;EAAO;;;EAAgF,KAAfW,CAAAA,EAAAA,CAqC/GtB,UArC+GsB,GAqClGrB,UArCkGqB,CAAAA,EAAAA;EAAO;;AAAzB;AAM1G;;;;;;;;;;;;AAE4B;AAK5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwJ+B;AAC9B;;;;;;;AAK4D;AAO7D;;;;;;;EAAuI,YAA4CgB,CAAAA,EA9EhKpB,gBA8EgKoB,CA9E/I3B,OA8E+I2B,EA9EtIN,YA8EsIM,CAAAA;EAAC;;AAAxB;AAK5J;;;;;;;;;;AAAiK;AAIjK;;;;;;;;;;;;;;EAA0S,aAA1BD,CAAAA,EA1D5PjB,iBA0D4PiB,CA1D1O1B,OA0D0O0B,EA1DjOL,YA0DiOK,CAAAA;EAAoB;AAIpS;;;;;;;EAA2L,WAAGK,CAAAA,EArD5KnB,eAqD4KmB,CArD5J/B,OAqD4J+B,EArDnJV,YAqDmJU,CAAAA;EAAI;;;;;;;AAAqI;EAgB3TM,WAAAA,CAAAA,EA5DMvB,eA4DNuB,CA5DsBrC,OA4DK,EA5DIqB,YA4DJ,CAAA;EAAA;;;;;;;;EAA6I,UAAsBe,CAAAA,EAnDzLpB,cAmDyLoB,CAnD1KpC,OAmD0KoC,EAnDjKf,YAmDiKe,CAAAA;EAAC;;;AAAgD;AAIkF;;;;EAIlR,UAAoCM,CAAAA,EAlD9ExB,cAkD8EwB,CAlD/D1C,OAkD+D0C,EAlDtDrB,YAkDsDqB,CAAAA;;;;;KA7C1FpB,kBA6CsKmB,CAAAA,CAAAA,CAAAA,GAAAA,QAAKC,MA5ChK9C,CA4CgK8C,IA5C3JnB,CA4C2JmB,SAAAA,IAAAA,MAAAA,EAAAA,GAAAA,KAAAA,GA5C1HnB,CA4C0HmB,GA5CtH9C,CA4CsH8C,CA5CpHnB,CA4CoHmB,CAAAA,EAAC;;;AAAyB;AAI1M;AAAwC,KAzC5BhB,oBAyC4B,CAAA,UAzCGP,eAyCH,CAAA,GAzCsBvB,CAyCtB,SAzCgCuB,eAyChC,CAAA,KAAA,EAAA,EAAA,GAAA,EAAA,GAAA,CAAA,GAzCqEQ,CAyCrE,SAzC+EjD,gBAyC/E,GAzCkG4C,kBAyClG,CAzCqHxC,qBAyCrH,CAzC2I6C,CAyC3I,CAAA,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;;;;;AAAsIR,KApClKS,yBAoCkKT,CAAAA,UApC9HA,eAoC8HA,CAAAA,GApC3GvB,CAoC2GuB,SApCjGA,eAoCiGA,CAAAA,KAAAA,EAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GApC5DQ,CAoC4DR,SApClDzC,gBAoCkDyC,GApC/BG,kBAoC+BH,CApCZtC,oBAoCYsC,CApCSQ,CAoCTR,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA8EkB,KAhChPR,qBAgCgPQ,CAAAA,IAhCtNlB,eAgCsNkB,EAAAA,CAAAA,GAhCjMzC,CAgCiMyC,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAhCpKzC,CAgCoKyC,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAhClHP,KAgCkHO,SAhCpGlB,eAgCoGkB,GAhClFN,IAgCkFM,SAAAA,SAhC5DlB,eAgC4DkB,EAAAA,GAhCxCX,oBAgCwCW,CAhCnBP,KAgCmBO,CAAAA,GAhCVR,qBAgCUQ,CAhCYN,IAgCZM,CAAAA,GAhCoBX,oBAgCpBW,CAhCyCP,KAgCzCO,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAAsGP,KA5BtVE,0BA4BsVF,CAAAA,UAAAA,SA5BxSX,eA4BwSW,EAAAA,CAAAA,GA5BnRlC,CA4BmRkC,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GA5BtPlC,CA4BsPkC,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GA5BpMA,KA4BoMA,SA5BtLX,eA4BsLW,GA5BpKC,IA4BoKD,SAAAA,SA5B9IX,eA4B8IW,EAAAA,GA5B1HF,yBA4B0HE,CA5BhGA,KA4BgGA,CAAAA,GA5BvFE,0BA4BuFF,CA5B5DC,IA4B4DD,CAAAA,GA5BpDF,yBA4BoDE,CA5B1BA,KA4B0BA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;AAAD;AAIjW;;;;AAAqO;AACzNL,KAjBAY,2BAiBgB,CAAA,UAjBsBlB,eAiBtB,CAAA,GAjByCvB,CAiBzC,SAjBmDuB,eAiBnD,CAAA,GAAA,EAAA,KAAA,EAAA,EAAA,GAAA,CAAA,GAjBwFiB,CAiBxF,SAjBkGxD,kBAiBlG,CAAA,KAAA,MAAA,CAAA,GAjBoIC,oBAiBpI,CAjByJyD,KAiBzJ,CAAA,GAAA,SAAA,GAjB8KF,CAiB9K,SAjBwL1D,gBAiBxL,GAjB2MG,oBAiB3M,CAjBgOuD,CAiBhO,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;;;;;;;;KATvBI,iBASsH9D,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CAT3F+D,CAS2F/D,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,CATjEgE,CASiEhE,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,SAAAA,GAT5BgE,CAS4BhE,GAAAA,SAAAA,GAAAA,CATXgE,CASWhE,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GATc+D,CASd/D,GAAAA,SAAAA,GAAAA,CAT+B+D,CAS/B/D,CAAAA,SAAAA,CAT2CgE,CAS3ChE,CAAAA,GATgD+D,CAShD/D,GAAAA,CATqDgE,CASrDhE,CAAAA,SAAAA,CATiE+D,CASjE/D,CAAAA,GATsEgE,CAStEhE,GAT0E+D,CAS1E/D,GAT8EgE,CAS9EhE;;;;AAAiC,KALhJiE,4BAKgJ,CAAA,UAAA,SALhGxB,eAKgG,EAAA,CAAA,GAL3EvB,CAK2E,SAAA,SAAA,EAAA,GAAA,CAAA,CAAA,GAL9CA,CAK8C,SAAA,SAAA,CAAA,KAAA,MAAA,EAAA,GAAA,KAAA,KAAA,CAAA,GALIkC,KAKJ,SALkBX,eAKlB,GALoCY,IAKpC,SAAA,SAL0DZ,eAK1D,EAAA,GAL8EqB,iBAK9E,CALgGH,2BAKhG,CAL4HP,KAK5H,CAAA,EALoIa,4BAKpI,CALiKZ,IAKjK,CAAA,CAAA,GAL0KM,2BAK1K,CALsMP,KAKtM,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;AAC5J;;;AAA2DpD,KAF/CkE,iBAE+ClE,CAAAA,sBAFPoB,iBAEOpB,GAFaA,gBAEbA,CAAAA,GAFiCmE,aAEjCnE,SAFuDA,gBAEvDA,GAF0EG,oBAE1EH,CAF+FmE,aAE/FnE,CAAAA,GAFgHmE,aAEhHnE,SAFsIoB,iBAEtIpB,GAF0J+C,gBAE1J/C,CAF2KmE,aAE3KnE,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAgC+D,KAD/EhB,gBAC+EgB,CAAAA,UADpD3C,iBACoD2C,GADhC/D,gBACgC+D,CAAAA,GADZA,CACYA,SADF3C,iBACE2C,GADkBA,CAClBA,GADsBA,CACtBA,SADgC/D,gBAChC+D,GADmDzD,cACnDyD,CADkE1D,2BAClE0D,CAD8FA,CAC9FA,CAAAA,CAAAA,GAAAA,KAAAA;AAAU3C,KAAzFgD,gBAAyFhD,CAAAA,UAA9DA,iBAA8DA,GAA1CpB,gBAA0CoB,GAAAA,SAAAA,CAAAA,GAAV2C,CAAU3C,SAAAA,iBAAAA,GAAoBpB,gBAApBoB,GAAuC2B,gBAAvC3B,CAAwD2C,CAAxD3C,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA"}