langchain 1.2.24 → 1.2.25

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

@@ -1 +1 @@
-{"version":3,"file":"toolRetry.js","names":["config: ToolRetryMiddlewareConfig","toolFilter: string[]","toolName: string","error: 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, calculateRetryDelay } from \"./utils.js\";\nimport { RetrySchema } from \"./constants.js\";\nimport { InvalidRetryConfigError } from \"./error.js\";\n\n/**\n * Configuration options for the Tool Retry Middleware.\n */\nexport const ToolRetryMiddlewareOptionsSchema = z\n  .object({\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     * Behavior when all retries are exhausted. Options:\n     * - `\"continue\"` (default): Return an AIMessage with error details, allowing\n     *   the agent to potentially handle the failure gracefully.\n     * - `\"error\"`: Re-raise the exception, stopping agent execution.\n     * - Custom function: Function that takes the exception and returns a string\n     *   for the AIMessage content, allowing custom error formatting.\n     *\n     * Deprecated values:\n     * - `\"raise\"`: use `\"error\"` instead.\n     * - `\"return_message\"`: use `\"continue\"` instead.\n     */\n    onFailure: z\n      .union([\n        z.literal(\"error\"),\n        z.literal(\"continue\"),\n        /**\n         * @deprecated Use `\"error\"` instead.\n         */\n        z.literal(\"raise\"),\n        /**\n         * @deprecated Use `\"continue\"` instead.\n         */\n        z.literal(\"return_message\"),\n        z.function().args(z.instanceof(Error)).returns(z.string()),\n      ])\n      .default(\"continue\"),\n  })\n  .merge(RetrySchema);\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 { success, error, data } =\n    ToolRetryMiddlewareOptionsSchema.safeParse(config);\n  if (!success) {\n    throw new InvalidRetryConfigError(error);\n  }\n  const {\n    maxRetries,\n    tools,\n    retryOn,\n    onFailure: onFailureConfig,\n    backoffFactor,\n    initialDelayMs,\n    maxDelayMs,\n    jitter,\n  } = data;\n\n  let onFailure = onFailureConfig;\n  if (onFailureConfig === \"raise\") {\n    console.warn(\n      \"⚠️ `onFailure: 'raise'` is deprecated. Use `onFailure: 'error'` instead.\"\n    );\n    onFailure = \"error\";\n  } else if (onFailureConfig === \"return_message\") {\n    console.warn(\n      \"⚠️ `onFailure: 'return_message'` is deprecated. Use `onFailure: 'continue'` instead.\"\n    );\n    onFailure = \"continue\";\n  }\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((ErrorConstructor) => {\n      // eslint-disable-next-line no-instanceof/no-instanceof\n      return error instanceof 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 === \"error\") {\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 ?? request.toolCall.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":";;;;;;;;;;;AAgBA,MAAa,mCAAmC,EAC7C,OAAO;CAMN,OAAO,EACJ,MACC,EAAE,MAAM;EAAC,EAAE,QAAoB;EAAE,EAAE,QAAoB;EAAE,EAAE,QAAQ;CAAC,EAAC,CACtE,CACA,UAAU;CAcb,WAAW,EACR,MAAM;EACL,EAAE,QAAQ,QAAQ;EAClB,EAAE,QAAQ,WAAW;EAIrB,EAAE,QAAQ,QAAQ;EAIlB,EAAE,QAAQ,iBAAiB;EAC3B,EAAE,UAAU,CAAC,KAAK,EAAE,WAAW,MAAM,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC;CAC3D,EAAC,CACD,QAAQ,WAAW;AACvB,EAAC,CACD,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyGrB,SAAgB,oBACdA,SAAoC,CAAE,GACrB;CACjB,MAAM,EAAE,SAAS,OAAO,MAAM,GAC5B,iCAAiC,UAAU,OAAO;AACpD,KAAI,CAAC,QACH,OAAM,IAAI,wBAAwB;CAEpC,MAAM,EACJ,YACA,OACA,SACA,WAAW,iBACX,eACA,gBACA,YACA,QACD,GAAG;CAEJ,IAAI,YAAY;AAChB,KAAI,oBAAoB,SAAS;EAC/B,QAAQ,KACN,2EACD;EACD,YAAY;CACb,WAAU,oBAAoB,kBAAkB;EAC/C,QAAQ,KACN,uFACD;EACD,YAAY;CACb;CAGD,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,YAA0B;AACtD,MAAI,OAAO,YAAY,WACrB,QAAO,QAAQC,QAAM;AAGvB,SAAO,QAAQ,KAAK,CAAC,qBAAqB;AAExC,UAAOA,mBAAiB;EACzB,EAAC;CACH;CAGD,MAAM,cAAc;EAAE;EAAe;EAAgB;EAAY;CAAQ;;;;CAKzE,MAAM,uBAAuB,CAC3BF,UACAC,SACAE,iBACW;EACX,MAAM,YAAYD,QAAM,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,UACAI,YACAH,SACAE,iBACgB;AAChB,MAAI,cAAc,QAChB,OAAMD;EAGR,IAAIG;AACJ,MAAI,OAAO,cAAc,YACvB,UAAU,UAAUH,QAAM;OAE1B,UAAU,qBAAqB,UAAUA,SAAO,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,WAAY,QAAQ,MAAM,QAAQ,QAAQ,SAAS;AAGzD,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,SAAQA,SAAO;IACd,MAAM,eAAe,UAAU;IAG/B,MAAM,MACJA,WAAS,OAAOA,YAAU,YAAY,aAAaA,UAC9CA,UACD,IAAI,MAAM,OAAOA,QAAM;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"}
+{"version":3,"file":"toolRetry.js","names":[],"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, calculateRetryDelay } from \"./utils.js\";\nimport { RetrySchema } from \"./constants.js\";\nimport { InvalidRetryConfigError } from \"./error.js\";\n\n/**\n * Configuration options for the Tool Retry Middleware.\n */\nexport const ToolRetryMiddlewareOptionsSchema = z\n  .object({\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     * Behavior when all retries are exhausted. Options:\n     * - `\"continue\"` (default): Return an AIMessage with error details, allowing\n     *   the agent to potentially handle the failure gracefully.\n     * - `\"error\"`: Re-raise the exception, stopping agent execution.\n     * - Custom function: Function that takes the exception and returns a string\n     *   for the AIMessage content, allowing custom error formatting.\n     *\n     * Deprecated values:\n     * - `\"raise\"`: use `\"error\"` instead.\n     * - `\"return_message\"`: use `\"continue\"` instead.\n     */\n    onFailure: z\n      .union([\n        z.literal(\"error\"),\n        z.literal(\"continue\"),\n        /**\n         * @deprecated Use `\"error\"` instead.\n         */\n        z.literal(\"raise\"),\n        /**\n         * @deprecated Use `\"continue\"` instead.\n         */\n        z.literal(\"return_message\"),\n        z.function().args(z.instanceof(Error)).returns(z.string()),\n      ])\n      .default(\"continue\"),\n  })\n  .merge(RetrySchema);\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 { success, error, data } =\n    ToolRetryMiddlewareOptionsSchema.safeParse(config);\n  if (!success) {\n    throw new InvalidRetryConfigError(error);\n  }\n  const {\n    maxRetries,\n    tools,\n    retryOn,\n    onFailure: onFailureConfig,\n    backoffFactor,\n    initialDelayMs,\n    maxDelayMs,\n    jitter,\n  } = data;\n\n  let onFailure = onFailureConfig;\n  if (onFailureConfig === \"raise\") {\n    console.warn(\n      \"⚠️ `onFailure: 'raise'` is deprecated. Use `onFailure: 'error'` instead.\"\n    );\n    onFailure = \"error\";\n  } else if (onFailureConfig === \"return_message\") {\n    console.warn(\n      \"⚠️ `onFailure: 'return_message'` is deprecated. Use `onFailure: 'continue'` instead.\"\n    );\n    onFailure = \"continue\";\n  }\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((ErrorConstructor) => {\n      // eslint-disable-next-line no-instanceof/no-instanceof\n      return error instanceof 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 === \"error\") {\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 ?? request.toolCall.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":";;;;;;;;;;;;;;AAgBA,MAAa,mCAAmC,EAC7C,OAAO;CAMN,OAAO,EACJ,MACC,EAAE,MAAM;EAAC,EAAE,QAAoB;EAAE,EAAE,QAAoB;EAAE,EAAE,QAAQ;EAAC,CAAC,CACtE,CACA,UAAU;CAcb,WAAW,EACR,MAAM;EACL,EAAE,QAAQ,QAAQ;EAClB,EAAE,QAAQ,WAAW;EAIrB,EAAE,QAAQ,QAAQ;EAIlB,EAAE,QAAQ,iBAAiB;EAC3B,EAAE,UAAU,CAAC,KAAK,EAAE,WAAW,MAAM,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC;EAC3D,CAAC,CACD,QAAQ,WAAW;CACvB,CAAC,CACD,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyGrB,SAAgB,oBACd,SAAoC,EAAE,EACrB;CACjB,MAAM,EAAE,SAAS,OAAO,SACtB,iCAAiC,UAAU,OAAO;AACpD,KAAI,CAAC,QACH,OAAM,IAAI,wBAAwB,MAAM;CAE1C,MAAM,EACJ,YACA,OACA,SACA,WAAW,iBACX,eACA,gBACA,YACA,WACE;CAEJ,IAAI,YAAY;AAChB,KAAI,oBAAoB,SAAS;AAC/B,UAAQ,KACN,2EACD;AACD,cAAY;YACH,oBAAoB,kBAAkB;AAC/C,UAAQ,KACN,uFACD;AACD,cAAY;;CAId,MAAM,aAAuB,EAAE;AAC/B,MAAK,MAAM,QAAQ,SAAS,EAAE,CAC5B,KAAI,OAAO,SAAS,SAClB,YAAW,KAAK,KAAK;UACZ,UAAU,QAAQ,OAAO,KAAK,SAAS,SAChD,YAAW,KAAK,KAAK,KAAK;KAE1B,OAAM,IAAI,UACR,mFACD;;;;CAOL,MAAM,mBAAmB,aAA8B;AACrD,MAAI,WAAW,WAAW,EACxB,QAAO;AAET,SAAO,WAAW,SAAS,SAAS;;;;;CAMtC,MAAM,wBAAwB,UAA0B;AACtD,MAAI,OAAO,YAAY,WACrB,QAAO,QAAQ,MAAM;AAGvB,SAAO,QAAQ,MAAM,qBAAqB;AAExC,UAAO,iBAAiB;IACxB;;CAIJ,MAAM,cAAc;EAAE;EAAe;EAAgB;EAAY;EAAQ;;;;CAKzE,MAAM,wBACJ,UACA,OACA,iBACW;EACX,MAAM,YAAY,MAAM,YAAY;AAEpC,SAAO,SAAS,SAAS,iBAAiB,aAAa,GADnC,iBAAiB,IAAI,YAAY,WACiB,QAAQ;;;;;CAMhF,MAAM,iBACJ,UACA,YACA,OACA,iBACgB;AAChB,MAAI,cAAc,QAChB,OAAM;EAGR,IAAI;AACJ,MAAI,OAAO,cAAc,WACvB,WAAU,UAAU,MAAM;MAE1B,WAAU,qBAAqB,UAAU,OAAO,aAAa;AAG/D,SAAO,IAAI,YAAY;GACrB;GACA,cAAc;GACd,MAAM;GACN,QAAQ;GACT,CAAC;;AAGJ,QAAO,iBAAiB;EACtB,MAAM;EACN,eAAe;EACf,cAAc,OAAO,SAAS,YAAY;GACxC,MAAM,WAAY,QAAQ,MAAM,QAAQ,QAAQ,SAAS;AAGzD,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;YACtB,OAAO;IACd,MAAM,eAAe,UAAU;IAG/B,MAAM,MACJ,SAAS,OAAO,UAAU,YAAY,aAAa,QAC9C,QACD,IAAI,MAAM,OAAO,MAAM,CAAC;AAG9B,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,EACV,OAAM,MAAM,MAAM;UAKpB,QAAO,cAAc,UAAU,YAAY,KAAK,aAAa;;AAMnE,SAAM,IAAI,MAAM,qDAAqD;;EAExE,CAAC"}

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

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

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

@@ -1,5 +1,5 @@
 import { JumpToTarget } from "../constants.cjs";
-import { AgentBuiltInState, Runtime } from "../runtime.cjs";
+import { AgentBuiltInState, Runtime as Runtime$1 } from "../runtime.cjs";
 import { ModelRequest } from "../nodes/types.cjs";
 import { AIMessage, SystemMessage, ToolMessage } from "@langchain/core/messages";
 import { ClientTool, ServerTool } from "@langchain/core/tools";
@@ -39,11 +39,11 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  * >;
  * ```
  */
-interface MiddlewareTypeConfig<TSchema$1 extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema$1 extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
   /** The middleware state schema type */
-  Schema: TSchema$1;
+  Schema: TSchema;
   /** The middleware context schema type */
-  ContextSchema: TContextSchema$1;
+  ContextSchema: TContextSchema;
   /** The full context type */
   FullContext: TFullContext;
   /** The tools array type */
@@ -54,7 +54,7 @@ interface MiddlewareTypeConfig<TSchema$1 extends StateDefinitionInit | undefined
  * Used when no explicit type parameters are provided.
  */
 type DefaultMiddlewareTypeConfig = MiddlewareTypeConfig;
-type NormalizedSchemaInput<TSchema$1 extends StateDefinitionInit | undefined | never = any> = [TSchema$1] extends [never] ? AgentBuiltInState : TSchema$1 extends InteropZodObject ? InferInteropZodOutput<TSchema$1> & AgentBuiltInState : TSchema$1 extends StateDefinitionInit ? InferSchemaInput<TSchema$1> & AgentBuiltInState : AgentBuiltInState;
+type NormalizedSchemaInput<TSchema extends StateDefinitionInit | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends StateDefinitionInit ? InferSchemaInput<TSchema> & AgentBuiltInState : AgentBuiltInState;
 /**
  * Result type for middleware functions.
  */
@@ -98,18 +98,18 @@ interface ToolCallRequest<TState extends Record<string, unknown> = Record<string
   /**
    * The runtime context containing metadata, signal, writer, interrupt, etc.
    */
-  runtime: Runtime<TContext>;
+  runtime: Runtime$1<TContext>;
 }
 /**
  * Handler function type for wrapping tool calls.
  * Takes a tool call request and returns the tool result or a command.
  */
-type ToolCallHandler<TSchema$1 extends Record<string, unknown> = AgentBuiltInState, TContext = unknown> = (request: ToolCallRequest<TSchema$1, 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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema$1>, TContext>, handler: ToolCallHandler<NormalizedSchemaInput<TSchema$1>, TContext>) => PromiseOrValue<ToolMessage | Command>;
+type WrapToolCallHook<TSchema extends StateDefinitionInit | 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.
@@ -117,7 +117,7 @@ type WrapToolCallHook<TSchema$1 extends StateDefinitionInit | undefined = undefi
  * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime
  * @returns The AI message response from the model
  */
-type WrapModelCallHandler<TSchema$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: Omit<ModelRequest<NormalizedSchemaInput<TSchema$1>, TContext>,
+type WrapModelCallHandler<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: Omit<ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,
 /**
  * allow to reset the system prompt or system message
  */
@@ -138,7 +138,7 @@ type WrapModelCallHandler<TSchema$1 extends StateDefinitionInit | undefined = un
  * @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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema$1>, TContext>, handler: WrapModelCallHandler<TSchema$1, TContext>) => PromiseOrValue<AIMessage | Command>;
+type WrapModelCallHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage | Command>;
 /**
  * Handler function type for the beforeAgent hook.
  * Called once at the start of agent invocation before any model calls or tool executions.
@@ -147,14 +147,14 @@ type WrapModelCallHook<TSchema$1 extends StateDefinitionInit | undefined = undef
  * @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$1, TContext> = (state: TSchema$1, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema$1>>>;
+type BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime$1<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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema$1>, TContext> | {
-  hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema$1>, TContext>;
+type BeforeAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -165,14 +165,14 @@ type BeforeAgentHook<TSchema$1 extends StateDefinitionInit | undefined = undefin
  * @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$1, TContext> = (state: TSchema$1, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema$1>>>;
+type BeforeModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime$1<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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema$1>, TContext> | {
-  hook: BeforeModelHandler<NormalizedSchemaInput<TSchema$1>, TContext>;
+type BeforeModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -184,14 +184,14 @@ type BeforeModelHook<TSchema$1 extends StateDefinitionInit | undefined = undefin
  * @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$1, TContext> = (state: TSchema$1, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema$1>>>;
+type AfterModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime$1<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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema$1>, TContext> | {
-  hook: AfterModelHandler<NormalizedSchemaInput<TSchema$1>, TContext>;
+type AfterModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -202,14 +202,14 @@ type AfterModelHook<TSchema$1 extends StateDefinitionInit | undefined = undefine
  * @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$1, TContext> = (state: TSchema$1, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema$1>>>;
+type AfterAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime$1<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$1 extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema$1>, TContext> | {
-  hook: AfterAgentHandler<NormalizedSchemaInput<TSchema$1>, TContext>;
+type AfterAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
+  hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -235,7 +235,7 @@ declare const MIDDLEWARE_BRAND: unique symbol;
  * });
  * ```
  */
-interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = any, TContextSchema$1 extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
+interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
   /**
    * Brand property to distinguish middleware instances from plain objects or functions.
    * This is required and prevents accidental assignment of functions to middleware arrays.
@@ -246,7 +246,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * This is a phantom property used only for type inference.
    * @internal
    */
-  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema$1, TContextSchema$1, TFullContext, TTools>;
+  readonly "~middlewareTypes"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools>;
   /**
    * The name of the middleware.
    */
@@ -258,7 +258,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * - An AnnotationRoot
    * - Undefined
    */
-  stateSchema?: TSchema$1;
+  stateSchema?: TSchema;
   /**
    * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:
    * - A Zod object
@@ -266,7 +266,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * - A Zod default object
    * - Undefined
    */
-  contextSchema?: TContextSchema$1;
+  contextSchema?: TContextSchema;
   /**
    * Additional tools registered by the middleware.
    */
@@ -333,7 +333,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * }
    * ```
    */
-  wrapToolCall?: WrapToolCallHook<TSchema$1, TFullContext>;
+  wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;
   /**
    * Wraps the model invocation with custom logic. This allows you to:
    * - Modify the request before calling the model
@@ -362,7 +362,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * }
    * ```
    */
-  wrapModelCall?: WrapModelCallHook<TSchema$1, TFullContext>;
+  wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;
   /**
    * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.
    * It allows to modify the state of the agent before any model calls or tool executions.
@@ -371,7 +371,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * @param runtime - The middleware runtime
    * @returns The modified middleware state or undefined to pass through
    */
-  beforeAgent?: BeforeAgentHook<TSchema$1, TFullContext>;
+  beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;
   /**
    * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.
    * It allows to modify the state of the agent.
@@ -380,7 +380,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * @param runtime - The middleware runtime
    * @returns The modified middleware state or undefined to pass through
    */
-  beforeModel?: BeforeModelHook<TSchema$1, TFullContext>;
+  beforeModel?: BeforeModelHook<TSchema, TFullContext>;
   /**
    * The function to run after the model call. This function is called after the model is invoked and before any tools are called.
    * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.
@@ -389,7 +389,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * @param runtime - The middleware runtime
    * @returns The modified middleware state or undefined to pass through
    */
-  afterModel?: AfterModelHook<TSchema$1, TFullContext>;
+  afterModel?: AfterModelHook<TSchema, TFullContext>;
   /**
    * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.
    * It allows to modify the final state of the agent after all model calls and tool executions are complete.
@@ -398,7 +398,7 @@ interface AgentMiddleware<TSchema$1 extends StateDefinitionInit | undefined = an
    * @param runtime - The middleware runtime
    * @returns The modified middleware state or undefined to pass through
    */
-  afterAgent?: AfterAgentHook<TSchema$1, TFullContext>;
+  afterAgent?: AfterAgentHook<TSchema, TFullContext>;
 }
 /**
  * Helper type to filter out properties that start with underscore (private properties)
@@ -418,7 +418,7 @@ type ResolveMiddlewareTypeConfig<T> = T extends {
  * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from
  * @typeParam K - The property key to extract ("Schema" | "ContextSchema" | "FullContext" | "Tools")
  */
-type InferMiddlewareType<T, K$1 extends keyof MiddlewareTypeConfig> = ResolveMiddlewareTypeConfig<T>[K$1];
+type InferMiddlewareType<T, K extends keyof MiddlewareTypeConfig> = ResolveMiddlewareTypeConfig<T>[K];
 /**
  * Shorthand helper to extract the Schema type from a MiddlewareTypeConfig or AgentMiddleware.
  */

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","InteropZodToStateDefinition","AnnotationRoot","StateSchema","InferStateSchemaUpdate","StateDefinitionInit","AIMessage","SystemMessage","ToolMessage","ToolCall","Command","ClientTool","ServerTool","JumpToTarget","Runtime","AgentBuiltInState","ModelRequest","PromiseOrValue","T","Promise","AnyAnnotationRoot","MiddlewareTypeConfig","TSchema","TContextSchema","TFullContext","TTools","DefaultMiddlewareTypeConfig","NormalizedSchemaInput","InferSchemaInput","MiddlewareResult","TState","ToolCallRequest","TContext","Record","ToolCallHandler","WrapToolCallHook","WrapModelCallHandler","Omit","WrapModelCallHook","BeforeAgentHandler","Partial","BeforeAgentHook","BeforeModelHandler","BeforeModelHook","AfterModelHandler","AfterModelHook","AfterAgentHandler","AfterAgentHook","MIDDLEWARE_BRAND","AgentMiddleware","FilterPrivateProps","K","ResolveMiddlewareTypeConfig","Types","InferMiddlewareType","InferMiddlewareSchema","InferMiddlewareContextSchema","InferMiddlewareFullContext","InferMiddlewareToolsFromConfig","InferChannelType","ToAnnotationRoot","InferMiddlewareState","InferMiddlewareInputState","InferMiddlewareStates","First","Rest","InferMiddlewareInputStates","InferMergedState","InferMergedInputState","InferMiddlewareContext","InferMiddlewareContextInput","Inner","InferMiddlewareContexts","MergeContextTypes","A","B","InferMiddlewareContextInputs","InferContextInput","ContextSchema","TFields"],"sources":["../../../src/agents/middleware/types.d.ts"],"sourcesContent":["import type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot, StateSchema, InferStateSchemaUpdate, StateDefinitionInit } from \"@langchain/langgraph\";\nimport type { AIMessage, SystemMessage, 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>;\n/**\n * Type bag that encapsulates all middleware type parameters.\n *\n * This interface bundles all the generic type parameters used throughout the middleware system\n * into a single configuration object. This pattern simplifies type signatures and makes\n * it easier to add new type parameters without changing multiple function signatures.\n *\n * @typeParam TSchema - The middleware state schema type. Can be a `StateDefinitionInit`\n *   (including `InteropZodObject`, `StateSchema`, or `AnnotationRoot`) or `undefined`.\n *\n * @typeParam TContextSchema - The middleware context schema type. Can be an `InteropZodObject`,\n *   `InteropZodDefault`, `InteropZodOptional`, or `undefined`.\n *\n * @typeParam TFullContext - The full context type available to middleware hooks.\n *\n * @typeParam TTools - The tools array type registered by the middleware.\n *\n * @example\n * ```typescript\n * // Define a type configuration\n * type MyMiddlewareTypes = MiddlewareTypeConfig<\n *   typeof myStateSchema,\n *   typeof myContextSchema,\n *   MyContextType,\n *   typeof myTools\n * >;\n * ```\n */\nexport interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined = StateDefinitionInit | undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {\n    /** The middleware state schema type */\n    Schema: TSchema;\n    /** The middleware context schema type */\n    ContextSchema: TContextSchema;\n    /** The full context type */\n    FullContext: TFullContext;\n    /** The tools array type */\n    Tools: TTools;\n}\n/**\n * Default type configuration for middleware.\n * Used when no explicit type parameters are provided.\n */\nexport type DefaultMiddlewareTypeConfig = MiddlewareTypeConfig;\nexport type NormalizedSchemaInput<TSchema extends StateDefinitionInit | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends StateDefinitionInit ? InferSchemaInput<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     * This will be `undefined` for dynamically registered tools that aren't\n     * declared upfront when creating the agent. In such cases, middleware\n     * should provide the tool implementation by spreading the request with\n     * the tool property.\n     *\n     * @example Dynamic tool handling\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   if (request.toolCall.name === \"dynamic_tool\" && !request.tool) {\n     *     // Provide the tool implementation for dynamically registered tools\n     *     return handler({ ...request, tool: myDynamicTool });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     */\n    tool: ClientTool | ServerTool | undefined;\n    /**\n     * The current agent state (includes both middleware state and built-in state).\n     */\n    state: TState & AgentBuiltInState;\n    /**\n     * The runtime context containing metadata, signal, writer, interrupt, etc.\n     */\n    runtime: Runtime<TContext>;\n}\n/**\n * 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 StateDefinitionInit | 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 StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: Omit<ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, \n/**\n * allow to reset the system prompt or system message\n */\n\"systemPrompt\" | \"systemMessage\"> & {\n    systemPrompt?: string;\n    systemMessage?: SystemMessage;\n}) => 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 StateDefinitionInit | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage | Command>;\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 StateDefinitionInit | 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 StateDefinitionInit | 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 StateDefinitionInit | 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 StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Unique symbol used to brand middleware instances.\n * This prevents functions from being accidentally assignable to AgentMiddleware\n * since functions have a 'name' property that would otherwise make them structurally compatible.\n */\nexport declare const MIDDLEWARE_BRAND: unique symbol;\n/**\n * Base middleware interface.\n *\n * @typeParam TSchema - The middleware state schema type\n * @typeParam TContextSchema - The middleware context schema type\n * @typeParam TFullContext - The full context type available to hooks\n * @typeParam TTools - The tools array type registered by the middleware\n *\n * @example\n * ```typescript\n * const middleware = createMiddleware({\n *   name: \"myMiddleware\",\n *   stateSchema: z.object({ count: z.number() }),\n *   tools: [myTool],\n * });\n * ```\n */\nexport interface AgentMiddleware<TSchema extends StateDefinitionInit | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any, TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {\n    /**\n     * Brand property to distinguish middleware instances from plain objects or functions.\n     * This is required and prevents accidental assignment of functions to middleware arrays.\n     */\n    readonly [MIDDLEWARE_BRAND]: true;\n    /**\n     * Type marker for extracting the MiddlewareTypeConfig from a middleware instance.\n     * This is a phantom property used only for type inference.\n     * @internal\n     */\n    readonly \"~middlewareTypes\"?: MiddlewareTypeConfig<TSchema, TContextSchema, TFullContext, TTools>;\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 (InteropZodObject)\n     * - A StateSchema from LangGraph (supports ReducedValue, UntrackedValue)\n     * - An AnnotationRoot\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?: TTools;\n    /**\n     * Wraps tool execution with custom logic. This allows you to:\n     * - Modify tool call parameters before execution\n     * - Handle errors and retry with different parameters\n     * - Post-process tool results\n     * - Implement caching, logging, authentication, or other cross-cutting concerns\n     * - Return Command objects for advanced control flow\n     *\n     * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n     * along with a handler function to execute the actual tool.\n     *\n     * @param request - The tool call request containing toolCall, state, and runtime.\n     * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n     * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n     *\n     * @example\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   console.log(`Calling tool: ${request.tool.name}`);\n     *   console.log(`Tool description: ${request.tool.description}`);\n     *\n     *   try {\n     *     // Execute the tool\n     *     const result = await handler(request);\n     *     console.log(`Tool ${request.tool.name} succeeded`);\n     *     return result;\n     *   } catch (error) {\n     *     console.error(`Tool ${request.tool.name} failed:`, error);\n     *     // Could return a custom error message or retry\n     *     throw error;\n     *   }\n     * }\n     * ```\n     *\n     * @example Authentication\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   // Check if user is authorized for this tool\n     *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n     *     return new ToolMessage({\n     *       content: \"Unauthorized to call this tool\",\n     *       tool_call_id: request.toolCall.id,\n     *     });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     *\n     * @example Caching\n     * ```ts\n     * const cache = new Map();\n     * wrapToolCall: async (request, handler) => {\n     *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n     *   if (cache.has(cacheKey)) {\n     *     return cache.get(cacheKey);\n     *   }\n     *   const result = await handler(request);\n     *   cache.set(cacheKey, result);\n     *   return result;\n     * }\n     * ```\n     */\n    wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n    /**\n     * 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};\n/**\n * Helper type to resolve a MiddlewareTypeConfig from either:\n * - A MiddlewareTypeConfig directly\n * - An AgentMiddleware instance (using `typeof middleware`)\n */\nexport type ResolveMiddlewareTypeConfig<T> = T extends {\n    \"~middlewareTypes\"?: infer Types;\n} ? Types extends MiddlewareTypeConfig ? Types : never : T extends MiddlewareTypeConfig ? T : never;\n/**\n * Helper type to extract any property from a MiddlewareTypeConfig or AgentMiddleware.\n *\n * @typeParam T - The MiddlewareTypeConfig or AgentMiddleware to extract from\n * @typeParam K - The property key to extract (\"Schema\" | \"ContextSchema\" | \"FullContext\" | \"Tools\")\n */\nexport type InferMiddlewareType<T, K extends keyof MiddlewareTypeConfig> = ResolveMiddlewareTypeConfig<T>[K];\n/**\n * Shorthand helper to extract the Schema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareSchema<T> = InferMiddlewareType<T, \"Schema\">;\n/**\n * Shorthand helper to extract the ContextSchema type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareContextSchema<T> = InferMiddlewareType<T, \"ContextSchema\">;\n/**\n * Shorthand helper to extract the FullContext type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareFullContext<T> = InferMiddlewareType<T, \"FullContext\">;\n/**\n * Shorthand helper to extract the Tools type from a MiddlewareTypeConfig or AgentMiddleware.\n */\nexport type InferMiddlewareToolsFromConfig<T> = InferMiddlewareType<T, \"Tools\">;\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 * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T extends readonly AgentMiddleware[]> = 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 TContextSchema, any, any> ? TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};\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 TContextSchema, any, any> ? TContextSchema extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : TContextSchema extends InteropZodObject ? InferInteropZodInput<TContextSchema> : {} : {};\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 StateDefinitionInit> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? InteropZodToStateDefinition<A> : never;\nexport type InferSchemaInput<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A[\"State\"] : {};\nexport {};\n//# sourceMappingURL=types.d.ts.map"],"mappings":";;;;;;;;;;;KAUKqB,oBAAoBC,IAAIC,QAAQD;AAAhCD,KACOG,iBAAAA,GAAoBlB,cADb,CAAA,GAAA,CAAA;;;;AAAiB;AACpC;AA6BA;;;;;;;;;;;;;;;;;;;;;AAQiB;AAMjB;AACYyB,UAfKN,oBAegBC,CAAAA,kBAfqBjB,mBAerB,GAAA,SAAA,GAfuDA,mBAevD,GAAA,SAAA,EAAA,yBAf+GT,gBAe/G,GAfkIC,iBAelI,CAfoJD,gBAepJ,CAAA,GAfwKE,kBAexK,CAf2LF,gBAe3L,CAAA,GAAA,SAAA,GAf2NA,gBAe3N,GAf8OC,iBAe9O,CAfgQD,gBAehQ,CAAA,GAfoRE,kBAepR,CAfuSF,gBAevS,CAAA,GAAA,SAAA,EAAA,eAAA,GAAA,EAAA,eAAA,SAAA,CAfmXe,UAenX,GAfgYC,UAehY,CAAA,EAAA,GAAA,SAAA,CAf0ZD,UAe1Z,GAfuaC,UAeva,CAAA,EAAA,CAAA,CAAA;EAAiBP;EAAkDiB,MAAAA,EAbxFA,SAawFA;EAA2BP;EAAoBO,aAAAA,EAXhIC,gBAWgID;EAAgB1B;EAAyC0B,WAAAA,EAT3LE,YAS2LF;EAAtBtB;EAAiCe,KAAAA,EAP5MU,MAO4MV;;;;;;AAA0GA,KADrTW,2BAAAA,GAA8BL,oBACuRN;AAAiB,KAAtUY,qBAAsU,CAAA,kBAAhStB,mBAAgS,GAAA,SAAA,GAAA,KAAA,GAAA,GAAA,CAAA,GAAA,CAA9OiB,SAA8O,CAAA,SAAA,CAAA,KAAA,CAAA,GAAnNP,iBAAmN,GAA/LO,SAA+L,SAA/K1B,gBAA+K,GAA5JI,qBAA4J,CAAtIsB,SAAsI,CAAA,GAA3HP,iBAA2H,GAAvGO,SAAuG,SAAvFjB,mBAAuF,GAAjEuB,gBAAiE,CAAhDN,SAAgD,CAAA,GAArCP,iBAAqC,GAAjBA,iBAAiB;AAIlV;AAOA;;AAA0EkB,KAP9DJ,gBAO8DI,CAAAA,MAAAA,CAAAA,GAAAA,CAPlCH,MAOkCG,GAAAA;EAI5DxB,MAAAA,CAAAA,EAVDI,YAUCJ;CAqBJE,CAAAA,GAAAA,IAAAA;;;;;AAQGG,UAjCIiB,eAiCJjB,CAAAA,eAjCmCmB,MAiCnCnB,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAjC6DmB,MAiC7DnB,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,WAAAA,OAAAA,CAAAA,CAAAA;EAAO;AAMpB;;EAAsEC,QAAAA,EAnCxDN,UAmCwDM;EAAmEO;;;;;;AAAqC;AAK9K;;;;;;;;;;;;;EAA0Q,IAAA,EAnBhQX,UAmBgQ,GAnBnPC,UAmBmP,GAAA,SAAA;EAQ9PwB;;;EAAoIT,KAAAA,EAvBrIG,MAuBqIH,GAvB5HZ,iBAuB4HY;EAAgCK;;;EAM5JzB,OAAAA,EAzBPO,OAyBOP,CAzBCyB,QAyBDzB,CAAAA;;;AACA;AAcpB;;AAA8Je,KAlClJY,eAkCkJZ,CAAAA,kBAlClHW,MAkCkHX,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAlCxFP,iBAkCwFO,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAlCrCS,eAkCqCT,CAlCrBA,SAkCqBA,EAlCZU,QAkCYV,CAAAA,EAAAA,GAlCEL,cAkCFK,CAlCiBd,WAkCjBc,GAlC+BZ,OAkC/BY,CAAAA;;;;;AAA4DU,KA7B9MG,gBA6B8MH,CAAAA,kBA7B7K3B,mBA6B6K2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EA7BhGD,eA6BgGC,CA7BhFL,qBA6BgFK,CA7B1DV,SA6B0DU,CAAAA,EA7BhDA,QA6BgDA,CAAAA,EAAAA,OAAAA,EA7B5BE,eA6B4BF,CA7BZL,qBA6BYK,CA7BUV,SA6BVU,CAAAA,EA7BoBA,QA6BpBA,CAAAA,EAAAA,GA7BkCf,cA6BlCe,CA7BiDxB,WA6BjDwB,GA7B+DtB,OA6B/DsB,CAAAA;;;;;AAA4B;AAAsB;;AAS7LA,KA9BnEI,oBA8BmEJ,CAAAA,kBA9B9B3B,mBA8B8B2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EA9B+CK,IA8B/CL,CA9BoDhB,YA8BpDgB,CA9BiEL,qBA8BjEK,CA9BuFV,SA8BvFU,CAAAA,EA9BiGA,QA8BjGA,CAAAA;;;;cAA6BH,GAAAA,eAAAA,CAAAA,GAAAA;EAAfZ,YAAAA,CAAAA,EAAAA,MAAAA;EAAc,aAAA,CAAA,EAxBvFV,aAwBuF;AAM3G,CAAA,EAAA,GA7BMU,cA6BMwB,CA7BSnC,SA6BMgB,CAAAA;;;;;;;;;;;AAEC;AAC1B;;AAS6EU,KA3BnEM,iBA2BmEN,CAAAA,kBA3BjC3B,mBA2BiC2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EA3B4ChB,YA2B5CgB,CA3ByDL,qBA2BzDK,CA3B+EV,SA2B/EU,CAAAA,EA3ByFA,QA2BzFA,CAAAA,EAAAA,OAAAA,EA3B6GI,oBA2B7GJ,CA3BkIV,SA2BlIU,EA3B2IA,QA2B3IA,CAAAA,EAAAA,GA3ByJf,cA2BzJe,CA3BwK1B,SA2BxK0B,GA3BoLtB,OA2BpLsB,CAAAA;;;;;;AAA4B;AAM3G;;KAxBKO,kBAwBmJjB,CAAAA,SAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAxBnGA,SAwBmGA,EAAAA,OAAAA,EAxBjFR,OAwBiFQ,CAxBzEU,QAwByEV,CAAAA,EAAAA,GAxB3DL,cAwB2DK,CAxB5CO,gBAwB4CP,CAxB3BkB,OAwB2BlB,CAxBnBA,SAwBmBA,CAAAA,CAAAA,CAAAA;;;;;;AAC3FU,KAnBjDS,eAmBiDT,CAAAA,kBAnBjB3B,mBAmBiB2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAnBkDO,kBAmBlDP,CAnBqEL,qBAmBrEK,CAnB2FV,SAmB3FU,CAAAA,EAnBqGA,QAmBrGA,CAAAA,GAAAA;EAAnDU,IAAAA,EAlBAH,kBAkBAG,CAlBmBf,qBAkBnBe,CAlByCpB,SAkBzCoB,CAAAA,EAlBmDV,QAkBnDU,CAAAA;EACM7B,SAAAA,CAAAA,EAlBAA,YAkBAA,EAAAA;AAAY,CAAA;AAC1B;;;;;;;;AAUwG,KAnBrG6B,kBAmBqG,CAAA,SAAA,EAAA,QAAA,CAAA,GAAA,CAAA,KAAA,EAnBrDpB,SAmBqD,EAAA,OAAA,EAnBnCR,OAmBmC,CAnB3BkB,QAmB2B,CAAA,EAAA,GAnBbf,cAmBa,CAnBEY,gBAmBF,CAnBmBW,OAmBnB,CAnB2BlB,SAmB3B,CAAA,CAAA,CAAA;AAM1G;;;;;AAA8GsB,KAnBlGD,eAmBkGC,CAAAA,kBAnBlEvC,mBAmBkEuC,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAnBCF,kBAmBDE,CAnBoBjB,qBAmBpBiB,CAnB0CtB,SAmB1CsB,CAAAA,EAnBoDZ,QAmBpDY,CAAAA,GAAAA;EAC5DtB,IAAAA,EAnBxCoB,kBAmBwCpB,CAnBrBK,qBAmBqBL,CAnBCA,SAmBDA,CAAAA,EAnBWU,QAmBXV,CAAAA;EAAtBK,SAAAA,CAAAA,EAlBZd,YAkBYc,EAAAA;CAAgCK;;;AAChC;AAC1B;;;;;;KATGY,iBAkBsGf,CAAAA,SAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAlBvDP,SAkBuDO,EAAAA,OAAAA,EAlBrCf,OAkBqCe,CAlB7BG,QAkB6BH,CAAAA,EAAAA,GAlBfZ,cAkBeY,CAlBAA,gBAkBAA,CAlBiBW,OAkBjBX,CAlByBP,SAkBzBO,CAAAA,CAAAA,CAAAA;;AAAD;AAM1G;;;AAAgIF,KAlBpHkB,cAkBoHlB,CAAAA,kBAlBrFtB,mBAkBqFsB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAlBlBiB,iBAkBkBjB,CAlBAA,qBAkBAA,CAlBsBL,SAkBtBK,CAAAA,EAlBgCK,QAkBhCL,CAAAA,GAAAA;EAAgCK,IAAAA,EAjBtJY,iBAiBsJZ,CAjBpIL,qBAiBoIK,CAjB9GV,SAiB8GU,CAAAA,EAjBpGA,QAiBoGA,CAAAA;EAAlDc,SAAAA,CAAAA,EAhB9FjC,YAgB8FiC,EAAAA;CAC5DxB;;;;;AACtB;AAO5B;AAkBA;;KAjCKwB,iBAiC0GlD,CAAAA,SAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAjC3D0B,SAiC2D1B,EAAAA,OAAAA,EAjCzCkB,OAiCyClB,CAjCjCoC,QAiCiCpC,CAAAA,EAAAA,GAjCnBqB,cAiCmBrB,CAjCJiC,gBAiCIjC,CAjCa4C,OAiCb5C,CAjCqB0B,SAiCrB1B,CAAAA,CAAAA,CAAAA;;;;;;AAA2KgB,KA3B9QmC,cA2B8QnC,CAAAA,kBA3B/OP,mBA2B+OO,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GA3B5KkC,iBA2B4KlC,CA3B1Je,qBA2B0Jf,CA3BpIU,SA2BoIV,CAAAA,EA3B1HoB,QA2B0HpB,CAAAA,GAAAA;EAA0BD,IAAAA,EA1B1SmC,iBA0B0SnC,CA1BxRgB,qBA0BwRhB,CA1BlQW,SA0BkQX,CAAAA,EA1BxPqB,QA0BwPrB,CAAAA;EAAaC,SAAAA,CAAAA,EAzBjTC,YAyBiTD,EAAAA;CAKnToC;;;;;;AAkBI1B,cAzCG0B,gBAyCH1B,EAAAA,OAAAA,MAAAA;;;;;;;;;;;;;;;;;;AA4IcA,UAnKf2B,eAmKe3B,CAAAA,kBAnKiBjB,mBAmKjBiB,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,yBAnK+E1B,gBAmK/E0B,GAnKkGzB,iBAmKlGyB,CAnKoH1B,gBAmKpH0B,CAAAA,GAnKwIxB,kBAmKxIwB,CAnK2J1B,gBAmK3J0B,CAAAA,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,eAAAA,GAAAA,EAAAA,eAAAA,SAAAA,CAnK6OX,UAmK7OW,GAnK0PV,UAmK1PU,CAAAA,EAAAA,GAAAA,SAAAA,CAnKoRX,UAmKpRW,GAnKiSV,UAmKjSU,CAAAA,EAAAA,CAAAA,CAAAA;EAASE;;AAAV;AAC9B;EAKeN,UApKF8B,gBAAAA,CAoKE9B,EAAAA,IAAAA;EAAKiC;;;;AAAwC;EAOjDC,SAAAA,kBAAAA,CAAAA,EArKsB/B,oBAqKK,CArKgBC,SAqKhB,EArKyBC,gBAqKzB,EArKyCC,YAqKzC,EArKuDC,MAqKvD,CAAA;EAAMP;;;EAEJmC,IAAAA,EAAAA,MAAAA;EAAgBnC;;;AAAkC;AAO3F;;;EAA2EkC,WAAAA,CAAAA,EAlKzD9B,SAkKyD8B;EAA+BD;AAAC;AAI3G;AAIA;AAIA;AAIA;AACA;EAAuC/B,aAAAA,CAAAA,EA3KnBG,gBA2KmBH;EAAoBxB;;;EAAmEsB,KAAAA,CAAAA,EAvKlHO,MAuKkHP;EAAjB0C;;;;;AAAgF;AAM7L;;;;;;;;;;;;;;AAAwQ;AAMxQ;;;;;;;;;;;;;;AAA4Q;AAI5Q;;;;;;;;;;;;;;AAAmT;AAInT;;;;;;;;;;;EAA2QM,YAAAA,CAAAA,EA5HxP/B,gBA4HwP+B,CA5HvO5C,SA4HuO4C,EA5H9N1C,YA4H8N0C,CAAAA;EAA6DF;;AAAD;AAIvU;;;;;AAAiH;AAIjH;;;;;AAA2H;AAI3H;;;;;;;;AAA+L;AAI/L;;;;EAAsIzC,aAAAA,CAAAA,EA/GlHe,iBA+GkHf,CA/GhGD,SA+GgGC,EA/GvFC,YA+GuFD,CAAAA;EAAuBzB;;;;;;;AAA0I;EAI3R0E,WAAAA,CAAAA,EA1GM/B,eA0GiB,CA1GDnB,SA0GCJ,EA1GQM,YA0GR,CAAA;EAAoByB;;;;;;;;EAA8KoB,WAAAA,CAAAA,EAjGnN1B,eAiGmN0B,CAjGnM/C,SAiGmM+C,EAjG1L7C,YAiG0L6C,CAAAA;EAAwDJ;;;;AAA8B;AAAkB;;;EAI9OU,UAAAA,CAAAA,EA5F9E9B,cA4F8E8B,CA5F/DrD,SA4F+DqD,EA5FtDnD,YA4FsDmD,CAAAA;EAAiBA;;;;;;;;EAAqFD,UAAAA,CAAAA,EAnFpL3B,cAmFoL2B,CAnFrKpD,SAmFqKoD,EAnF5JlD,YAmF4JkD,CAAAA;;AAAK;AAI1M;;KAlFKxB,kBAkF4EhC,CAAAA,CAAAA,CAAAA,GAAAA,QAA6BA,MAjF9FA,CAiF8FA,IAjFzFiC,CAiFyFjC,SAAAA,IAAAA,MAAAA,EAAAA,GAAAA,KAAAA,GAjFxDiC,CAiFwDjC,GAjFpDA,CAiFoDA,CAjFlDiC,CAiFkDjC,CAAAA,EAAkD8C;;;;;;AAA6JC,KA1EjTb,2BA0EiTa,CAAAA,CAAAA,CAAAA,GA1EhR/C,CA0EgR+C,SAAAA;EAA7BW,kBAAAA,CAAAA,EAAAA,KAAAA,MAAAA;CAAtDH,GAxEtOpB,KAwEsOoB,SAxExNpD,oBAwEwNoD,GAxEjMpB,KAwEiMoB,GAAAA,KAAAA,GAxEjLvD,CAwEiLuD,SAxEvKpD,oBAwEuKoD,GAxEhJvD,CAwEgJuD,GAAAA,KAAAA;;;AAAuH;AAIjW;;;AAA4FK,KArEhFxB,mBAqEgFwB,CAAAA,CAAAA,EAAAA,YAAAA,MArEzCzD,oBAqEyCyD,CAAAA,GArEjB1B,2BAqEiB0B,CArEW5D,CAqEX4D,CAAAA,CArEc3B,GAqEd2B,CAAAA;;;;AAA+EA,KAjE/JvB,qBAiE+JuB,CAAAA,CAAAA,CAAAA,GAjEpIxB,mBAiEoIwB,CAjEhH5D,CAiEgH4D,EAAAA,QAAAA,CAAAA;;;;AAA0D,KA7DzNtB,4BA6DyN,CAAA,CAAA,CAAA,GA7DvLF,mBA6DuL,CA7DnKpC,CA6DmK,EAAA,eAAA,CAAA;AACrO;;;AAAwEE,KA1D5DqC,0BA0D4DrC,CAAAA,CAAAA,CAAAA,GA1D5BkC,mBA0D4BlC,CA1DRF,CA0DQE,EAAAA,aAAAA,CAAAA;;;;AAAiFsD,KAtD7IhB,8BAsD6IgB,CAAAA,CAAAA,CAAAA,GAtDzGpB,mBAsDyGoB,CAtDrFxD,CAsDqFwD,EAAAA,OAAAA,CAAAA;AAA5BzE,KArDjH0D,gBAqDiH1D,CAAAA,UArDtFmB,iBAqDsFnB,GArDlEL,gBAqDkEK,CAAAA,GArD9CiB,CAqD8CjB,SArDpCmB,iBAqDoCnB,GArDhB2D,gBAqDgB3D,CArDCiB,CAqDDjB,CAAAA,CAAAA,OAAAA,CAAAA,GArDeiB,CAqDfjB,SArDyBL,gBAqDzBK,GArD4CF,oBAqD5CE,CArDiEiB,CAqDjEjB,CAAAA,GAAAA,CAAAA,CAAAA;AAA2B;AACxJ;;;;AAAwI8E,KAhD5HlB,oBAgD4HkB,CAAAA,UAhD7F9B,eAgD6F8B,CAAAA,GAhD1E7D,CAgD0E6D,SAhDhE9B,eAgDgE8B,CAAAA,KAAAA,QAAAA,EAAAA,GAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GAhDhBzD,OAgDgByD,SAhDAnF,gBAgDAmF,GAhDmB7B,kBAgDnB6B,CAhDsC/E,qBAgDtC+E,CAhD4DzD,OAgD5DyD,CAAAA,CAAAA,GAhDwEzD,OAgDxEyD,SAhDwF1E,mBAgDxF0E,GAhD8G7B,kBAgD9G6B,CAhDiInD,gBAgDjImD,CAhDkJzD,OAgDlJyD,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;;;AAAmEL,KA1C/LZ,yBA0C+LY,CAAAA,UA1C3JzB,eA0C2JyB,CAAAA,GA1CxIxD,CA0CwIwD,SA1C9HzB,eA0C8HyB,CAAAA,KAAAA,QAAAA,EAAAA,GAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GA1C9EpD,OA0C8EoD,SA1C9D9E,gBA0C8D8E,GA1C3CxB,kBA0C2CwB,CA1CxB3E,oBA0CwB2E,CA1CHpD,OA0CGoD,CAAAA,CAAAA,GA1CSpD,OA0CToD,SA1CyBrE,mBA0CzBqE,GA1C+CxB,kBA0C/CwB,CA1CkE9C,gBA0ClE8C,CA1CmFpD,OA0CnFoD,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;AAA+B;KAtC9NX,yCAAyCd,qBAAqB/B,6BAA6BA,kDAAkD8C,cAAcf,kBAAkBgB,sBAAsBhB,oBAAoBY,qBAAqBG,SAASD,sBAAsBE,QAAQJ,qBAAqBG;;;;KAIxSE,8CAA8CjB,qBAAqB/B,6BAA6BA,kDAAkD8C,cAAcf,kBAAkBgB,sBAAsBhB,oBAAoBa,0BAA0BE,SAASE,2BAA2BD,QAAQH,0BAA0BE;;;;KAI5TG,oCAAoClB,qBAAqBc,sBAAsB7C,KAAKH;;;;KAIpFqD,yCAAyCnB,qBAAqBiB,2BAA2BhD,KAAKH;;;;KAI9FsD,iCAAiCpB,mBAAmB/B,UAAU+B,uDAAuD1B,uBAAuB3B,mBAAmBG,qBAAqBwB;;;;KAIpL+C,sCAAsCrB,mBAAmB/B,UAAU+B,uDAAuD1B,uBAAuBzB,kCAAkCC,qBAAqBwE,qBAAqBhD,uBAAuB3B,mBAAmBG,qBAAqBwB;;;;KAI5RiD,2CAA2CvB,qBAAqB/B,6BAA6BA,kDAAkD8C,cAAcf,kBAAkBgB,sBAAsBhB,oBAAoBoB,uBAAuBL,SAASQ,wBAAwBP,QAAQI,uBAAuBL;;;;KAIvTS,2BAA2BC,0BAA0BC,qCAAqCA,iBAAiBA,yBAAyBD,iBAAiBA,YAAYC,KAAKD,KAAKC,YAAYD,KAAKC,IAAID,IAAIC;;;;KAI7LC,gDAAgD3B,qBAAqB/B,6BAA6BA,kDAAkD8C,cAAcf,kBAAkBgB,sBAAsBhB,oBAAoBwB,kBAAkBH,4BAA4BN,QAAQY,6BAA6BX,SAASK,4BAA4BN;;;;KAItVa,wCAAwCzD,oBAAoBxB,oBAAoBkF,sBAAsBlF,mBAAmBG,qBAAqB+E,iBAAiBA,sBAAsB1D,oBAAoBwC,iBAAiBkB;KAC1NlB,2BAA2BvD,uBAAuBqE,UAAUtD,oBAAoBsD,IAAIA,UAAU9E,mBAAmBK,4BAA4ByE;KAC7I9C,2BAA2BvB,mCAAmCqE,UAAUvE,6BAA6BC,uBAAuB2E,WAAWL,UAAU9E,mBAAmBI,sBAAsB0E,KAAKA,UAAUtD,oBAAoBsD"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA4BK,cAAA,MAAoB,CAAA,GAAI,OAAA,CAAQ,CAAA;AAAA,KAEzB,iBAAA,GAAoB,cAAA;;;;;;;;;;;;;;AAAhC;;;;;AA8BA;;;;;;;;;;UAAiB,oBAAA,iBACC,mBAAA,eACZ,mBAAA,qCAGA,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,gBAEnB,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,4DAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EANkB;EAUtB,MAAA,EAAQ,OAAA;EATe;EAWvB,aAAA,EAAe,cAAA;EARU;EAUzB,WAAA,EAAa,YAAA;EATT;EAWJ,KAAA,EAAO,MAAA;AAAA;;;;;KAOG,2BAAA,GAA8B,oBAAA;AAAA,KAE9B,qBAAA,iBACM,mBAAA,+BACb,OAAA,oBACD,iBAAA,GACA,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;;;;KAKI,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EAhDrC;;;EAsDJ,QAAA,EAAU,UAAA;EAnDe;;;;;;;;;;;;;;AAmB3B;;;;;AAEA;EAmDE,IAAA,EAAM,UAAA,GAAa,UAAA;EAnDY;;;EAuD/B,KAAA,EAAO,MAAA,GAAS,iBAAA;EAnDd;;;EAuDF,OAAA,EAAS,SAAA,CAAQ,QAAA;AAAA;;;;;KAOP,eAAA,iBACM,MAAA,oBAA0B,iBAAA,yBAG1C,OAAA,EAAS,eAAA,CAAgB,OAAA,EAAS,QAAA,MAC/B,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;KAMtB,gBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACzD,OAAA,EAAS,eAAA,CAAgB,qBAAA,CAAsB,OAAA,GAAU,QAAA,MACtD,cAAA,CAAe,WAAA,GAAc,OAAA;;;;;;;;KAStB,oBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,IAAA,CACP,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA;;;;;EAKzC,YAAA;EAAuB,aAAA,GAAgB,aAAA;AAAA,MAC1C,cAAA,CAAe,SAAA;AA1FpB;;;;;;;;;;AAUA;;;AAVA,KAyGY,iBAAA,iBACM,mBAAA,iDAGhB,OAAA,EAAS,YAAA,CAAa,qBAAA,CAAsB,OAAA,GAAU,QAAA,GACtD,OAAA,EAAS,oBAAA,CAAqB,OAAA,EAAS,QAAA,MACpC,cAAA,CAAe,SAAA,GAAY,OAAA;;;;;;;;;KAU3B,kBAAA,uBACH,KAAA,EAAO,OAAA,EACP,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,OAAA;;;;;;KAOjC,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EAEjD,IAAA,EAAM,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EACzD,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,OAAA,EACP,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,OAAA;AAnG7C;;;;;AAAA,KA0GY,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EAEjD,IAAA,EAAM,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EACzD,SAAA,GAAY,YAAA;AAAA;;;;;;;;;;KAYb,iBAAA,uBACH,KAAA,EAAO,OAAA,EACP,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,OAAA;;;;;;KAOjC,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EAEhD,IAAA,EAAM,iBAAA,CAAkB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EACxD,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,iBAAA,uBACH,KAAA,EAAO,OAAA,EACP,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,OAAA;;;;;;KAOjC,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EAEhD,IAAA,EAAM,iBAAA,CAAkB,qBAAA,CAAsB,OAAA,GAAU,QAAA;EACxD,SAAA,GAAY,YAAA;AAAA;;;;;;cAQL,gBAAA;;;;;;;;;;AAxJb;;;;;;;;UA2KiB,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EA3KyC;;;;EAAA,UAkLnC,gBAAA;EA3LV;;;;;EAAA,SAkMS,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA;EAlMmC;;;EAwMrC,IAAA;EAnM6C;;;;;;AAgB/C;EA4LE,WAAA,GAAc,OAAA;EA5La;;;;;;;EAqM3B,aAAA,GAAgB,cAAA;EAhMP;;;EAqMT,KAAA,GAAQ,MAAA;EApMS;;;;;;;;;;;;;;;;;;;AAAsB;;;;;;;;;;;;;;;;;;;;;;;;;AAoBzC;;;;;;;;;;;;;;;;;;EA+OE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EA3OpB;;;;;;;;;;;;AAIjB;;;;;;;;;;;;;;;;EAqQJ,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EAzP1B;;;;;;;;EAmQjB,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA3Pd;;;;;;;;EAqQzB,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EA/PsB;;;;;;;;EAyQ7D,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EA3QhB;;;;;;;;EAqRrB,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;AAjRjC;KAuRD,kBAAA,oBACS,CAAA,IAAK,CAAA,gCAAiC,CAAA,GAAI,CAAA,CAAE,CAAA;;;;;;KAQ9C,2BAAA,MAAiC,CAAA;EAC3C,kBAAA;AAAA,IAEE,KAAA,SAAc,oBAAA,GACZ,KAAA,WAEF,CAAA,SAAU,oBAAA,GACR,CAAA;;;;;;;KASM,mBAAA,oBAEM,oBAAA,IACd,2BAAA,CAA4B,CAAA,EAAG,CAAA;;;;KAKvB,qBAAA,MAA2B,mBAAA,CAAoB,CAAA;;;AAnS3D;KAwSY,4BAAA,MAAkC,mBAAA,CAC5C,CAAA;;;;KAOU,0BAAA,MAAgC,mBAAA,CAC1C,CAAA;;;;KAOU,8BAAA,MAAoC,mBAAA,CAAoB,CAAA;AAAA,KAExD,gBAAA,WAA2B,iBAAA,GAAoB,gBAAA,IACzD,CAAA,SAAU,iBAAA,GACN,gBAAA,CAAiB,CAAA,aACjB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA;;;;;;KAQjB,oBAAA,WAA+B,eAAA,IACzC,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;;KASlC,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOlC,qBAAA,oBAAyC,eAAA,MACnD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,oBAAA,CAAqB,KAAA,IAAS,qBAAA,CAAsB,IAAA,IACpD,oBAAA,CAAqB,KAAA;;;;KAOrB,0BAAA,oBAA8C,eAAA,MACxD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,yBAAA,CAA0B,KAAA,IAAS,0BAAA,CAA2B,IAAA,IAC9D,yBAAA,CAA0B,KAAA;AA7WhC;;;AAAA,KAoXM,gBAAA,oBAAoC,eAAA,MAC9C,qBAAA,CAAsB,CAAA,IAAK,iBAAA;;;;KAKjB,qBAAA,oBAAyC,eAAA,MACnD,0BAAA,CAA2B,CAAA,IAAK,iBAAA;;;;KAKtB,sBAAA,WAAiC,eAAA,IAC3C,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOjB,2BAAA,WAAsC,eAAA,IAChD,CAAA,SAAU,eAAA,wCACN,cAAA,SAAuB,kBAAA,gBACrB,oBAAA,CAAqB,KAAA,gBACrB,cAAA,SAAuB,gBAAA,GACrB,oBAAA,CAAqB,cAAA;;;;KAOnB,uBAAA,oBAA2C,eAAA,MACrD,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,sBAAA,CAAuB,KAAA,IAAS,uBAAA,CAAwB,IAAA,IACxD,sBAAA,CAAuB,KAAA;;;;KAO9B,iBAAA,UAA2B,CAAA,yBAC3B,CAAA,oCAEC,CAAA,gBACD,CAAA,wBACC,CAAA,gBACC,CAAA,WAAY,CAAA,IACX,CAAA,IACC,CAAA,WAAY,CAAA,IACX,CAAA,GACA,CAAA,GAAI,CAAA;;AA1Zd;;KA+ZY,4BAAA,oBAAgD,eAAA,MAC1D,CAAA,4BAEI,CAAA,iDACE,KAAA,SAAc,eAAA,GACZ,IAAA,kBAAsB,eAAA,KACpB,iBAAA,CACE,2BAAA,CAA4B,KAAA,GAC5B,4BAAA,CAA6B,IAAA,KAE/B,2BAAA,CAA4B,KAAA;;;;KAO5B,iBAAA,uBACY,iBAAA,GAAoB,gBAAA,IACxC,aAAA,SAAsB,gBAAA,GACtB,oBAAA,CAAqB,aAAA,IACrB,aAAA,SAAsB,iBAAA,GACpB,gBAAA,CAAiB,aAAA;AAAA,KAGX,gBAAA,WAA2B,mBAAA,IACrC,CAAA,SAAU,iBAAA,GACN,CAAA,GACA,CAAA,SAAU,gBAAA,GACR,2BAAA,CAA4B,CAAA;AAAA,KAGxB,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA"}