langchain 1.2.26 → 1.2.27

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

@@ -152,7 +152,61 @@ type ToolRetryMiddlewareConfig = z.input<typeof ToolRetryMiddlewareOptionsSchema
  * @param config - Configuration options for the retry middleware
  * @returns A middleware instance that handles tool failures with retries
  */
-declare function toolRetryMiddleware(config?: ToolRetryMiddlewareConfig): AgentMiddleware;
+declare function toolRetryMiddleware(config?: ToolRetryMiddlewareConfig): AgentMiddleware<undefined, z.ZodObject<{
+  /**
+   * Optional list of tools or tool names to apply retry logic to.
+   * Can be a list of `BaseTool` instances or tool name strings.
+   * If `undefined`, applies to all tools. Default is `undefined`.
+   */
+  tools: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodType<ClientTool, z.ZodTypeDef, ClientTool>, z.ZodType<ServerTool, z.ZodTypeDef, ServerTool>, z.ZodString]>, "many">>;
+  /**
+   * Behavior when all retries are exhausted. Options:
+   * - `"continue"` (default): Return an AIMessage with error details, allowing
+   *   the agent to potentially handle the failure gracefully.
+   * - `"error"`: Re-raise the exception, stopping agent execution.
+   * - Custom function: Function that takes the exception and returns a string
+   *   for the AIMessage content, allowing custom error formatting.
+   *
+   * Deprecated values:
+   * - `"raise"`: use `"error"` instead.
+   * - `"return_message"`: use `"continue"` instead.
+   */
+  onFailure: z.ZodDefault<z.ZodUnion<[z.ZodLiteral<"error">, z.ZodLiteral<"continue">, z.ZodLiteral<"raise">, z.ZodLiteral<"return_message">, z.ZodFunction<z.ZodTuple<[z.ZodType<Error, z.ZodTypeDef, Error>], z.ZodUnknown>, z.ZodString>]>>;
+} & {
+  maxRetries: z.ZodDefault<z.ZodNumber>;
+  retryOn: z.ZodDefault<z.ZodUnion<[z.ZodFunction<z.ZodTuple<[z.ZodType<Error, z.ZodTypeDef, Error>], z.ZodUnknown>, z.ZodBoolean>, z.ZodArray<z.ZodType<new (...args: any[]) => Error, z.ZodTypeDef, new (...args: any[]) => Error>, "many">]>>;
+  backoffFactor: z.ZodDefault<z.ZodNumber>;
+  initialDelayMs: z.ZodDefault<z.ZodNumber>;
+  maxDelayMs: z.ZodDefault<z.ZodNumber>;
+  jitter: z.ZodDefault<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+  maxRetries: number;
+  retryOn: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean);
+  backoffFactor: number;
+  initialDelayMs: number;
+  maxDelayMs: number;
+  jitter: boolean;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string);
+}, {
+  maxRetries?: number | undefined;
+  retryOn?: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean) | undefined;
+  backoffFactor?: number | undefined;
+  initialDelayMs?: number | undefined;
+  maxDelayMs?: number | undefined;
+  jitter?: boolean | undefined;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure?: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string) | undefined;
+}>, {
+  maxRetries: number;
+  retryOn: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean);
+  backoffFactor: number;
+  initialDelayMs: number;
+  maxDelayMs: number;
+  jitter: boolean;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string);
+}, readonly (ServerTool | ClientTool)[]>;
 //#endregion
 export { ToolRetryMiddlewareConfig, toolRetryMiddleware };
 //# sourceMappingURL=toolRetry.d.cts.map

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

@@ -1 +1 @@
-{"version":3,"file":"toolRetry.d.cts","names":[],"sources":["../../../src/agents/middleware/toolRetry.ts"],"mappings":";;;;;;;;cAgBa,gCAAA,EAAgC,CAAA,CAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA2CjC,yBAAA,GAA4B,CAAA,CAAE,KAAA,QACjC,gCAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsGO,mBAAA,CACd,MAAA,GAAQ,yBAAA,GACP,eAAA"}
+{"version":3,"file":"toolRetry.d.cts","names":[],"sources":["../../../src/agents/middleware/toolRetry.ts"],"mappings":";;;;;AAeA;;;AAAA,cAAa,gCAAA,EAAgC,CAAA,CAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA2CjC,yBAAA,GAA4B,CAAA,CAAE,KAAA,QACjC,gCAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsGO,mBAAA,CAAoB,MAAA,GAAQ,yBAAA,8BAA8B,CAAA,CAAA,SAAA"}

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

@@ -152,7 +152,61 @@ type ToolRetryMiddlewareConfig = z.input<typeof ToolRetryMiddlewareOptionsSchema
  * @param config - Configuration options for the retry middleware
  * @returns A middleware instance that handles tool failures with retries
  */
-declare function toolRetryMiddleware(config?: ToolRetryMiddlewareConfig): AgentMiddleware;
+declare function toolRetryMiddleware(config?: ToolRetryMiddlewareConfig): AgentMiddleware<undefined, z.ZodObject<{
+  /**
+   * Optional list of tools or tool names to apply retry logic to.
+   * Can be a list of `BaseTool` instances or tool name strings.
+   * If `undefined`, applies to all tools. Default is `undefined`.
+   */
+  tools: z.ZodOptional<z.ZodArray<z.ZodUnion<[z.ZodType<ClientTool, z.ZodTypeDef, ClientTool>, z.ZodType<ServerTool, z.ZodTypeDef, ServerTool>, z.ZodString]>, "many">>;
+  /**
+   * Behavior when all retries are exhausted. Options:
+   * - `"continue"` (default): Return an AIMessage with error details, allowing
+   *   the agent to potentially handle the failure gracefully.
+   * - `"error"`: Re-raise the exception, stopping agent execution.
+   * - Custom function: Function that takes the exception and returns a string
+   *   for the AIMessage content, allowing custom error formatting.
+   *
+   * Deprecated values:
+   * - `"raise"`: use `"error"` instead.
+   * - `"return_message"`: use `"continue"` instead.
+   */
+  onFailure: z.ZodDefault<z.ZodUnion<[z.ZodLiteral<"error">, z.ZodLiteral<"continue">, z.ZodLiteral<"raise">, z.ZodLiteral<"return_message">, z.ZodFunction<z.ZodTuple<[z.ZodType<Error, z.ZodTypeDef, Error>], z.ZodUnknown>, z.ZodString>]>>;
+} & {
+  maxRetries: z.ZodDefault<z.ZodNumber>;
+  retryOn: z.ZodDefault<z.ZodUnion<[z.ZodFunction<z.ZodTuple<[z.ZodType<Error, z.ZodTypeDef, Error>], z.ZodUnknown>, z.ZodBoolean>, z.ZodArray<z.ZodType<new (...args: any[]) => Error, z.ZodTypeDef, new (...args: any[]) => Error>, "many">]>>;
+  backoffFactor: z.ZodDefault<z.ZodNumber>;
+  initialDelayMs: z.ZodDefault<z.ZodNumber>;
+  maxDelayMs: z.ZodDefault<z.ZodNumber>;
+  jitter: z.ZodDefault<z.ZodBoolean>;
+}, "strip", z.ZodTypeAny, {
+  maxRetries: number;
+  retryOn: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean);
+  backoffFactor: number;
+  initialDelayMs: number;
+  maxDelayMs: number;
+  jitter: boolean;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string);
+}, {
+  maxRetries?: number | undefined;
+  retryOn?: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean) | undefined;
+  backoffFactor?: number | undefined;
+  initialDelayMs?: number | undefined;
+  maxDelayMs?: number | undefined;
+  jitter?: boolean | undefined;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure?: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string) | undefined;
+}>, {
+  maxRetries: number;
+  retryOn: (new (...args: any[]) => Error)[] | ((args_0: Error, ...args: unknown[]) => boolean);
+  backoffFactor: number;
+  initialDelayMs: number;
+  maxDelayMs: number;
+  jitter: boolean;
+  tools?: (string | ServerTool | ClientTool)[] | undefined;
+  onFailure: "continue" | "error" | "raise" | "return_message" | ((args_0: Error, ...args: unknown[]) => string);
+}, readonly (ServerTool | ClientTool)[]>;
 //#endregion
 export { ToolRetryMiddlewareConfig, toolRetryMiddleware };
 //# sourceMappingURL=toolRetry.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"toolRetry.d.ts","names":[],"sources":["../../../src/agents/middleware/toolRetry.ts"],"mappings":";;;;;;;;cAgBa,gCAAA,EAAgC,CAAA,CAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA2CjC,yBAAA,GAA4B,CAAA,CAAE,KAAA,QACjC,gCAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsGO,mBAAA,CACd,MAAA,GAAQ,yBAAA,GACP,eAAA"}
+{"version":3,"file":"toolRetry.d.ts","names":[],"sources":["../../../src/agents/middleware/toolRetry.ts"],"mappings":";;;;;AAeA;;;AAAA,cAAa,gCAAA,EAAgC,CAAA,CAAA,SAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA2CjC,yBAAA,GAA4B,CAAA,CAAE,KAAA,QACjC,gCAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsGO,mBAAA,CAAoB,MAAA,GAAQ,yBAAA,8BAA8B,CAAA,CAAA,SAAA"}

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

@@ -1 +1 @@
-{"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"}
+{"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 { 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(config: ToolRetryMiddlewareConfig = {}) {\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":";;;;;;;;;;;;;;AAeA,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,oBAAoB,SAAoC,EAAE,EAAE;CAC1E,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":[],"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"}
+{"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  InferStateSchemaValue,\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 InferSchemaValueType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodOutput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaValue<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type InferSchemaUpdateType<TSchema> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields> & AgentBuiltInState\n    : TSchema extends InteropZodObject\n      ? InferInteropZodInput<TSchema> & AgentBuiltInState\n      : TSchema extends StateDefinitionInit\n        ? InferSchemaInput<TSchema> & AgentBuiltInState\n        : AgentBuiltInState;\n\nexport type NormalizedSchemaInput<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaValueType<TSchema>;\n\nexport type NormalizedSchemaUpdate<\n  TSchema extends StateDefinitionInit | undefined | never = any,\n> = InferSchemaUpdateType<TSchema>;\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: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<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<TSchema, TContext>\n  | {\n      hook: BeforeAgentHandler<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: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<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<TSchema, TContext>\n  | {\n      hook: BeforeModelHandler<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: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<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<TSchema, TContext>\n  | {\n      hook: AfterModelHandler<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: InferSchemaValueType<TSchema>,\n  runtime: Runtime<TContext>\n) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<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<TSchema, TContext>\n  | {\n      hook: AfterAgentHandler<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 StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaValue<TFields>>\n      : TSchema extends InteropZodObject\n        ? FilterPrivateProps<InferInteropZodOutput<TSchema>>\n        : TSchema extends StateDefinitionInit\n          ? FilterPrivateProps<InferSchemaValue<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 StateSchema<infer TFields>\n      ? FilterPrivateProps<InferStateSchemaUpdate<TFields>>\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 InferSchemaValue<A extends StateDefinitionInit | undefined> =\n  A extends StateSchema<infer TFields>\n    ? InferStateSchemaValue<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodOutput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"State\"]\n        : {};\n\nexport type InferSchemaInput<A extends StateDefinitionInit | undefined> =\n  A extends StateSchema<infer TFields>\n    ? InferStateSchemaUpdate<TFields>\n    : A extends InteropZodObject\n      ? InferInteropZodInput<A>\n      : A extends AnyAnnotationRoot\n        ? A[\"Update\"]\n        : {};\n"],"mappings":";;;;;;;AAsWA,MAAa,mBAAkC,OAAO,kBAAkB"}

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

@@ -4,7 +4,7 @@ import { ModelRequest } from "../nodes/types.cjs";
 import { AIMessage, SystemMessage, ToolMessage } from "@langchain/core/messages";
 import { ClientTool, ServerTool } from "@langchain/core/tools";
 import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional } from "@langchain/core/utils/types";
-import { AnnotationRoot, Command, InferStateSchemaUpdate, StateDefinitionInit, StateSchema } from "@langchain/langgraph";
+import { AnnotationRoot, Command, InferStateSchemaUpdate, InferStateSchemaValue, StateDefinitionInit, StateSchema } from "@langchain/langgraph";
 import { ToolCall as ToolCall$1 } from "@langchain/core/messages/tool";
 import { InteropZodToStateDefinition } from "@langchain/langgraph/zod";
 
@@ -54,7 +54,10 @@ interface MiddlewareTypeConfig<TSchema extends StateDefinitionInit | undefined =
  * Used when no explicit type parameters are provided.
  */
 type DefaultMiddlewareTypeConfig = MiddlewareTypeConfig;
-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;
+type InferSchemaValueType<TSchema> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> & AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends StateDefinitionInit ? InferSchemaValue<TSchema> & AgentBuiltInState : AgentBuiltInState;
+type InferSchemaUpdateType<TSchema> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> & AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> & AgentBuiltInState : TSchema extends StateDefinitionInit ? InferSchemaInput<TSchema> & AgentBuiltInState : AgentBuiltInState;
+type NormalizedSchemaInput<TSchema extends StateDefinitionInit | undefined | never = any> = InferSchemaValueType<TSchema>;
+type NormalizedSchemaUpdate<TSchema extends StateDefinitionInit | undefined | never = any> = InferSchemaUpdateType<TSchema>;
 /**
  * Result type for middleware functions.
  */
@@ -147,14 +150,14 @@ type WrapModelCallHook<TSchema 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 BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
+type BeforeAgentHandler<TSchema, TContext> = (state: InferSchemaValueType<TSchema>, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;
 /**
  * Hook type for the beforeAgent lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called once at the start of the agent invocation.
  */
-type BeforeAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
-  hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
+type BeforeAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeAgentHandler<TSchema, TContext> | {
+  hook: BeforeAgentHandler<TSchema, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -165,14 +168,14 @@ type BeforeAgentHook<TSchema extends StateDefinitionInit | undefined = undefined
  * @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, TContext> = (state: TSchema, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
+type BeforeModelHandler<TSchema, TContext> = (state: InferSchemaValueType<TSchema>, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;
 /**
  * Hook type for the beforeModel lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called before each model invocation.
  */
-type BeforeModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
-  hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
+type BeforeModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = BeforeModelHandler<TSchema, TContext> | {
+  hook: BeforeModelHandler<TSchema, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -184,14 +187,14 @@ type BeforeModelHook<TSchema extends StateDefinitionInit | undefined = undefined
  * @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, TContext> = (state: TSchema, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
+type AfterModelHandler<TSchema, TContext> = (state: InferSchemaValueType<TSchema>, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;
 /**
  * Hook type for the afterModel lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called after each model invocation.
  */
-type AfterModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {
-  hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;
+type AfterModelHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterModelHandler<TSchema, TContext> | {
+  hook: AfterModelHandler<TSchema, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -202,14 +205,14 @@ type AfterModelHook<TSchema extends StateDefinitionInit | undefined = undefined,
  * @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, TContext> = (state: TSchema, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;
+type AfterAgentHandler<TSchema, TContext> = (state: InferSchemaValueType<TSchema>, runtime: Runtime$1<TContext>) => PromiseOrValue<MiddlewareResult<Partial<InferSchemaUpdateType<TSchema>>>>;
 /**
  * Hook type for the afterAgent lifecycle event.
  * Can be either a handler function or an object with a handler and optional jump targets.
  * This hook is called once at the end of the agent invocation.
  */
-type AfterAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {
-  hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
+type AfterAgentHook<TSchema extends StateDefinitionInit | undefined = undefined, TContext = unknown> = AfterAgentHandler<TSchema, TContext> | {
+  hook: AfterAgentHandler<TSchema, TContext>;
   canJumpTo?: JumpToTarget[];
 };
 /**
@@ -441,13 +444,13 @@ type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extend
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-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>> : {} : {};
+type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaValue<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaValue<TSchema>> : {} : {};
 /**
  * Helper type to infer the input state schema type from a middleware (all properties optional)
  * This filters out private properties (those starting with underscore)
  * Supports both Zod schemas (InteropZodObject) and StateSchema from LangGraph
  */
-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>> : {} : {};
+type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer TSchema, any, any, any> ? TSchema extends StateSchema<infer TFields> ? FilterPrivateProps<InferStateSchemaUpdate<TFields>> : TSchema extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<TSchema>> : TSchema extends StateDefinitionInit ? FilterPrivateProps<InferSchemaInput<TSchema>> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (just the middleware states)
  */
@@ -489,7 +492,8 @@ type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T exte
  */
 type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>["State"] : {};
 type ToAnnotationRoot<A extends StateDefinitionInit> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? InteropZodToStateDefinition<A> : never;
-type InferSchemaInput<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
+type InferSchemaValue<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaValue<TFields> : A extends InteropZodObject ? InferInteropZodOutput<A> : A extends AnyAnnotationRoot ? A["State"] : {};
+type InferSchemaInput<A extends StateDefinitionInit | undefined> = A extends StateSchema<infer TFields> ? InferStateSchemaUpdate<TFields> : A extends InteropZodObject ? InferInteropZodInput<A> : A extends AnyAnnotationRoot ? A["Update"] : {};
 //#endregion
-export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
+export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, DefaultMiddlewareTypeConfig, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContextSchema, InferMiddlewareContexts, InferMiddlewareFullContext, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareSchema, InferMiddlewareState, InferMiddlewareStates, InferMiddlewareToolsFromConfig, InferMiddlewareType, InferSchemaInput, InferSchemaUpdateType, InferSchemaValue, InferSchemaValueType, MIDDLEWARE_BRAND, MiddlewareResult, MiddlewareTypeConfig, NormalizedSchemaInput, NormalizedSchemaUpdate, ResolveMiddlewareTypeConfig, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
 //# sourceMappingURL=types.d.cts.map

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

@@ -1 +1 @@
-{"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"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/agents/middleware/types.ts"],"mappings":";;;;;;;;;;;KA6BK,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,oBAAA,aAAiC,OAAA,oBACzC,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,gBAAA,GACd,qBAAA,CAAsB,OAAA,IAAW,iBAAA,GACjC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,aAAkC,OAAA,oBAC1C,iBAAA,GACA,OAAA,SAAgB,WAAA,kBACd,sBAAA,CAAuB,OAAA,IAAW,iBAAA,GAClC,OAAA,SAAgB,gBAAA,GACd,oBAAA,CAAqB,OAAA,IAAW,iBAAA,GAChC,OAAA,SAAgB,mBAAA,GACd,gBAAA,CAAiB,OAAA,IAAW,iBAAA,GAC5B,iBAAA;AAAA,KAEE,qBAAA,iBACM,mBAAA,8BACd,oBAAA,CAAqB,OAAA;AAAA,KAEb,sBAAA,iBACM,mBAAA,8BACd,qBAAA,CAAsB,OAAA;;;;KAKd,gBAAA,YACP,MAAA;EACC,MAAA,GAAS,YAAA;AAAA;;;;;UAQE,eAAA,gBACA,MAAA,oBAA0B,MAAA;EA/DzC;;;EAqEA,QAAA,EAAU,UAAA;EAnEN;;;;;;;;;;;AAiBN;;;;;AAEA;;;;EAqEE,IAAA,EAAM,UAAA,GAAa,UAAA;EAnEjB;;;EAuEF,KAAA,EAAO,MAAA,GAAS,iBAAA;EAtEqB;;;EA0ErC,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;;;;;;;AA/GpB;;;;;;;KA8HY,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,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,kBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,eAAA,iBACM,mBAAA,gDAGd,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAE1B,IAAA,EAAM,kBAAA,CAAmB,OAAA,EAAS,QAAA;EAClC,SAAA,GAAY,YAAA;AAAA;;AAjLlB;;;;;;;;KA6LK,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;;;;;KAOvD,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;;;;;;;KAWb,iBAAA,uBACH,KAAA,EAAO,oBAAA,CAAqB,OAAA,GAC5B,OAAA,EAAS,SAAA,CAAQ,QAAA,MACd,cAAA,CAAe,gBAAA,CAAiB,OAAA,CAAQ,qBAAA,CAAsB,OAAA;;AAjNnE;;;;KAwNY,cAAA,iBACM,mBAAA,gDAGd,iBAAA,CAAkB,OAAA,EAAS,QAAA;EAEzB,IAAA,EAAM,iBAAA,CAAkB,OAAA,EAAS,QAAA;EACjC,SAAA,GAAY,YAAA;AAAA;;;AArNlB;;;cA6Na,gBAAA;;;;;;;;;;;;;;;;;;UAmBI,eAAA,iBACC,mBAAA,2CAEZ,gBAAA,GACA,iBAAA,CAAkB,gBAAA,IAClB,kBAAA,CAAmB,gBAAA,kEAGE,UAAA,GAAa,UAAA,gBAClC,UAAA,GACA,UAAA;EA9Ne;;;;EAAA,UAqOT,gBAAA;EA7ND;;;;AAOX;EAPW,SAoOA,kBAAA,GAAqB,oBAAA,CAC5B,OAAA,EACA,cAAA,EACA,YAAA,EACA,MAAA;EAjOuB;;;EAuOzB,IAAA;EAnOkC;;;;;;;EA4OlC,WAAA,GAAc,OAAA;EA/OE;;;;;;;EAwPhB,aAAA,GAAgB,cAAA;EApPE;;;EAyPlB,KAAA,GAAQ,MAAA;EAnPE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAeZ;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmSE,YAAA,GAAe,gBAAA,CAAiB,OAAA,EAAS,YAAA;EAxRd;;AAe7B;;;;;;;;;;;;;;;;;;;;;;;;;;EAuSE,aAAA,GAAgB,iBAAA,CAAkB,OAAA,EAAS,YAAA;EAlS3C;;;;;;AACuC;;EA2SvC,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAhSX;;;;;;;;EA0S5B,WAAA,GAAc,eAAA,CAAgB,OAAA,EAAS,YAAA;EAxStB;;;;;;;;EAkTjB,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;EAnTrC;;;;;;;;EA6TA,UAAA,GAAa,cAAA,CAAe,OAAA,EAAS,YAAA;AAAA;;;;KAMlC,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;AApVrD;;;AAAA,KAyVM,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,WAAA,kBACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,qBAAA,CAAsB,OAAA,KACzC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;;;KASpC,yBAAA,WAAoC,eAAA,IAC9C,CAAA,SAAU,eAAA,iCACN,OAAA,SAAgB,WAAA,kBACd,kBAAA,CAAmB,sBAAA,CAAuB,OAAA,KAC1C,OAAA,SAAgB,gBAAA,GACd,kBAAA,CAAmB,oBAAA,CAAqB,OAAA,KACxC,OAAA,SAAgB,mBAAA,GACd,kBAAA,CAAmB,gBAAA,CAAiB,OAAA;;;;KAOpC,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;AAxYjC;;;AAAA,KA+YY,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;;;;KAO1B,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;;;;KAKF,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,qBAAA,CAAsB,OAAA,IACtB,CAAA,SAAU,gBAAA,GACR,qBAAA,CAAsB,CAAA,IACtB,CAAA,SAAU,iBAAA,GACR,CAAA;AAAA,KAGE,gBAAA,WAA2B,mBAAA,gBACrC,CAAA,SAAU,WAAA,kBACN,sBAAA,CAAuB,OAAA,IACvB,CAAA,SAAU,gBAAA,GACR,oBAAA,CAAqB,CAAA,IACrB,CAAA,SAAU,iBAAA,GACR,CAAA"}