langchain 1.1.6-dev-1765433794876 → 1.2.0

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

@@ -1 +1 @@
-{"version":3,"file":"summarization.js","names":["input: BaseLanguageModel","options: SummarizationMiddlewareConfig","z4","trigger: ContextSize | ContextSize[] | undefined","keep: ContextSize","triggerConditions: ContextSize[]","uuid","messages: BaseMessage[]","systemPrompt: SystemMessage | undefined","conversationMessages: BaseMessage[]","cutoffIndex: number","totalTokens: number","model: BaseLanguageModel","tokenCounter: TokenCounter","targetTokenCount: number","messagesToKeep: number","aiMessage: AIMessage","aiMessageIndex: number","toolCallIds: Set<string>","messagesToSummarize: BaseMessage[]","summaryPrompt: string","trimTokensToSummarize: number | undefined"],"sources":["../../../src/agents/middleware/summarization.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { z as z4 } from \"zod/v4\";\nimport { v4 as uuid } from \"uuid\";\nimport {\n  BaseMessage,\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n  RemoveMessage,\n  trimMessages,\n  HumanMessage,\n} from \"@langchain/core/messages\";\nimport {\n  BaseLanguageModel,\n  getModelContextSize,\n} from \"@langchain/core/language_models/base\";\nimport {\n  interopSafeParse,\n  InferInteropZodInput,\n  InferInteropZodOutput,\n} from \"@langchain/core/utils/types\";\nimport { REMOVE_ALL_MESSAGES } from \"@langchain/langgraph\";\nimport { createMiddleware } from \"../middleware.js\";\nimport { countTokensApproximately } from \"./utils.js\";\nimport { hasToolCalls } from \"../utils.js\";\nimport { initChatModel } from \"../../chat_models/universal.js\";\n\nexport const DEFAULT_SUMMARY_PROMPT = `<role>\nContext Extraction Assistant\n</role>\n\n<primary_objective>\nYour sole objective in this task is to extract the highest quality/most relevant context from the conversation history below.\n</primary_objective>\n\n<objective_information>\nYou're nearing the total number of input tokens you can accept, so you must extract the highest quality/most relevant pieces of information from your conversation history.\nThis context will then overwrite the conversation history presented below. Because of this, ensure the context you extract is only the most important information to your overall goal.\n</objective_information>\n\n<instructions>\nThe conversation history below will be replaced with the context you extract in this step. Because of this, you must do your very best to extract and record all of the most important context from the conversation history.\nYou want to ensure that you don't repeat any actions you've already completed, so the context you extract from the conversation history should be focused on the most important information to your overall goal.\n</instructions>\n\nThe user will message you with the full message history you'll be extracting context from, to then replace. Carefully read over it all, and think deeply about what information is most important to your overall goal that should be saved:\n\nWith all of this in mind, please carefully read over the entire conversation history, and extract the most important and relevant context to replace it so that you can free up space in the conversation history.\nRespond ONLY with the extracted context. Do not include any additional information, or text before or after the extracted context.\n\n<messages>\nMessages to summarize:\n{messages}\n</messages>`;\n\nconst DEFAULT_MESSAGES_TO_KEEP = 20;\nconst DEFAULT_TRIM_TOKEN_LIMIT = 4000;\nconst DEFAULT_FALLBACK_MESSAGE_COUNT = 15;\nconst SEARCH_RANGE_FOR_TOOL_PAIRS = 5;\n\nconst tokenCounterSchema = z\n  .function()\n  .args(z.array(z.custom<BaseMessage>()))\n  .returns(z.union([z.number(), z.promise(z.number())]));\nexport type TokenCounter = (\n  messages: BaseMessage[]\n) => number | Promise<number>;\n\nexport const contextSizeSchema = z\n  .object({\n    /**\n     * Fraction of the model's context size to use as the trigger\n     */\n    fraction: z\n      .number()\n      .gt(0, \"Fraction must be greater than 0\")\n      .max(1, \"Fraction must be less than or equal to 1\")\n      .optional(),\n    /**\n     * Number of tokens to use as the trigger\n     */\n    tokens: z.number().positive(\"Tokens must be greater than 0\").optional(),\n    /**\n     * Number of messages to use as the trigger\n     */\n    messages: z\n      .number()\n      .int(\"Messages must be an integer\")\n      .positive(\"Messages must be greater than 0\")\n      .optional(),\n  })\n  .refine(\n    (data) => {\n      const count = [data.fraction, data.tokens, data.messages].filter(\n        (v) => v !== undefined\n      ).length;\n      return count >= 1;\n    },\n    {\n      message: \"At least one of fraction, tokens, or messages must be provided\",\n    }\n  );\nexport type ContextSize = z.infer<typeof contextSizeSchema>;\n\nexport const keepSchema = z\n  .object({\n    /**\n     * Fraction of the model's context size to keep\n     */\n    fraction: z\n      .number()\n      .min(0, \"Messages must be non-negative\")\n      .max(1, \"Fraction must be less than or equal to 1\")\n      .optional(),\n    /**\n     * Number of tokens to keep\n     */\n    tokens: z\n      .number()\n      .min(0, \"Tokens must be greater than or equal to 0\")\n      .optional(),\n    messages: z\n      .number()\n      .int(\"Messages must be an integer\")\n      .min(0, \"Messages must be non-negative\")\n      .optional(),\n  })\n  .refine(\n    (data) => {\n      const count = [data.fraction, data.tokens, data.messages].filter(\n        (v) => v !== undefined\n      ).length;\n      return count === 1;\n    },\n    {\n      message: \"Exactly one of fraction, tokens, or messages must be provided\",\n    }\n  );\nexport type KeepSize = z.infer<typeof keepSchema>;\n\nconst contextSchema = z.object({\n  /**\n   * Model to use for summarization\n   */\n  model: z.custom<string | BaseLanguageModel>(),\n  /**\n   * Trigger conditions for summarization.\n   * Can be a single condition object (all properties must be met) or an array of conditions (any condition must be met).\n   *\n   * @example\n   * ```ts\n   * // Single condition: trigger if tokens >= 5000 AND messages >= 3\n   * trigger: { tokens: 5000, messages: 3 }\n   *\n   * // Multiple conditions: trigger if (tokens >= 5000 AND messages >= 3) OR (tokens >= 3000 AND messages >= 6)\n   * trigger: [\n   *   { tokens: 5000, messages: 3 },\n   *   { tokens: 3000, messages: 6 }\n   * ]\n   * ```\n   */\n  trigger: z.union([contextSizeSchema, z.array(contextSizeSchema)]).optional(),\n  /**\n   * Keep conditions for summarization\n   */\n  keep: keepSchema.optional(),\n  /**\n   * Token counter function to use for summarization\n   */\n  tokenCounter: tokenCounterSchema.optional(),\n  /**\n   * Summary prompt to use for summarization\n   * @default {@link DEFAULT_SUMMARY_PROMPT}\n   */\n  summaryPrompt: z.string().default(DEFAULT_SUMMARY_PROMPT),\n  /**\n   * Number of tokens to trim to before summarizing\n   */\n  trimTokensToSummarize: z.number().optional(),\n  /**\n   * Prefix to add to the summary\n   */\n  summaryPrefix: z.string().optional(),\n  /**\n   * @deprecated Use `trigger: { tokens: value }` instead.\n   */\n  maxTokensBeforeSummary: z.number().optional(),\n  /**\n   * @deprecated Use `keep: { messages: value }` instead.\n   */\n  messagesToKeep: z.number().optional(),\n});\n\nexport type SummarizationMiddlewareConfig = InferInteropZodInput<\n  typeof contextSchema\n>;\n\n/**\n * Get max input tokens from model profile or fallback to model name lookup\n */\nexport function getProfileLimits(input: BaseLanguageModel): number | undefined {\n  // Access maxInputTokens on the model profile directly if available\n  if (\n    \"profile\" in input &&\n    typeof input.profile === \"object\" &&\n    input.profile &&\n    \"maxInputTokens\" in input.profile &&\n    (typeof input.profile.maxInputTokens === \"number\" ||\n      input.profile.maxInputTokens == null)\n  ) {\n    return input.profile.maxInputTokens ?? undefined;\n  }\n\n  // Fallback to using model name if available\n  if (\"model\" in input && typeof input.model === \"string\") {\n    return getModelContextSize(input.model);\n  }\n  if (\"modelName\" in input && typeof input.modelName === \"string\") {\n    return getModelContextSize(input.modelName);\n  }\n\n  return undefined;\n}\n\n/**\n * Summarization middleware that automatically summarizes conversation history when token limits are approached.\n *\n * This middleware monitors message token counts and automatically summarizes older\n * messages when a threshold is reached, preserving recent messages and maintaining\n * context continuity by ensuring AI/Tool message pairs remain together.\n *\n * @param options Configuration options for the summarization middleware\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * import { summarizationMiddleware } from \"langchain\";\n * import { createAgent } from \"langchain\";\n *\n * // Single condition: trigger if tokens >= 4000 AND messages >= 10\n * const agent1 = createAgent({\n *   llm: model,\n *   tools: [getWeather],\n *   middleware: [\n *     summarizationMiddleware({\n *       model: new ChatOpenAI({ model: \"gpt-4o\" }),\n *       trigger: { tokens: 4000, messages: 10 },\n *       keep: { messages: 20 },\n *     })\n *   ],\n * });\n *\n * // Multiple conditions: trigger if (tokens >= 5000 AND messages >= 3) OR (tokens >= 3000 AND messages >= 6)\n * const agent2 = createAgent({\n *   llm: model,\n *   tools: [getWeather],\n *   middleware: [\n *     summarizationMiddleware({\n *       model: new ChatOpenAI({ model: \"gpt-4o\" }),\n *       trigger: [\n *         { tokens: 5000, messages: 3 },\n *         { tokens: 3000, messages: 6 },\n *       ],\n *       keep: { messages: 20 },\n *     })\n *   ],\n * });\n *\n * ```\n */\nexport function summarizationMiddleware(\n  options: SummarizationMiddlewareConfig\n) {\n  /**\n   * Parse user options to get their explicit values\n   */\n  const { data: userOptions, error } = interopSafeParse(contextSchema, options);\n  if (error) {\n    throw new Error(\n      `Invalid summarization middleware options: ${z4.prettifyError(error)}`\n    );\n  }\n\n  return createMiddleware({\n    name: \"SummarizationMiddleware\",\n    contextSchema: contextSchema.extend({\n      /**\n       * `model` should be required when initializing the middleware,\n       * but can be omitted within context when invoking the middleware.\n       */\n      model: z.custom<BaseLanguageModel>().optional(),\n    }),\n    beforeModel: async (state, runtime) => {\n      let trigger: ContextSize | ContextSize[] | undefined =\n        userOptions.trigger;\n      let keep: ContextSize = userOptions.keep as InferInteropZodOutput<\n        typeof keepSchema\n      >;\n\n      /**\n       * Handle deprecated parameters\n       */\n      if (userOptions.maxTokensBeforeSummary !== undefined) {\n        console.warn(\n          \"maxTokensBeforeSummary is deprecated. Use `trigger: { tokens: value }` instead.\"\n        );\n        if (trigger === undefined) {\n          trigger = { tokens: userOptions.maxTokensBeforeSummary };\n        }\n      }\n\n      /**\n       * Handle deprecated parameters\n       */\n      if (userOptions.messagesToKeep !== undefined) {\n        console.warn(\n          \"messagesToKeep is deprecated. Use `keep: { messages: value }` instead.\"\n        );\n        if (\n          !keep ||\n          (keep &&\n            \"messages\" in keep &&\n            keep.messages === DEFAULT_MESSAGES_TO_KEEP)\n        ) {\n          keep = { messages: userOptions.messagesToKeep };\n        }\n      }\n\n      /**\n       * Merge context with user options\n       */\n      const resolvedTrigger =\n        runtime.context?.trigger !== undefined\n          ? runtime.context.trigger\n          : trigger;\n      const resolvedKeep =\n        runtime.context?.keep !== undefined\n          ? runtime.context.keep\n          : keep ?? { messages: DEFAULT_MESSAGES_TO_KEEP };\n\n      const validatedKeep = keepSchema.parse(resolvedKeep);\n\n      /**\n       * Validate trigger conditions\n       */\n      let triggerConditions: ContextSize[] = [];\n      if (resolvedTrigger === undefined) {\n        triggerConditions = [];\n      } else if (Array.isArray(resolvedTrigger)) {\n        /**\n         * It's an array of ContextSize objects\n         */\n        triggerConditions = (resolvedTrigger as ContextSize[]).map((t) =>\n          contextSizeSchema.parse(t)\n        );\n      } else {\n        /**\n         * Single ContextSize object - all properties must be satisfied (AND logic)\n         */\n        triggerConditions = [contextSizeSchema.parse(resolvedTrigger)];\n      }\n\n      /**\n       * Check if profile is required\n       */\n      const requiresProfile =\n        triggerConditions.some((c) => \"fraction\" in c) ||\n        \"fraction\" in validatedKeep;\n\n      const model =\n        typeof userOptions.model === \"string\"\n          ? await initChatModel(userOptions.model)\n          : userOptions.model;\n\n      if (requiresProfile && !getProfileLimits(model)) {\n        throw new Error(\n          \"Model profile information is required to use fractional token limits. \" +\n            \"Use absolute token counts instead.\"\n        );\n      }\n\n      const summaryPrompt =\n        runtime.context?.summaryPrompt === DEFAULT_SUMMARY_PROMPT\n          ? userOptions.summaryPrompt ?? DEFAULT_SUMMARY_PROMPT\n          : runtime.context?.summaryPrompt ??\n            userOptions.summaryPrompt ??\n            DEFAULT_SUMMARY_PROMPT;\n      const trimTokensToSummarize =\n        runtime.context?.trimTokensToSummarize !== undefined\n          ? runtime.context.trimTokensToSummarize\n          : userOptions.trimTokensToSummarize ?? DEFAULT_TRIM_TOKEN_LIMIT;\n\n      /**\n       * Ensure all messages have IDs\n       */\n      ensureMessageIds(state.messages);\n\n      const tokenCounter =\n        runtime.context?.tokenCounter !== undefined\n          ? runtime.context.tokenCounter\n          : userOptions.tokenCounter ?? countTokensApproximately;\n      const totalTokens = await tokenCounter(state.messages);\n      const doSummarize = await shouldSummarize(\n        state.messages,\n        totalTokens,\n        triggerConditions,\n        model\n      );\n\n      if (!doSummarize) {\n        return;\n      }\n\n      const { systemPrompt, conversationMessages } = splitSystemMessage(\n        state.messages\n      );\n      const cutoffIndex = await determineCutoffIndex(\n        conversationMessages,\n        validatedKeep,\n        tokenCounter,\n        model\n      );\n\n      if (cutoffIndex <= 0) {\n        return;\n      }\n\n      const { messagesToSummarize, preservedMessages } = partitionMessages(\n        systemPrompt,\n        conversationMessages,\n        cutoffIndex\n      );\n\n      const summary = await createSummary(\n        messagesToSummarize,\n        model,\n        summaryPrompt,\n        tokenCounter,\n        trimTokensToSummarize\n      );\n\n      const summaryMessage = new HumanMessage({\n        content: `Here is a summary of the conversation to date:\\n\\n${summary}`,\n        id: uuid(),\n      });\n\n      return {\n        messages: [\n          new RemoveMessage({ id: REMOVE_ALL_MESSAGES }),\n          summaryMessage,\n          ...preservedMessages,\n        ],\n      };\n    },\n  });\n}\n\n/**\n * Ensure all messages have unique IDs\n */\nfunction ensureMessageIds(messages: BaseMessage[]): void {\n  for (const msg of messages) {\n    if (!msg.id) {\n      msg.id = uuid();\n    }\n  }\n}\n\n/**\n * Separate system message from conversation messages\n */\nfunction splitSystemMessage(messages: BaseMessage[]): {\n  systemPrompt?: SystemMessage;\n  conversationMessages: BaseMessage[];\n} {\n  if (messages.length > 0 && SystemMessage.isInstance(messages[0])) {\n    return {\n      systemPrompt: messages[0] as SystemMessage,\n      conversationMessages: messages.slice(1),\n    };\n  }\n  return {\n    conversationMessages: messages,\n  };\n}\n\n/**\n * Partition messages into those to summarize and those to preserve\n */\nfunction partitionMessages(\n  systemPrompt: SystemMessage | undefined,\n  conversationMessages: BaseMessage[],\n  cutoffIndex: number\n): { messagesToSummarize: BaseMessage[]; preservedMessages: BaseMessage[] } {\n  const messagesToSummarize = conversationMessages.slice(0, cutoffIndex);\n  const preservedMessages = conversationMessages.slice(cutoffIndex);\n\n  // Include system message in messages to summarize to capture previous summaries\n  if (systemPrompt) {\n    messagesToSummarize.unshift(systemPrompt);\n  }\n\n  return { messagesToSummarize, preservedMessages };\n}\n\n/**\n * Determine whether summarization should run for the current token usage\n *\n * @param messages - Current messages in the conversation\n * @param totalTokens - Total token count for all messages\n * @param triggerConditions - Array of trigger conditions. Returns true if ANY condition is satisfied (OR logic).\n *                           Within each condition, ALL specified properties must be satisfied (AND logic).\n * @param model - The language model being used\n * @returns true if summarization should be triggered\n */\nasync function shouldSummarize(\n  messages: BaseMessage[],\n  totalTokens: number,\n  triggerConditions: ContextSize[],\n  model: BaseLanguageModel\n): Promise<boolean> {\n  if (triggerConditions.length === 0) {\n    return false;\n  }\n\n  /**\n   * Check each condition (OR logic between conditions)\n   */\n  for (const trigger of triggerConditions) {\n    /**\n     * Within a single condition, all specified properties must be satisfied (AND logic)\n     */\n    let conditionMet = true;\n    let hasAnyProperty = false;\n\n    if (trigger.messages !== undefined) {\n      hasAnyProperty = true;\n      if (messages.length < trigger.messages) {\n        conditionMet = false;\n      }\n    }\n\n    if (trigger.tokens !== undefined) {\n      hasAnyProperty = true;\n      if (totalTokens < trigger.tokens) {\n        conditionMet = false;\n      }\n    }\n\n    if (trigger.fraction !== undefined) {\n      hasAnyProperty = true;\n      const maxInputTokens = getProfileLimits(model);\n      if (typeof maxInputTokens === \"number\") {\n        const threshold = Math.floor(maxInputTokens * trigger.fraction);\n        if (totalTokens < threshold) {\n          conditionMet = false;\n        }\n      } else {\n        /**\n         * If fraction is specified but we can't get model limits, skip this condition\n         */\n        conditionMet = false;\n      }\n    }\n\n    /**\n     * If condition has at least one property and all properties are satisfied, trigger summarization\n     */\n    if (hasAnyProperty && conditionMet) {\n      return true;\n    }\n  }\n\n  return false;\n}\n\n/**\n * Determine cutoff index respecting retention configuration\n */\nasync function determineCutoffIndex(\n  messages: BaseMessage[],\n  keep: ContextSize,\n  tokenCounter: TokenCounter,\n  model: BaseLanguageModel\n): Promise<number> {\n  if (\"tokens\" in keep || \"fraction\" in keep) {\n    const tokenBasedCutoff = await findTokenBasedCutoff(\n      messages,\n      keep,\n      tokenCounter,\n      model\n    );\n    if (typeof tokenBasedCutoff === \"number\") {\n      return tokenBasedCutoff;\n    }\n    /**\n     * Fallback to message count if token-based fails\n     */\n    return findSafeCutoff(messages, DEFAULT_MESSAGES_TO_KEEP);\n  }\n  /**\n   * find cutoff index based on message count\n   */\n  return findSafeCutoff(messages, keep.messages ?? DEFAULT_MESSAGES_TO_KEEP);\n}\n\n/**\n * Find cutoff index based on target token retention\n */\nasync function findTokenBasedCutoff(\n  messages: BaseMessage[],\n  keep: ContextSize,\n  tokenCounter: TokenCounter,\n  model: BaseLanguageModel\n): Promise<number | undefined> {\n  if (messages.length === 0) {\n    return 0;\n  }\n\n  let targetTokenCount: number;\n\n  if (\"fraction\" in keep && keep.fraction !== undefined) {\n    const maxInputTokens = getProfileLimits(model);\n    if (typeof maxInputTokens !== \"number\") {\n      return;\n    }\n    targetTokenCount = Math.floor(maxInputTokens * keep.fraction);\n  } else if (\"tokens\" in keep && keep.tokens !== undefined) {\n    targetTokenCount = Math.floor(keep.tokens);\n  } else {\n    return;\n  }\n\n  if (targetTokenCount <= 0) {\n    targetTokenCount = 1;\n  }\n\n  const totalTokens = await tokenCounter(messages);\n  if (totalTokens <= targetTokenCount) {\n    return 0;\n  }\n\n  /**\n   * Use binary search to identify the earliest message index that keeps the\n   * suffix within the token budget.\n   */\n  let left = 0;\n  let right = messages.length;\n  let cutoffCandidate = messages.length;\n  const maxIterations = Math.floor(Math.log2(messages.length)) + 1;\n\n  for (let i = 0; i < maxIterations; i++) {\n    if (left >= right) {\n      break;\n    }\n\n    const mid = Math.floor((left + right) / 2);\n    const suffixTokens = await tokenCounter(messages.slice(mid));\n    if (suffixTokens <= targetTokenCount) {\n      cutoffCandidate = mid;\n      right = mid;\n    } else {\n      left = mid + 1;\n    }\n  }\n\n  if (cutoffCandidate === messages.length) {\n    cutoffCandidate = left;\n  }\n\n  if (cutoffCandidate >= messages.length) {\n    if (messages.length === 1) {\n      return 0;\n    }\n    cutoffCandidate = messages.length - 1;\n  }\n\n  /**\n   * Find safe cutoff point that preserves tool pairs\n   */\n  for (let i = cutoffCandidate; i >= 0; i--) {\n    if (isSafeCutoffPoint(messages, i)) {\n      return i;\n    }\n  }\n\n  return 0;\n}\n\n/**\n * Find safe cutoff point that preserves AI/Tool message pairs\n */\nfunction findSafeCutoff(\n  messages: BaseMessage[],\n  messagesToKeep: number\n): number {\n  if (messages.length <= messagesToKeep) {\n    return 0;\n  }\n\n  const targetCutoff = messages.length - messagesToKeep;\n\n  for (let i = targetCutoff; i >= 0; i--) {\n    if (isSafeCutoffPoint(messages, i)) {\n      return i;\n    }\n  }\n\n  return 0;\n}\n\n/**\n * Check if cutting at index would separate AI/Tool message pairs\n */\nfunction isSafeCutoffPoint(\n  messages: BaseMessage[],\n  cutoffIndex: number\n): boolean {\n  if (cutoffIndex >= messages.length) {\n    return true;\n  }\n\n  /**\n   * Prevent preserved messages from starting with AI message containing tool calls\n   */\n  if (\n    cutoffIndex < messages.length &&\n    AIMessage.isInstance(messages[cutoffIndex]) &&\n    hasToolCalls(messages[cutoffIndex])\n  ) {\n    return false;\n  }\n\n  const searchStart = Math.max(0, cutoffIndex - SEARCH_RANGE_FOR_TOOL_PAIRS);\n  const searchEnd = Math.min(\n    messages.length,\n    cutoffIndex + SEARCH_RANGE_FOR_TOOL_PAIRS\n  );\n\n  for (let i = searchStart; i < searchEnd; i++) {\n    if (!hasToolCalls(messages[i])) {\n      continue;\n    }\n\n    const toolCallIds = extractToolCallIds(messages[i] as AIMessage);\n    if (cutoffSeparatesToolPair(messages, i, cutoffIndex, toolCallIds)) {\n      return false;\n    }\n  }\n\n  return true;\n}\n\n/**\n * Extract tool call IDs from an AI message\n */\nfunction extractToolCallIds(aiMessage: AIMessage): Set<string> {\n  const toolCallIds = new Set<string>();\n  if (aiMessage.tool_calls) {\n    for (const toolCall of aiMessage.tool_calls) {\n      const id =\n        typeof toolCall === \"object\" && \"id\" in toolCall ? toolCall.id : null;\n      if (id) {\n        toolCallIds.add(id);\n      }\n    }\n  }\n  return toolCallIds;\n}\n\n/**\n * Check if cutoff separates an AI message from its corresponding tool messages\n */\nfunction cutoffSeparatesToolPair(\n  messages: BaseMessage[],\n  aiMessageIndex: number,\n  cutoffIndex: number,\n  toolCallIds: Set<string>\n): boolean {\n  for (let j = aiMessageIndex + 1; j < messages.length; j++) {\n    const message = messages[j];\n    if (\n      ToolMessage.isInstance(message) &&\n      toolCallIds.has(message.tool_call_id)\n    ) {\n      const aiBeforeCutoff = aiMessageIndex < cutoffIndex;\n      const toolBeforeCutoff = j < cutoffIndex;\n      if (aiBeforeCutoff !== toolBeforeCutoff) {\n        return true;\n      }\n    }\n  }\n  return false;\n}\n\n/**\n * Generate summary for the given messages\n */\nasync function createSummary(\n  messagesToSummarize: BaseMessage[],\n  model: BaseLanguageModel,\n  summaryPrompt: string,\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined\n): Promise<string> {\n  if (!messagesToSummarize.length) {\n    return \"No previous conversation history.\";\n  }\n\n  const trimmedMessages = await trimMessagesForSummary(\n    messagesToSummarize,\n    tokenCounter,\n    trimTokensToSummarize\n  );\n\n  if (!trimmedMessages.length) {\n    return \"Previous conversation was too long to summarize.\";\n  }\n\n  try {\n    const formattedPrompt = summaryPrompt.replace(\n      \"{messages}\",\n      JSON.stringify(trimmedMessages, null, 2)\n    );\n    const response = await model.invoke(formattedPrompt);\n    const content = response.content;\n    /**\n     * Handle both string content and MessageContent array\n     */\n    if (typeof content === \"string\") {\n      return content.trim();\n    } else if (Array.isArray(content)) {\n      /**\n       * Extract text from MessageContent array\n       */\n      const textContent = content\n        .map((item) => {\n          if (typeof item === \"string\") return item;\n          if (typeof item === \"object\" && item !== null && \"text\" in item) {\n            return (item as { text: string }).text;\n          }\n          return \"\";\n        })\n        .join(\"\");\n      return textContent.trim();\n    }\n    return \"Error generating summary: Invalid response format\";\n  } catch (e) {\n    return `Error generating summary: ${e}`;\n  }\n}\n\n/**\n * Trim messages to fit within summary generation limits\n */\nasync function trimMessagesForSummary(\n  messages: BaseMessage[],\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined\n): Promise<BaseMessage[]> {\n  if (trimTokensToSummarize === undefined) {\n    return messages;\n  }\n\n  try {\n    return await trimMessages(messages, {\n      maxTokens: trimTokensToSummarize,\n      tokenCounter: async (msgs) => tokenCounter(msgs),\n      strategy: \"last\",\n      allowPartial: true,\n      includeSystem: true,\n    });\n  } catch {\n    /**\n     * Fallback to last N messages if trimming fails\n     */\n    return messages.slice(-DEFAULT_FALLBACK_MESSAGE_COUNT);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA2BA,MAAa,yBAAyB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;WA0B5B,CAAC;AAEZ,MAAM,2BAA2B;AACjC,MAAM,2BAA2B;AACjC,MAAM,iCAAiC;AACvC,MAAM,8BAA8B;AAEpC,MAAM,qBAAqB,EACxB,UAAU,CACV,KAAK,EAAE,MAAM,EAAE,QAAqB,CAAC,CAAC,CACtC,QAAQ,EAAE,MAAM,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,AAAC,EAAC,CAAC;AAKxD,MAAa,oBAAoB,EAC9B,OAAO;CAIN,UAAU,EACP,QAAQ,CACR,GAAG,GAAG,kCAAkC,CACxC,IAAI,GAAG,2CAA2C,CAClD,UAAU;CAIb,QAAQ,EAAE,QAAQ,CAAC,SAAS,gCAAgC,CAAC,UAAU;CAIvE,UAAU,EACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,SAAS,kCAAkC,CAC3C,UAAU;AACd,EAAC,CACD,OACC,CAAC,SAAS;CACR,MAAM,QAAQ;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAS,EAAC,OACxD,CAAC,MAAM,MAAM,OACd,CAAC;AACF,QAAO,SAAS;AACjB,GACD,EACE,SAAS,iEACV,EACF;AAGH,MAAa,aAAa,EACvB,OAAO;CAIN,UAAU,EACP,QAAQ,CACR,IAAI,GAAG,gCAAgC,CACvC,IAAI,GAAG,2CAA2C,CAClD,UAAU;CAIb,QAAQ,EACL,QAAQ,CACR,IAAI,GAAG,4CAA4C,CACnD,UAAU;CACb,UAAU,EACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,IAAI,GAAG,gCAAgC,CACvC,UAAU;AACd,EAAC,CACD,OACC,CAAC,SAAS;CACR,MAAM,QAAQ;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAS,EAAC,OACxD,CAAC,MAAM,MAAM,OACd,CAAC;AACF,QAAO,UAAU;AAClB,GACD,EACE,SAAS,gEACV,EACF;AAGH,MAAM,gBAAgB,EAAE,OAAO;CAI7B,OAAO,EAAE,QAAoC;CAiB7C,SAAS,EAAE,MAAM,CAAC,mBAAmB,EAAE,MAAM,kBAAkB,AAAC,EAAC,CAAC,UAAU;CAI5E,MAAM,WAAW,UAAU;CAI3B,cAAc,mBAAmB,UAAU;CAK3C,eAAe,EAAE,QAAQ,CAAC,QAAQ,uBAAuB;CAIzD,uBAAuB,EAAE,QAAQ,CAAC,UAAU;CAI5C,eAAe,EAAE,QAAQ,CAAC,UAAU;CAIpC,wBAAwB,EAAE,QAAQ,CAAC,UAAU;CAI7C,gBAAgB,EAAE,QAAQ,CAAC,UAAU;AACtC,EAAC;;;;AASF,SAAgB,iBAAiBA,OAA8C;AAE7E,KACE,aAAa,SACb,OAAO,MAAM,YAAY,YACzB,MAAM,WACN,oBAAoB,MAAM,YACzB,OAAO,MAAM,QAAQ,mBAAmB,YACvC,MAAM,QAAQ,kBAAkB,MAElC,QAAO,MAAM,QAAQ,kBAAkB;AAIzC,KAAI,WAAW,SAAS,OAAO,MAAM,UAAU,SAC7C,QAAO,oBAAoB,MAAM,MAAM;AAEzC,KAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,QAAO,oBAAoB,MAAM,UAAU;AAG7C,QAAO;AACR;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDD,SAAgB,wBACdC,SACA;;;;CAIA,MAAM,EAAE,MAAM,aAAa,OAAO,GAAG,iBAAiB,eAAe,QAAQ;AAC7E,KAAI,MACF,OAAM,IAAI,MACR,CAAC,0CAA0C,EAAEC,IAAG,cAAc,MAAM,EAAE;AAI1E,QAAO,iBAAiB;EACtB,MAAM;EACN,eAAe,cAAc,OAAO,EAKlC,OAAO,EAAE,QAA2B,CAAC,UAAU,CAChD,EAAC;EACF,aAAa,OAAO,OAAO,YAAY;GACrC,IAAIC,UACF,YAAY;GACd,IAAIC,OAAoB,YAAY;;;;AAOpC,OAAI,YAAY,2BAA2B,QAAW;IACpD,QAAQ,KACN,kFACD;AACD,QAAI,YAAY,QACd,UAAU,EAAE,QAAQ,YAAY,uBAAwB;GAE3D;;;;AAKD,OAAI,YAAY,mBAAmB,QAAW;IAC5C,QAAQ,KACN,yEACD;AACD,QACE,CAAC,QACA,QACC,cAAc,QACd,KAAK,aAAa,0BAEpB,OAAO,EAAE,UAAU,YAAY,eAAgB;GAElD;;;;GAKD,MAAM,kBACJ,QAAQ,SAAS,YAAY,SACzB,QAAQ,QAAQ,UAChB;GACN,MAAM,eACJ,QAAQ,SAAS,SAAS,SACtB,QAAQ,QAAQ,OAChB,QAAQ,EAAE,UAAU,yBAA0B;GAEpD,MAAM,gBAAgB,WAAW,MAAM,aAAa;;;;GAKpD,IAAIC,oBAAmC,CAAE;AACzC,OAAI,oBAAoB,QACtB,oBAAoB,CAAE;YACb,MAAM,QAAQ,gBAAgB;;;;GAIvC,oBAAqB,gBAAkC,IAAI,CAAC,MAC1D,kBAAkB,MAAM,EAAE,CAC3B;;;;;GAKD,oBAAoB,CAAC,kBAAkB,MAAM,gBAAgB,AAAC;;;;GAMhE,MAAM,kBACJ,kBAAkB,KAAK,CAAC,MAAM,cAAc,EAAE,IAC9C,cAAc;GAEhB,MAAM,QACJ,OAAO,YAAY,UAAU,WACzB,MAAM,cAAc,YAAY,MAAM,GACtC,YAAY;AAElB,OAAI,mBAAmB,CAAC,iBAAiB,MAAM,CAC7C,OAAM,IAAI,MACR;GAKJ,MAAM,gBACJ,QAAQ,SAAS,kBAAkB,yBAC/B,YAAY,iBAAiB,yBAC7B,QAAQ,SAAS,iBACjB,YAAY,iBACZ;GACN,MAAM,wBACJ,QAAQ,SAAS,0BAA0B,SACvC,QAAQ,QAAQ,wBAChB,YAAY,yBAAyB;;;;GAK3C,iBAAiB,MAAM,SAAS;GAEhC,MAAM,eACJ,QAAQ,SAAS,iBAAiB,SAC9B,QAAQ,QAAQ,eAChB,YAAY,gBAAgB;GAClC,MAAM,cAAc,MAAM,aAAa,MAAM,SAAS;GACtD,MAAM,cAAc,MAAM,gBACxB,MAAM,UACN,aACA,mBACA,MACD;AAED,OAAI,CAAC,YACH;GAGF,MAAM,EAAE,cAAc,sBAAsB,GAAG,mBAC7C,MAAM,SACP;GACD,MAAM,cAAc,MAAM,qBACxB,sBACA,eACA,cACA,MACD;AAED,OAAI,eAAe,EACjB;GAGF,MAAM,EAAE,qBAAqB,mBAAmB,GAAG,kBACjD,cACA,sBACA,YACD;GAED,MAAM,UAAU,MAAM,cACpB,qBACA,OACA,eACA,cACA,sBACD;GAED,MAAM,iBAAiB,IAAI,aAAa;IACtC,SAAS,CAAC,kDAAkD,EAAE,SAAS;IACvE,IAAIC,IAAM;GACX;AAED,UAAO,EACL,UAAU;IACR,IAAI,cAAc,EAAE,IAAI,oBAAqB;IAC7C;IACA,GAAG;GACJ,EACF;EACF;CACF,EAAC;AACH;;;;AAKD,SAAS,iBAAiBC,UAA+B;AACvD,MAAK,MAAM,OAAO,SAChB,KAAI,CAAC,IAAI,IACP,IAAI,KAAKD,IAAM;AAGpB;;;;AAKD,SAAS,mBAAmBC,UAG1B;AACA,KAAI,SAAS,SAAS,KAAK,cAAc,WAAW,SAAS,GAAG,CAC9D,QAAO;EACL,cAAc,SAAS;EACvB,sBAAsB,SAAS,MAAM,EAAE;CACxC;AAEH,QAAO,EACL,sBAAsB,SACvB;AACF;;;;AAKD,SAAS,kBACPC,cACAC,sBACAC,aAC0E;CAC1E,MAAM,sBAAsB,qBAAqB,MAAM,GAAG,YAAY;CACtE,MAAM,oBAAoB,qBAAqB,MAAM,YAAY;AAGjE,KAAI,cACF,oBAAoB,QAAQ,aAAa;AAG3C,QAAO;EAAE;EAAqB;CAAmB;AAClD;;;;;;;;;;;AAYD,eAAe,gBACbH,UACAI,aACAN,mBACAO,OACkB;AAClB,KAAI,kBAAkB,WAAW,EAC/B,QAAO;;;;AAMT,MAAK,MAAM,WAAW,mBAAmB;;;;EAIvC,IAAI,eAAe;EACnB,IAAI,iBAAiB;AAErB,MAAI,QAAQ,aAAa,QAAW;GAClC,iBAAiB;AACjB,OAAI,SAAS,SAAS,QAAQ,UAC5B,eAAe;EAElB;AAED,MAAI,QAAQ,WAAW,QAAW;GAChC,iBAAiB;AACjB,OAAI,cAAc,QAAQ,QACxB,eAAe;EAElB;AAED,MAAI,QAAQ,aAAa,QAAW;GAClC,iBAAiB;GACjB,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,OAAI,OAAO,mBAAmB,UAAU;IACtC,MAAM,YAAY,KAAK,MAAM,iBAAiB,QAAQ,SAAS;AAC/D,QAAI,cAAc,WAChB,eAAe;GAElB;;;;GAIC,eAAe;EAElB;;;;AAKD,MAAI,kBAAkB,aACpB,QAAO;CAEV;AAED,QAAO;AACR;;;;AAKD,eAAe,qBACbL,UACAH,MACAS,cACAD,OACiB;AACjB,KAAI,YAAY,QAAQ,cAAc,MAAM;EAC1C,MAAM,mBAAmB,MAAM,qBAC7B,UACA,MACA,cACA,MACD;AACD,MAAI,OAAO,qBAAqB,SAC9B,QAAO;;;;AAKT,SAAO,eAAe,UAAU,yBAAyB;CAC1D;;;;AAID,QAAO,eAAe,UAAU,KAAK,YAAY,yBAAyB;AAC3E;;;;AAKD,eAAe,qBACbL,UACAH,MACAS,cACAD,OAC6B;AAC7B,KAAI,SAAS,WAAW,EACtB,QAAO;CAGT,IAAIE;AAEJ,KAAI,cAAc,QAAQ,KAAK,aAAa,QAAW;EACrD,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,MAAI,OAAO,mBAAmB,SAC5B;EAEF,mBAAmB,KAAK,MAAM,iBAAiB,KAAK,SAAS;CAC9D,WAAU,YAAY,QAAQ,KAAK,WAAW,QAC7C,mBAAmB,KAAK,MAAM,KAAK,OAAO;KAE1C;AAGF,KAAI,oBAAoB,GACtB,mBAAmB;CAGrB,MAAM,cAAc,MAAM,aAAa,SAAS;AAChD,KAAI,eAAe,iBACjB,QAAO;;;;;CAOT,IAAI,OAAO;CACX,IAAI,QAAQ,SAAS;CACrB,IAAI,kBAAkB,SAAS;CAC/B,MAAM,gBAAgB,KAAK,MAAM,KAAK,KAAK,SAAS,OAAO,CAAC,GAAG;AAE/D,MAAK,IAAI,IAAI,GAAG,IAAI,eAAe,KAAK;AACtC,MAAI,QAAQ,MACV;EAGF,MAAM,MAAM,KAAK,OAAO,OAAO,SAAS,EAAE;EAC1C,MAAM,eAAe,MAAM,aAAa,SAAS,MAAM,IAAI,CAAC;AAC5D,MAAI,gBAAgB,kBAAkB;GACpC,kBAAkB;GAClB,QAAQ;EACT,OACC,OAAO,MAAM;CAEhB;AAED,KAAI,oBAAoB,SAAS,QAC/B,kBAAkB;AAGpB,KAAI,mBAAmB,SAAS,QAAQ;AACtC,MAAI,SAAS,WAAW,EACtB,QAAO;EAET,kBAAkB,SAAS,SAAS;CACrC;;;;AAKD,MAAK,IAAI,IAAI,iBAAiB,KAAK,GAAG,IACpC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;AACR;;;;AAKD,SAAS,eACPP,UACAQ,gBACQ;AACR,KAAI,SAAS,UAAU,eACrB,QAAO;CAGT,MAAM,eAAe,SAAS,SAAS;AAEvC,MAAK,IAAI,IAAI,cAAc,KAAK,GAAG,IACjC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;AACR;;;;AAKD,SAAS,kBACPR,UACAG,aACS;AACT,KAAI,eAAe,SAAS,OAC1B,QAAO;;;;AAMT,KACE,cAAc,SAAS,UACvB,UAAU,WAAW,SAAS,aAAa,IAC3C,aAAa,SAAS,aAAa,CAEnC,QAAO;CAGT,MAAM,cAAc,KAAK,IAAI,GAAG,cAAc,4BAA4B;CAC1E,MAAM,YAAY,KAAK,IACrB,SAAS,QACT,cAAc,4BACf;AAED,MAAK,IAAI,IAAI,aAAa,IAAI,WAAW,KAAK;AAC5C,MAAI,CAAC,aAAa,SAAS,GAAG,CAC5B;EAGF,MAAM,cAAc,mBAAmB,SAAS,GAAgB;AAChE,MAAI,wBAAwB,UAAU,GAAG,aAAa,YAAY,CAChE,QAAO;CAEV;AAED,QAAO;AACR;;;;AAKD,SAAS,mBAAmBM,WAAmC;CAC7D,MAAM,8BAAc,IAAI;AACxB,KAAI,UAAU,WACZ,MAAK,MAAM,YAAY,UAAU,YAAY;EAC3C,MAAM,KACJ,OAAO,aAAa,YAAY,QAAQ,WAAW,SAAS,KAAK;AACnE,MAAI,IACF,YAAY,IAAI,GAAG;CAEtB;AAEH,QAAO;AACR;;;;AAKD,SAAS,wBACPT,UACAU,gBACAP,aACAQ,aACS;AACT,MAAK,IAAI,IAAI,iBAAiB,GAAG,IAAI,SAAS,QAAQ,KAAK;EACzD,MAAM,UAAU,SAAS;AACzB,MACE,YAAY,WAAW,QAAQ,IAC/B,YAAY,IAAI,QAAQ,aAAa,EACrC;GACA,MAAM,iBAAiB,iBAAiB;GACxC,MAAM,mBAAmB,IAAI;AAC7B,OAAI,mBAAmB,iBACrB,QAAO;EAEV;CACF;AACD,QAAO;AACR;;;;AAKD,eAAe,cACbC,qBACAP,OACAQ,eACAP,cACAQ,uBACiB;AACjB,KAAI,CAAC,oBAAoB,OACvB,QAAO;CAGT,MAAM,kBAAkB,MAAM,uBAC5B,qBACA,cACA,sBACD;AAED,KAAI,CAAC,gBAAgB,OACnB,QAAO;AAGT,KAAI;EACF,MAAM,kBAAkB,cAAc,QACpC,cACA,KAAK,UAAU,iBAAiB,MAAM,EAAE,CACzC;EACD,MAAM,WAAW,MAAM,MAAM,OAAO,gBAAgB;EACpD,MAAM,UAAU,SAAS;;;;AAIzB,MAAI,OAAO,YAAY,SACrB,QAAO,QAAQ,MAAM;WACZ,MAAM,QAAQ,QAAQ,EAAE;;;;GAIjC,MAAM,cAAc,QACjB,IAAI,CAAC,SAAS;AACb,QAAI,OAAO,SAAS,SAAU,QAAO;AACrC,QAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,KACzD,QAAQ,KAA0B;AAEpC,WAAO;GACR,EAAC,CACD,KAAK,GAAG;AACX,UAAO,YAAY,MAAM;EAC1B;AACD,SAAO;CACR,SAAQ,GAAG;AACV,SAAO,CAAC,0BAA0B,EAAE,GAAG;CACxC;AACF;;;;AAKD,eAAe,uBACbd,UACAM,cACAQ,uBACwB;AACxB,KAAI,0BAA0B,OAC5B,QAAO;AAGT,KAAI;AACF,SAAO,MAAM,aAAa,UAAU;GAClC,WAAW;GACX,cAAc,OAAO,SAAS,aAAa,KAAK;GAChD,UAAU;GACV,cAAc;GACd,eAAe;EAChB,EAAC;CACH,QAAO;;;;AAIN,SAAO,SAAS,MAAM,CAAC,+BAA+B;CACvD;AACF"}
+{"version":3,"file":"summarization.js","names":["input: BaseLanguageModel","options: SummarizationMiddlewareConfig","z4","trigger: ContextSize | ContextSize[] | undefined","keep: ContextSize","triggerConditions: ContextSize[]","uuid","messages: BaseMessage[]","systemPrompt: SystemMessage | undefined","conversationMessages: BaseMessage[]","cutoffIndex: number","totalTokens: number","model: BaseLanguageModel","tokenCounter: TokenCounter","targetTokenCount: number","messagesToKeep: number","aiMessage: AIMessage","aiMessageIndex: number","toolCallIds: Set<string>","messagesToSummarize: BaseMessage[]","summaryPrompt: string","trimTokensToSummarize: number | undefined"],"sources":["../../../src/agents/middleware/summarization.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { z as z4 } from \"zod/v4\";\nimport { v4 as uuid } from \"uuid\";\nimport {\n  BaseMessage,\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n  RemoveMessage,\n  trimMessages,\n  HumanMessage,\n} from \"@langchain/core/messages\";\nimport {\n  BaseLanguageModel,\n  getModelContextSize,\n} from \"@langchain/core/language_models/base\";\nimport {\n  interopSafeParse,\n  InferInteropZodInput,\n  InferInteropZodOutput,\n} from \"@langchain/core/utils/types\";\nimport { REMOVE_ALL_MESSAGES } from \"@langchain/langgraph\";\nimport { createMiddleware } from \"../middleware.js\";\nimport { countTokensApproximately } from \"./utils.js\";\nimport { hasToolCalls } from \"../utils.js\";\nimport { initChatModel } from \"../../chat_models/universal.js\";\n\nexport const DEFAULT_SUMMARY_PROMPT = `<role>\nContext Extraction Assistant\n</role>\n\n<primary_objective>\nYour sole objective in this task is to extract the highest quality/most relevant context from the conversation history below.\n</primary_objective>\n\n<objective_information>\nYou're nearing the total number of input tokens you can accept, so you must extract the highest quality/most relevant pieces of information from your conversation history.\nThis context will then overwrite the conversation history presented below. Because of this, ensure the context you extract is only the most important information to your overall goal.\n</objective_information>\n\n<instructions>\nThe conversation history below will be replaced with the context you extract in this step. Because of this, you must do your very best to extract and record all of the most important context from the conversation history.\nYou want to ensure that you don't repeat any actions you've already completed, so the context you extract from the conversation history should be focused on the most important information to your overall goal.\n</instructions>\n\nThe user will message you with the full message history you'll be extracting context from, to then replace. Carefully read over it all, and think deeply about what information is most important to your overall goal that should be saved:\n\nWith all of this in mind, please carefully read over the entire conversation history, and extract the most important and relevant context to replace it so that you can free up space in the conversation history.\nRespond ONLY with the extracted context. Do not include any additional information, or text before or after the extracted context.\n\n<messages>\nMessages to summarize:\n{messages}\n</messages>`;\n\nconst DEFAULT_SUMMARY_PREFIX = \"Here is a summary of the conversation to date:\";\nconst DEFAULT_MESSAGES_TO_KEEP = 20;\nconst DEFAULT_TRIM_TOKEN_LIMIT = 4000;\nconst DEFAULT_FALLBACK_MESSAGE_COUNT = 15;\nconst SEARCH_RANGE_FOR_TOOL_PAIRS = 5;\n\nconst tokenCounterSchema = z\n  .function()\n  .args(z.array(z.custom<BaseMessage>()))\n  .returns(z.union([z.number(), z.promise(z.number())]));\nexport type TokenCounter = (\n  messages: BaseMessage[]\n) => number | Promise<number>;\n\nexport const contextSizeSchema = z\n  .object({\n    /**\n     * Fraction of the model's context size to use as the trigger\n     */\n    fraction: z\n      .number()\n      .gt(0, \"Fraction must be greater than 0\")\n      .max(1, \"Fraction must be less than or equal to 1\")\n      .optional(),\n    /**\n     * Number of tokens to use as the trigger\n     */\n    tokens: z.number().positive(\"Tokens must be greater than 0\").optional(),\n    /**\n     * Number of messages to use as the trigger\n     */\n    messages: z\n      .number()\n      .int(\"Messages must be an integer\")\n      .positive(\"Messages must be greater than 0\")\n      .optional(),\n  })\n  .refine(\n    (data) => {\n      const count = [data.fraction, data.tokens, data.messages].filter(\n        (v) => v !== undefined\n      ).length;\n      return count >= 1;\n    },\n    {\n      message: \"At least one of fraction, tokens, or messages must be provided\",\n    }\n  );\nexport type ContextSize = z.infer<typeof contextSizeSchema>;\n\nexport const keepSchema = z\n  .object({\n    /**\n     * Fraction of the model's context size to keep\n     */\n    fraction: z\n      .number()\n      .min(0, \"Messages must be non-negative\")\n      .max(1, \"Fraction must be less than or equal to 1\")\n      .optional(),\n    /**\n     * Number of tokens to keep\n     */\n    tokens: z\n      .number()\n      .min(0, \"Tokens must be greater than or equal to 0\")\n      .optional(),\n    messages: z\n      .number()\n      .int(\"Messages must be an integer\")\n      .min(0, \"Messages must be non-negative\")\n      .optional(),\n  })\n  .refine(\n    (data) => {\n      const count = [data.fraction, data.tokens, data.messages].filter(\n        (v) => v !== undefined\n      ).length;\n      return count === 1;\n    },\n    {\n      message: \"Exactly one of fraction, tokens, or messages must be provided\",\n    }\n  );\nexport type KeepSize = z.infer<typeof keepSchema>;\n\nconst contextSchema = z.object({\n  /**\n   * Model to use for summarization\n   */\n  model: z.custom<string | BaseLanguageModel>(),\n  /**\n   * Trigger conditions for summarization.\n   * Can be a single condition object (all properties must be met) or an array of conditions (any condition must be met).\n   *\n   * @example\n   * ```ts\n   * // Single condition: trigger if tokens >= 5000 AND messages >= 3\n   * trigger: { tokens: 5000, messages: 3 }\n   *\n   * // Multiple conditions: trigger if (tokens >= 5000 AND messages >= 3) OR (tokens >= 3000 AND messages >= 6)\n   * trigger: [\n   *   { tokens: 5000, messages: 3 },\n   *   { tokens: 3000, messages: 6 }\n   * ]\n   * ```\n   */\n  trigger: z.union([contextSizeSchema, z.array(contextSizeSchema)]).optional(),\n  /**\n   * Keep conditions for summarization\n   */\n  keep: keepSchema.optional(),\n  /**\n   * Token counter function to use for summarization\n   */\n  tokenCounter: tokenCounterSchema.optional(),\n  /**\n   * Summary prompt to use for summarization\n   * @default {@link DEFAULT_SUMMARY_PROMPT}\n   */\n  summaryPrompt: z.string().default(DEFAULT_SUMMARY_PROMPT),\n  /**\n   * Number of tokens to trim to before summarizing\n   */\n  trimTokensToSummarize: z.number().optional(),\n  /**\n   * Prefix to add to the summary\n   */\n  summaryPrefix: z.string().optional(),\n  /**\n   * @deprecated Use `trigger: { tokens: value }` instead.\n   */\n  maxTokensBeforeSummary: z.number().optional(),\n  /**\n   * @deprecated Use `keep: { messages: value }` instead.\n   */\n  messagesToKeep: z.number().optional(),\n});\n\nexport type SummarizationMiddlewareConfig = InferInteropZodInput<\n  typeof contextSchema\n>;\n\n/**\n * Get max input tokens from model profile or fallback to model name lookup\n */\nexport function getProfileLimits(input: BaseLanguageModel): number | undefined {\n  // Access maxInputTokens on the model profile directly if available\n  if (\n    \"profile\" in input &&\n    typeof input.profile === \"object\" &&\n    input.profile &&\n    \"maxInputTokens\" in input.profile &&\n    (typeof input.profile.maxInputTokens === \"number\" ||\n      input.profile.maxInputTokens == null)\n  ) {\n    return input.profile.maxInputTokens ?? undefined;\n  }\n\n  // Fallback to using model name if available\n  if (\"model\" in input && typeof input.model === \"string\") {\n    return getModelContextSize(input.model);\n  }\n  if (\"modelName\" in input && typeof input.modelName === \"string\") {\n    return getModelContextSize(input.modelName);\n  }\n\n  return undefined;\n}\n\n/**\n * Summarization middleware that automatically summarizes conversation history when token limits are approached.\n *\n * This middleware monitors message token counts and automatically summarizes older\n * messages when a threshold is reached, preserving recent messages and maintaining\n * context continuity by ensuring AI/Tool message pairs remain together.\n *\n * @param options Configuration options for the summarization middleware\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * import { summarizationMiddleware } from \"langchain\";\n * import { createAgent } from \"langchain\";\n *\n * // Single condition: trigger if tokens >= 4000 AND messages >= 10\n * const agent1 = createAgent({\n *   llm: model,\n *   tools: [getWeather],\n *   middleware: [\n *     summarizationMiddleware({\n *       model: new ChatOpenAI({ model: \"gpt-4o\" }),\n *       trigger: { tokens: 4000, messages: 10 },\n *       keep: { messages: 20 },\n *     })\n *   ],\n * });\n *\n * // Multiple conditions: trigger if (tokens >= 5000 AND messages >= 3) OR (tokens >= 3000 AND messages >= 6)\n * const agent2 = createAgent({\n *   llm: model,\n *   tools: [getWeather],\n *   middleware: [\n *     summarizationMiddleware({\n *       model: new ChatOpenAI({ model: \"gpt-4o\" }),\n *       trigger: [\n *         { tokens: 5000, messages: 3 },\n *         { tokens: 3000, messages: 6 },\n *       ],\n *       keep: { messages: 20 },\n *     })\n *   ],\n * });\n *\n * ```\n */\nexport function summarizationMiddleware(\n  options: SummarizationMiddlewareConfig\n) {\n  /**\n   * Parse user options to get their explicit values\n   */\n  const { data: userOptions, error } = interopSafeParse(contextSchema, options);\n  if (error) {\n    throw new Error(\n      `Invalid summarization middleware options: ${z4.prettifyError(error)}`\n    );\n  }\n\n  return createMiddleware({\n    name: \"SummarizationMiddleware\",\n    contextSchema: contextSchema.extend({\n      /**\n       * `model` should be required when initializing the middleware,\n       * but can be omitted within context when invoking the middleware.\n       */\n      model: z.custom<BaseLanguageModel>().optional(),\n    }),\n    beforeModel: async (state, runtime) => {\n      let trigger: ContextSize | ContextSize[] | undefined =\n        userOptions.trigger;\n      let keep: ContextSize = userOptions.keep as InferInteropZodOutput<\n        typeof keepSchema\n      >;\n\n      /**\n       * Handle deprecated parameters\n       */\n      if (userOptions.maxTokensBeforeSummary !== undefined) {\n        console.warn(\n          \"maxTokensBeforeSummary is deprecated. Use `trigger: { tokens: value }` instead.\"\n        );\n        if (trigger === undefined) {\n          trigger = { tokens: userOptions.maxTokensBeforeSummary };\n        }\n      }\n\n      /**\n       * Handle deprecated parameters\n       */\n      if (userOptions.messagesToKeep !== undefined) {\n        console.warn(\n          \"messagesToKeep is deprecated. Use `keep: { messages: value }` instead.\"\n        );\n        if (\n          !keep ||\n          (keep &&\n            \"messages\" in keep &&\n            keep.messages === DEFAULT_MESSAGES_TO_KEEP)\n        ) {\n          keep = { messages: userOptions.messagesToKeep };\n        }\n      }\n\n      /**\n       * Merge context with user options\n       */\n      const resolvedTrigger =\n        runtime.context?.trigger !== undefined\n          ? runtime.context.trigger\n          : trigger;\n      const resolvedKeep =\n        runtime.context?.keep !== undefined\n          ? runtime.context.keep\n          : keep ?? { messages: DEFAULT_MESSAGES_TO_KEEP };\n\n      const validatedKeep = keepSchema.parse(resolvedKeep);\n\n      /**\n       * Validate trigger conditions\n       */\n      let triggerConditions: ContextSize[] = [];\n      if (resolvedTrigger === undefined) {\n        triggerConditions = [];\n      } else if (Array.isArray(resolvedTrigger)) {\n        /**\n         * It's an array of ContextSize objects\n         */\n        triggerConditions = (resolvedTrigger as ContextSize[]).map((t) =>\n          contextSizeSchema.parse(t)\n        );\n      } else {\n        /**\n         * Single ContextSize object - all properties must be satisfied (AND logic)\n         */\n        triggerConditions = [contextSizeSchema.parse(resolvedTrigger)];\n      }\n\n      /**\n       * Check if profile is required\n       */\n      const requiresProfile =\n        triggerConditions.some((c) => \"fraction\" in c) ||\n        \"fraction\" in validatedKeep;\n\n      const model =\n        typeof userOptions.model === \"string\"\n          ? await initChatModel(userOptions.model)\n          : userOptions.model;\n\n      if (requiresProfile && !getProfileLimits(model)) {\n        throw new Error(\n          \"Model profile information is required to use fractional token limits. \" +\n            \"Use absolute token counts instead.\"\n        );\n      }\n\n      const summaryPrompt =\n        runtime.context?.summaryPrompt === DEFAULT_SUMMARY_PROMPT\n          ? userOptions.summaryPrompt ?? DEFAULT_SUMMARY_PROMPT\n          : runtime.context?.summaryPrompt ??\n            userOptions.summaryPrompt ??\n            DEFAULT_SUMMARY_PROMPT;\n      const summaryPrefix =\n        runtime.context.summaryPrefix ??\n        userOptions.summaryPrefix ??\n        DEFAULT_SUMMARY_PREFIX;\n      const trimTokensToSummarize =\n        runtime.context?.trimTokensToSummarize !== undefined\n          ? runtime.context.trimTokensToSummarize\n          : userOptions.trimTokensToSummarize ?? DEFAULT_TRIM_TOKEN_LIMIT;\n\n      /**\n       * Ensure all messages have IDs\n       */\n      ensureMessageIds(state.messages);\n\n      const tokenCounter =\n        runtime.context?.tokenCounter !== undefined\n          ? runtime.context.tokenCounter\n          : userOptions.tokenCounter ?? countTokensApproximately;\n      const totalTokens = await tokenCounter(state.messages);\n      const doSummarize = await shouldSummarize(\n        state.messages,\n        totalTokens,\n        triggerConditions,\n        model\n      );\n\n      if (!doSummarize) {\n        return;\n      }\n\n      const { systemPrompt, conversationMessages } = splitSystemMessage(\n        state.messages\n      );\n      const cutoffIndex = await determineCutoffIndex(\n        conversationMessages,\n        validatedKeep,\n        tokenCounter,\n        model\n      );\n\n      if (cutoffIndex <= 0) {\n        return;\n      }\n\n      const { messagesToSummarize, preservedMessages } = partitionMessages(\n        systemPrompt,\n        conversationMessages,\n        cutoffIndex\n      );\n\n      const summary = await createSummary(\n        messagesToSummarize,\n        model,\n        summaryPrompt,\n        tokenCounter,\n        trimTokensToSummarize\n      );\n\n      const summaryMessage = new HumanMessage({\n        content: `${summaryPrefix}\\n\\n${summary}`,\n        id: uuid(),\n      });\n\n      return {\n        messages: [\n          new RemoveMessage({ id: REMOVE_ALL_MESSAGES }),\n          summaryMessage,\n          ...preservedMessages,\n        ],\n      };\n    },\n  });\n}\n\n/**\n * Ensure all messages have unique IDs\n */\nfunction ensureMessageIds(messages: BaseMessage[]): void {\n  for (const msg of messages) {\n    if (!msg.id) {\n      msg.id = uuid();\n    }\n  }\n}\n\n/**\n * Separate system message from conversation messages\n */\nfunction splitSystemMessage(messages: BaseMessage[]): {\n  systemPrompt?: SystemMessage;\n  conversationMessages: BaseMessage[];\n} {\n  if (messages.length > 0 && SystemMessage.isInstance(messages[0])) {\n    return {\n      systemPrompt: messages[0] as SystemMessage,\n      conversationMessages: messages.slice(1),\n    };\n  }\n  return {\n    conversationMessages: messages,\n  };\n}\n\n/**\n * Partition messages into those to summarize and those to preserve\n */\nfunction partitionMessages(\n  systemPrompt: SystemMessage | undefined,\n  conversationMessages: BaseMessage[],\n  cutoffIndex: number\n): { messagesToSummarize: BaseMessage[]; preservedMessages: BaseMessage[] } {\n  const messagesToSummarize = conversationMessages.slice(0, cutoffIndex);\n  const preservedMessages = conversationMessages.slice(cutoffIndex);\n\n  // Include system message in messages to summarize to capture previous summaries\n  if (systemPrompt) {\n    messagesToSummarize.unshift(systemPrompt);\n  }\n\n  return { messagesToSummarize, preservedMessages };\n}\n\n/**\n * Determine whether summarization should run for the current token usage\n *\n * @param messages - Current messages in the conversation\n * @param totalTokens - Total token count for all messages\n * @param triggerConditions - Array of trigger conditions. Returns true if ANY condition is satisfied (OR logic).\n *                           Within each condition, ALL specified properties must be satisfied (AND logic).\n * @param model - The language model being used\n * @returns true if summarization should be triggered\n */\nasync function shouldSummarize(\n  messages: BaseMessage[],\n  totalTokens: number,\n  triggerConditions: ContextSize[],\n  model: BaseLanguageModel\n): Promise<boolean> {\n  if (triggerConditions.length === 0) {\n    return false;\n  }\n\n  /**\n   * Check each condition (OR logic between conditions)\n   */\n  for (const trigger of triggerConditions) {\n    /**\n     * Within a single condition, all specified properties must be satisfied (AND logic)\n     */\n    let conditionMet = true;\n    let hasAnyProperty = false;\n\n    if (trigger.messages !== undefined) {\n      hasAnyProperty = true;\n      if (messages.length < trigger.messages) {\n        conditionMet = false;\n      }\n    }\n\n    if (trigger.tokens !== undefined) {\n      hasAnyProperty = true;\n      if (totalTokens < trigger.tokens) {\n        conditionMet = false;\n      }\n    }\n\n    if (trigger.fraction !== undefined) {\n      hasAnyProperty = true;\n      const maxInputTokens = getProfileLimits(model);\n      if (typeof maxInputTokens === \"number\") {\n        const threshold = Math.floor(maxInputTokens * trigger.fraction);\n        if (totalTokens < threshold) {\n          conditionMet = false;\n        }\n      } else {\n        /**\n         * If fraction is specified but we can't get model limits, skip this condition\n         */\n        conditionMet = false;\n      }\n    }\n\n    /**\n     * If condition has at least one property and all properties are satisfied, trigger summarization\n     */\n    if (hasAnyProperty && conditionMet) {\n      return true;\n    }\n  }\n\n  return false;\n}\n\n/**\n * Determine cutoff index respecting retention configuration\n */\nasync function determineCutoffIndex(\n  messages: BaseMessage[],\n  keep: ContextSize,\n  tokenCounter: TokenCounter,\n  model: BaseLanguageModel\n): Promise<number> {\n  if (\"tokens\" in keep || \"fraction\" in keep) {\n    const tokenBasedCutoff = await findTokenBasedCutoff(\n      messages,\n      keep,\n      tokenCounter,\n      model\n    );\n    if (typeof tokenBasedCutoff === \"number\") {\n      return tokenBasedCutoff;\n    }\n    /**\n     * Fallback to message count if token-based fails\n     */\n    return findSafeCutoff(messages, DEFAULT_MESSAGES_TO_KEEP);\n  }\n  /**\n   * find cutoff index based on message count\n   */\n  return findSafeCutoff(messages, keep.messages ?? DEFAULT_MESSAGES_TO_KEEP);\n}\n\n/**\n * Find cutoff index based on target token retention\n */\nasync function findTokenBasedCutoff(\n  messages: BaseMessage[],\n  keep: ContextSize,\n  tokenCounter: TokenCounter,\n  model: BaseLanguageModel\n): Promise<number | undefined> {\n  if (messages.length === 0) {\n    return 0;\n  }\n\n  let targetTokenCount: number;\n\n  if (\"fraction\" in keep && keep.fraction !== undefined) {\n    const maxInputTokens = getProfileLimits(model);\n    if (typeof maxInputTokens !== \"number\") {\n      return;\n    }\n    targetTokenCount = Math.floor(maxInputTokens * keep.fraction);\n  } else if (\"tokens\" in keep && keep.tokens !== undefined) {\n    targetTokenCount = Math.floor(keep.tokens);\n  } else {\n    return;\n  }\n\n  if (targetTokenCount <= 0) {\n    targetTokenCount = 1;\n  }\n\n  const totalTokens = await tokenCounter(messages);\n  if (totalTokens <= targetTokenCount) {\n    return 0;\n  }\n\n  /**\n   * Use binary search to identify the earliest message index that keeps the\n   * suffix within the token budget.\n   */\n  let left = 0;\n  let right = messages.length;\n  let cutoffCandidate = messages.length;\n  const maxIterations = Math.floor(Math.log2(messages.length)) + 1;\n\n  for (let i = 0; i < maxIterations; i++) {\n    if (left >= right) {\n      break;\n    }\n\n    const mid = Math.floor((left + right) / 2);\n    const suffixTokens = await tokenCounter(messages.slice(mid));\n    if (suffixTokens <= targetTokenCount) {\n      cutoffCandidate = mid;\n      right = mid;\n    } else {\n      left = mid + 1;\n    }\n  }\n\n  if (cutoffCandidate === messages.length) {\n    cutoffCandidate = left;\n  }\n\n  if (cutoffCandidate >= messages.length) {\n    if (messages.length === 1) {\n      return 0;\n    }\n    cutoffCandidate = messages.length - 1;\n  }\n\n  /**\n   * Find safe cutoff point that preserves tool pairs\n   */\n  for (let i = cutoffCandidate; i >= 0; i--) {\n    if (isSafeCutoffPoint(messages, i)) {\n      return i;\n    }\n  }\n\n  return 0;\n}\n\n/**\n * Find safe cutoff point that preserves AI/Tool message pairs\n */\nfunction findSafeCutoff(\n  messages: BaseMessage[],\n  messagesToKeep: number\n): number {\n  if (messages.length <= messagesToKeep) {\n    return 0;\n  }\n\n  const targetCutoff = messages.length - messagesToKeep;\n\n  for (let i = targetCutoff; i >= 0; i--) {\n    if (isSafeCutoffPoint(messages, i)) {\n      return i;\n    }\n  }\n\n  return 0;\n}\n\n/**\n * Check if cutting at index would separate AI/Tool message pairs\n */\nfunction isSafeCutoffPoint(\n  messages: BaseMessage[],\n  cutoffIndex: number\n): boolean {\n  if (cutoffIndex >= messages.length) {\n    return true;\n  }\n\n  /**\n   * Prevent preserved messages from starting with AI message containing tool calls\n   */\n  if (\n    cutoffIndex < messages.length &&\n    AIMessage.isInstance(messages[cutoffIndex]) &&\n    hasToolCalls(messages[cutoffIndex])\n  ) {\n    return false;\n  }\n\n  const searchStart = Math.max(0, cutoffIndex - SEARCH_RANGE_FOR_TOOL_PAIRS);\n  const searchEnd = Math.min(\n    messages.length,\n    cutoffIndex + SEARCH_RANGE_FOR_TOOL_PAIRS\n  );\n\n  for (let i = searchStart; i < searchEnd; i++) {\n    if (!hasToolCalls(messages[i])) {\n      continue;\n    }\n\n    const toolCallIds = extractToolCallIds(messages[i] as AIMessage);\n    if (cutoffSeparatesToolPair(messages, i, cutoffIndex, toolCallIds)) {\n      return false;\n    }\n  }\n\n  return true;\n}\n\n/**\n * Extract tool call IDs from an AI message\n */\nfunction extractToolCallIds(aiMessage: AIMessage): Set<string> {\n  const toolCallIds = new Set<string>();\n  if (aiMessage.tool_calls) {\n    for (const toolCall of aiMessage.tool_calls) {\n      const id =\n        typeof toolCall === \"object\" && \"id\" in toolCall ? toolCall.id : null;\n      if (id) {\n        toolCallIds.add(id);\n      }\n    }\n  }\n  return toolCallIds;\n}\n\n/**\n * Check if cutoff separates an AI message from its corresponding tool messages\n */\nfunction cutoffSeparatesToolPair(\n  messages: BaseMessage[],\n  aiMessageIndex: number,\n  cutoffIndex: number,\n  toolCallIds: Set<string>\n): boolean {\n  for (let j = aiMessageIndex + 1; j < messages.length; j++) {\n    const message = messages[j];\n    if (\n      ToolMessage.isInstance(message) &&\n      toolCallIds.has(message.tool_call_id)\n    ) {\n      const aiBeforeCutoff = aiMessageIndex < cutoffIndex;\n      const toolBeforeCutoff = j < cutoffIndex;\n      if (aiBeforeCutoff !== toolBeforeCutoff) {\n        return true;\n      }\n    }\n  }\n  return false;\n}\n\n/**\n * Generate summary for the given messages\n */\nasync function createSummary(\n  messagesToSummarize: BaseMessage[],\n  model: BaseLanguageModel,\n  summaryPrompt: string,\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined\n): Promise<string> {\n  if (!messagesToSummarize.length) {\n    return \"No previous conversation history.\";\n  }\n\n  const trimmedMessages = await trimMessagesForSummary(\n    messagesToSummarize,\n    tokenCounter,\n    trimTokensToSummarize\n  );\n\n  if (!trimmedMessages.length) {\n    return \"Previous conversation was too long to summarize.\";\n  }\n\n  try {\n    const formattedPrompt = summaryPrompt.replace(\n      \"{messages}\",\n      JSON.stringify(trimmedMessages, null, 2)\n    );\n    /**\n     * Invoke the model with an empty callbacks array to prevent the internal\n     * summarization call from being streamed to the UI. This ensures the\n     * summarization is an internal housekeeping step that doesn't leak\n     * assistant messages or streaming events.\n     */\n    const response = await model.invoke(formattedPrompt, { callbacks: [] });\n    const content = response.content;\n    /**\n     * Handle both string content and MessageContent array\n     */\n    if (typeof content === \"string\") {\n      return content.trim();\n    } else if (Array.isArray(content)) {\n      /**\n       * Extract text from MessageContent array\n       */\n      const textContent = content\n        .map((item) => {\n          if (typeof item === \"string\") return item;\n          if (typeof item === \"object\" && item !== null && \"text\" in item) {\n            return (item as { text: string }).text;\n          }\n          return \"\";\n        })\n        .join(\"\");\n      return textContent.trim();\n    }\n    return \"Error generating summary: Invalid response format\";\n  } catch (e) {\n    return `Error generating summary: ${e}`;\n  }\n}\n\n/**\n * Trim messages to fit within summary generation limits\n */\nasync function trimMessagesForSummary(\n  messages: BaseMessage[],\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined\n): Promise<BaseMessage[]> {\n  if (trimTokensToSummarize === undefined) {\n    return messages;\n  }\n\n  try {\n    return await trimMessages(messages, {\n      maxTokens: trimTokensToSummarize,\n      tokenCounter: async (msgs) => tokenCounter(msgs),\n      strategy: \"last\",\n      allowPartial: true,\n      includeSystem: true,\n    });\n  } catch {\n    /**\n     * Fallback to last N messages if trimming fails\n     */\n    return messages.slice(-DEFAULT_FALLBACK_MESSAGE_COUNT);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA2BA,MAAa,yBAAyB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;WA0B5B,CAAC;AAEZ,MAAM,yBAAyB;AAC/B,MAAM,2BAA2B;AACjC,MAAM,2BAA2B;AACjC,MAAM,iCAAiC;AACvC,MAAM,8BAA8B;AAEpC,MAAM,qBAAqB,EACxB,UAAU,CACV,KAAK,EAAE,MAAM,EAAE,QAAqB,CAAC,CAAC,CACtC,QAAQ,EAAE,MAAM,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,AAAC,EAAC,CAAC;AAKxD,MAAa,oBAAoB,EAC9B,OAAO;CAIN,UAAU,EACP,QAAQ,CACR,GAAG,GAAG,kCAAkC,CACxC,IAAI,GAAG,2CAA2C,CAClD,UAAU;CAIb,QAAQ,EAAE,QAAQ,CAAC,SAAS,gCAAgC,CAAC,UAAU;CAIvE,UAAU,EACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,SAAS,kCAAkC,CAC3C,UAAU;AACd,EAAC,CACD,OACC,CAAC,SAAS;CACR,MAAM,QAAQ;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAS,EAAC,OACxD,CAAC,MAAM,MAAM,OACd,CAAC;AACF,QAAO,SAAS;AACjB,GACD,EACE,SAAS,iEACV,EACF;AAGH,MAAa,aAAa,EACvB,OAAO;CAIN,UAAU,EACP,QAAQ,CACR,IAAI,GAAG,gCAAgC,CACvC,IAAI,GAAG,2CAA2C,CAClD,UAAU;CAIb,QAAQ,EACL,QAAQ,CACR,IAAI,GAAG,4CAA4C,CACnD,UAAU;CACb,UAAU,EACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,IAAI,GAAG,gCAAgC,CACvC,UAAU;AACd,EAAC,CACD,OACC,CAAC,SAAS;CACR,MAAM,QAAQ;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAS,EAAC,OACxD,CAAC,MAAM,MAAM,OACd,CAAC;AACF,QAAO,UAAU;AAClB,GACD,EACE,SAAS,gEACV,EACF;AAGH,MAAM,gBAAgB,EAAE,OAAO;CAI7B,OAAO,EAAE,QAAoC;CAiB7C,SAAS,EAAE,MAAM,CAAC,mBAAmB,EAAE,MAAM,kBAAkB,AAAC,EAAC,CAAC,UAAU;CAI5E,MAAM,WAAW,UAAU;CAI3B,cAAc,mBAAmB,UAAU;CAK3C,eAAe,EAAE,QAAQ,CAAC,QAAQ,uBAAuB;CAIzD,uBAAuB,EAAE,QAAQ,CAAC,UAAU;CAI5C,eAAe,EAAE,QAAQ,CAAC,UAAU;CAIpC,wBAAwB,EAAE,QAAQ,CAAC,UAAU;CAI7C,gBAAgB,EAAE,QAAQ,CAAC,UAAU;AACtC,EAAC;;;;AASF,SAAgB,iBAAiBA,OAA8C;AAE7E,KACE,aAAa,SACb,OAAO,MAAM,YAAY,YACzB,MAAM,WACN,oBAAoB,MAAM,YACzB,OAAO,MAAM,QAAQ,mBAAmB,YACvC,MAAM,QAAQ,kBAAkB,MAElC,QAAO,MAAM,QAAQ,kBAAkB;AAIzC,KAAI,WAAW,SAAS,OAAO,MAAM,UAAU,SAC7C,QAAO,oBAAoB,MAAM,MAAM;AAEzC,KAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,QAAO,oBAAoB,MAAM,UAAU;AAG7C,QAAO;AACR;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDD,SAAgB,wBACdC,SACA;;;;CAIA,MAAM,EAAE,MAAM,aAAa,OAAO,GAAG,iBAAiB,eAAe,QAAQ;AAC7E,KAAI,MACF,OAAM,IAAI,MACR,CAAC,0CAA0C,EAAEC,IAAG,cAAc,MAAM,EAAE;AAI1E,QAAO,iBAAiB;EACtB,MAAM;EACN,eAAe,cAAc,OAAO,EAKlC,OAAO,EAAE,QAA2B,CAAC,UAAU,CAChD,EAAC;EACF,aAAa,OAAO,OAAO,YAAY;GACrC,IAAIC,UACF,YAAY;GACd,IAAIC,OAAoB,YAAY;;;;AAOpC,OAAI,YAAY,2BAA2B,QAAW;IACpD,QAAQ,KACN,kFACD;AACD,QAAI,YAAY,QACd,UAAU,EAAE,QAAQ,YAAY,uBAAwB;GAE3D;;;;AAKD,OAAI,YAAY,mBAAmB,QAAW;IAC5C,QAAQ,KACN,yEACD;AACD,QACE,CAAC,QACA,QACC,cAAc,QACd,KAAK,aAAa,0BAEpB,OAAO,EAAE,UAAU,YAAY,eAAgB;GAElD;;;;GAKD,MAAM,kBACJ,QAAQ,SAAS,YAAY,SACzB,QAAQ,QAAQ,UAChB;GACN,MAAM,eACJ,QAAQ,SAAS,SAAS,SACtB,QAAQ,QAAQ,OAChB,QAAQ,EAAE,UAAU,yBAA0B;GAEpD,MAAM,gBAAgB,WAAW,MAAM,aAAa;;;;GAKpD,IAAIC,oBAAmC,CAAE;AACzC,OAAI,oBAAoB,QACtB,oBAAoB,CAAE;YACb,MAAM,QAAQ,gBAAgB;;;;GAIvC,oBAAqB,gBAAkC,IAAI,CAAC,MAC1D,kBAAkB,MAAM,EAAE,CAC3B;;;;;GAKD,oBAAoB,CAAC,kBAAkB,MAAM,gBAAgB,AAAC;;;;GAMhE,MAAM,kBACJ,kBAAkB,KAAK,CAAC,MAAM,cAAc,EAAE,IAC9C,cAAc;GAEhB,MAAM,QACJ,OAAO,YAAY,UAAU,WACzB,MAAM,cAAc,YAAY,MAAM,GACtC,YAAY;AAElB,OAAI,mBAAmB,CAAC,iBAAiB,MAAM,CAC7C,OAAM,IAAI,MACR;GAKJ,MAAM,gBACJ,QAAQ,SAAS,kBAAkB,yBAC/B,YAAY,iBAAiB,yBAC7B,QAAQ,SAAS,iBACjB,YAAY,iBACZ;GACN,MAAM,gBACJ,QAAQ,QAAQ,iBAChB,YAAY,iBACZ;GACF,MAAM,wBACJ,QAAQ,SAAS,0BAA0B,SACvC,QAAQ,QAAQ,wBAChB,YAAY,yBAAyB;;;;GAK3C,iBAAiB,MAAM,SAAS;GAEhC,MAAM,eACJ,QAAQ,SAAS,iBAAiB,SAC9B,QAAQ,QAAQ,eAChB,YAAY,gBAAgB;GAClC,MAAM,cAAc,MAAM,aAAa,MAAM,SAAS;GACtD,MAAM,cAAc,MAAM,gBACxB,MAAM,UACN,aACA,mBACA,MACD;AAED,OAAI,CAAC,YACH;GAGF,MAAM,EAAE,cAAc,sBAAsB,GAAG,mBAC7C,MAAM,SACP;GACD,MAAM,cAAc,MAAM,qBACxB,sBACA,eACA,cACA,MACD;AAED,OAAI,eAAe,EACjB;GAGF,MAAM,EAAE,qBAAqB,mBAAmB,GAAG,kBACjD,cACA,sBACA,YACD;GAED,MAAM,UAAU,MAAM,cACpB,qBACA,OACA,eACA,cACA,sBACD;GAED,MAAM,iBAAiB,IAAI,aAAa;IACtC,SAAS,GAAG,cAAc,IAAI,EAAE,SAAS;IACzC,IAAIC,IAAM;GACX;AAED,UAAO,EACL,UAAU;IACR,IAAI,cAAc,EAAE,IAAI,oBAAqB;IAC7C;IACA,GAAG;GACJ,EACF;EACF;CACF,EAAC;AACH;;;;AAKD,SAAS,iBAAiBC,UAA+B;AACvD,MAAK,MAAM,OAAO,SAChB,KAAI,CAAC,IAAI,IACP,IAAI,KAAKD,IAAM;AAGpB;;;;AAKD,SAAS,mBAAmBC,UAG1B;AACA,KAAI,SAAS,SAAS,KAAK,cAAc,WAAW,SAAS,GAAG,CAC9D,QAAO;EACL,cAAc,SAAS;EACvB,sBAAsB,SAAS,MAAM,EAAE;CACxC;AAEH,QAAO,EACL,sBAAsB,SACvB;AACF;;;;AAKD,SAAS,kBACPC,cACAC,sBACAC,aAC0E;CAC1E,MAAM,sBAAsB,qBAAqB,MAAM,GAAG,YAAY;CACtE,MAAM,oBAAoB,qBAAqB,MAAM,YAAY;AAGjE,KAAI,cACF,oBAAoB,QAAQ,aAAa;AAG3C,QAAO;EAAE;EAAqB;CAAmB;AAClD;;;;;;;;;;;AAYD,eAAe,gBACbH,UACAI,aACAN,mBACAO,OACkB;AAClB,KAAI,kBAAkB,WAAW,EAC/B,QAAO;;;;AAMT,MAAK,MAAM,WAAW,mBAAmB;;;;EAIvC,IAAI,eAAe;EACnB,IAAI,iBAAiB;AAErB,MAAI,QAAQ,aAAa,QAAW;GAClC,iBAAiB;AACjB,OAAI,SAAS,SAAS,QAAQ,UAC5B,eAAe;EAElB;AAED,MAAI,QAAQ,WAAW,QAAW;GAChC,iBAAiB;AACjB,OAAI,cAAc,QAAQ,QACxB,eAAe;EAElB;AAED,MAAI,QAAQ,aAAa,QAAW;GAClC,iBAAiB;GACjB,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,OAAI,OAAO,mBAAmB,UAAU;IACtC,MAAM,YAAY,KAAK,MAAM,iBAAiB,QAAQ,SAAS;AAC/D,QAAI,cAAc,WAChB,eAAe;GAElB;;;;GAIC,eAAe;EAElB;;;;AAKD,MAAI,kBAAkB,aACpB,QAAO;CAEV;AAED,QAAO;AACR;;;;AAKD,eAAe,qBACbL,UACAH,MACAS,cACAD,OACiB;AACjB,KAAI,YAAY,QAAQ,cAAc,MAAM;EAC1C,MAAM,mBAAmB,MAAM,qBAC7B,UACA,MACA,cACA,MACD;AACD,MAAI,OAAO,qBAAqB,SAC9B,QAAO;;;;AAKT,SAAO,eAAe,UAAU,yBAAyB;CAC1D;;;;AAID,QAAO,eAAe,UAAU,KAAK,YAAY,yBAAyB;AAC3E;;;;AAKD,eAAe,qBACbL,UACAH,MACAS,cACAD,OAC6B;AAC7B,KAAI,SAAS,WAAW,EACtB,QAAO;CAGT,IAAIE;AAEJ,KAAI,cAAc,QAAQ,KAAK,aAAa,QAAW;EACrD,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,MAAI,OAAO,mBAAmB,SAC5B;EAEF,mBAAmB,KAAK,MAAM,iBAAiB,KAAK,SAAS;CAC9D,WAAU,YAAY,QAAQ,KAAK,WAAW,QAC7C,mBAAmB,KAAK,MAAM,KAAK,OAAO;KAE1C;AAGF,KAAI,oBAAoB,GACtB,mBAAmB;CAGrB,MAAM,cAAc,MAAM,aAAa,SAAS;AAChD,KAAI,eAAe,iBACjB,QAAO;;;;;CAOT,IAAI,OAAO;CACX,IAAI,QAAQ,SAAS;CACrB,IAAI,kBAAkB,SAAS;CAC/B,MAAM,gBAAgB,KAAK,MAAM,KAAK,KAAK,SAAS,OAAO,CAAC,GAAG;AAE/D,MAAK,IAAI,IAAI,GAAG,IAAI,eAAe,KAAK;AACtC,MAAI,QAAQ,MACV;EAGF,MAAM,MAAM,KAAK,OAAO,OAAO,SAAS,EAAE;EAC1C,MAAM,eAAe,MAAM,aAAa,SAAS,MAAM,IAAI,CAAC;AAC5D,MAAI,gBAAgB,kBAAkB;GACpC,kBAAkB;GAClB,QAAQ;EACT,OACC,OAAO,MAAM;CAEhB;AAED,KAAI,oBAAoB,SAAS,QAC/B,kBAAkB;AAGpB,KAAI,mBAAmB,SAAS,QAAQ;AACtC,MAAI,SAAS,WAAW,EACtB,QAAO;EAET,kBAAkB,SAAS,SAAS;CACrC;;;;AAKD,MAAK,IAAI,IAAI,iBAAiB,KAAK,GAAG,IACpC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;AACR;;;;AAKD,SAAS,eACPP,UACAQ,gBACQ;AACR,KAAI,SAAS,UAAU,eACrB,QAAO;CAGT,MAAM,eAAe,SAAS,SAAS;AAEvC,MAAK,IAAI,IAAI,cAAc,KAAK,GAAG,IACjC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;AACR;;;;AAKD,SAAS,kBACPR,UACAG,aACS;AACT,KAAI,eAAe,SAAS,OAC1B,QAAO;;;;AAMT,KACE,cAAc,SAAS,UACvB,UAAU,WAAW,SAAS,aAAa,IAC3C,aAAa,SAAS,aAAa,CAEnC,QAAO;CAGT,MAAM,cAAc,KAAK,IAAI,GAAG,cAAc,4BAA4B;CAC1E,MAAM,YAAY,KAAK,IACrB,SAAS,QACT,cAAc,4BACf;AAED,MAAK,IAAI,IAAI,aAAa,IAAI,WAAW,KAAK;AAC5C,MAAI,CAAC,aAAa,SAAS,GAAG,CAC5B;EAGF,MAAM,cAAc,mBAAmB,SAAS,GAAgB;AAChE,MAAI,wBAAwB,UAAU,GAAG,aAAa,YAAY,CAChE,QAAO;CAEV;AAED,QAAO;AACR;;;;AAKD,SAAS,mBAAmBM,WAAmC;CAC7D,MAAM,8BAAc,IAAI;AACxB,KAAI,UAAU,WACZ,MAAK,MAAM,YAAY,UAAU,YAAY;EAC3C,MAAM,KACJ,OAAO,aAAa,YAAY,QAAQ,WAAW,SAAS,KAAK;AACnE,MAAI,IACF,YAAY,IAAI,GAAG;CAEtB;AAEH,QAAO;AACR;;;;AAKD,SAAS,wBACPT,UACAU,gBACAP,aACAQ,aACS;AACT,MAAK,IAAI,IAAI,iBAAiB,GAAG,IAAI,SAAS,QAAQ,KAAK;EACzD,MAAM,UAAU,SAAS;AACzB,MACE,YAAY,WAAW,QAAQ,IAC/B,YAAY,IAAI,QAAQ,aAAa,EACrC;GACA,MAAM,iBAAiB,iBAAiB;GACxC,MAAM,mBAAmB,IAAI;AAC7B,OAAI,mBAAmB,iBACrB,QAAO;EAEV;CACF;AACD,QAAO;AACR;;;;AAKD,eAAe,cACbC,qBACAP,OACAQ,eACAP,cACAQ,uBACiB;AACjB,KAAI,CAAC,oBAAoB,OACvB,QAAO;CAGT,MAAM,kBAAkB,MAAM,uBAC5B,qBACA,cACA,sBACD;AAED,KAAI,CAAC,gBAAgB,OACnB,QAAO;AAGT,KAAI;EACF,MAAM,kBAAkB,cAAc,QACpC,cACA,KAAK,UAAU,iBAAiB,MAAM,EAAE,CACzC;;;;;;;EAOD,MAAM,WAAW,MAAM,MAAM,OAAO,iBAAiB,EAAE,WAAW,CAAE,EAAE,EAAC;EACvE,MAAM,UAAU,SAAS;;;;AAIzB,MAAI,OAAO,YAAY,SACrB,QAAO,QAAQ,MAAM;WACZ,MAAM,QAAQ,QAAQ,EAAE;;;;GAIjC,MAAM,cAAc,QACjB,IAAI,CAAC,SAAS;AACb,QAAI,OAAO,SAAS,SAAU,QAAO;AACrC,QAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,KACzD,QAAQ,KAA0B;AAEpC,WAAO;GACR,EAAC,CACD,KAAK,GAAG;AACX,UAAO,YAAY,MAAM;EAC1B;AACD,SAAO;CACR,SAAQ,GAAG;AACV,SAAO,CAAC,0BAA0B,EAAE,GAAG;CACxC;AACF;;;;AAKD,eAAe,uBACbd,UACAM,cACAQ,uBACwB;AACxB,KAAI,0BAA0B,OAC5B,QAAO;AAGT,KAAI;AACF,SAAO,MAAM,aAAa,UAAU;GAClC,WAAW;GACX,cAAc,OAAO,SAAS,aAAa,KAAK;GAChD,UAAU;GACV,cAAc;GACd,eAAe;EAChB,EAAC;CACH,QAAO;;;;AAIN,SAAO,SAAS,MAAM,CAAC,+BAA+B;CACvD;AACF"}

package/dist/agents/middleware/types.cjs

@@ -0,0 +1,12 @@
+
+//#region src/agents/middleware/types.ts
+/**
+* Unique symbol used to brand middleware instances.
+* This prevents functions from being accidentally assignable to AgentMiddleware
+* since functions have a 'name' property that would otherwise make them structurally compatible.
+*/
+const MIDDLEWARE_BRAND = Symbol("AgentMiddleware");
+
+//#endregion
+exports.MIDDLEWARE_BRAND = MIDDLEWARE_BRAND;
+//# sourceMappingURL=types.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.cjs","names":["MIDDLEWARE_BRAND: unique symbol"],"sources":["../../../src/agents/middleware/types.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport type {\n  InteropZodObject,\n  InteropZodDefault,\n  InteropZodOptional,\n  InferInteropZodInput,\n  InferInteropZodOutput,\n} from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot } 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\nexport type NormalizedSchemaInput<\n  TSchema extends InteropZodObject | undefined | never = any\n> = [TSchema] extends [never]\n  ? AgentBuiltInState\n  : TSchema extends InteropZodObject\n  ? InferInteropZodOutput<TSchema> & AgentBuiltInState\n  : TSchema extends Record<string, unknown>\n  ? 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  tool: ClientTool | ServerTool;\n  /**\n   * The current agent state (includes both middleware state and built-in state).\n   */\n  state: TState & AgentBuiltInState;\n  /**\n   * The runtime context containing metadata, signal, writer, interrupt, etc.\n   */\n  runtime: Runtime<TContext>;\n}\n\n/**\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 InteropZodObject | 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 InteropZodObject | 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 InteropZodObject | undefined = undefined,\n  TContext = unknown\n> = (\n  request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>,\n  handler: WrapModelCallHandler<TSchema, TContext>\n) => PromiseOrValue<AIMessage>;\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 InteropZodObject | 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 InteropZodObject | 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 InteropZodObject | 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 InteropZodObject | 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 */\nexport interface AgentMiddleware<\n  TSchema extends InteropZodObject | undefined = any,\n  TContextSchema extends\n    | InteropZodObject\n    | InteropZodDefault<InteropZodObject>\n    | InteropZodOptional<InteropZodObject>\n    | undefined = any,\n  TFullContext = any\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   * 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\n   * - A Zod optional object\n   * - A Zod default object\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?: (ClientTool | ServerTool)[];\n  /**\n   * Wraps tool execution with custom logic. This allows you to:\n   * - Modify tool call parameters before execution\n   * - Handle errors and retry with different parameters\n   * - Post-process tool results\n   * - Implement caching, logging, authentication, or other cross-cutting concerns\n   * - Return Command objects for advanced control flow\n   *\n   * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n   * along with a handler function to execute the actual tool.\n   *\n   * @param request - The tool call request containing toolCall, state, and runtime.\n   * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n   * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n   *\n   * @example\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   console.log(`Calling tool: ${request.tool.name}`);\n   *   console.log(`Tool description: ${request.tool.description}`);\n   *\n   *   try {\n   *     // Execute the tool\n   *     const result = await handler(request);\n   *     console.log(`Tool ${request.tool.name} succeeded`);\n   *     return result;\n   *   } catch (error) {\n   *     console.error(`Tool ${request.tool.name} failed:`, error);\n   *     // Could return a custom error message or retry\n   *     throw error;\n   *   }\n   * }\n   * ```\n   *\n   * @example Authentication\n   * ```ts\n   * wrapToolCall: async (request, handler) => {\n   *   // Check if user is authorized for this tool\n   *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n   *     return new ToolMessage({\n   *       content: \"Unauthorized to call this tool\",\n   *       tool_call_id: request.toolCall.id,\n   *     });\n   *   }\n   *   return handler(request);\n   * }\n   * ```\n   *\n   * @example Caching\n   * ```ts\n   * const cache = new Map();\n   * wrapToolCall: async (request, handler) => {\n   *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n   *   if (cache.has(cacheKey)) {\n   *     return cache.get(cacheKey);\n   *   }\n   *   const result = await handler(request);\n   *   cache.set(cacheKey, result);\n   *   return result;\n   * }\n   * ```\n   */\n  wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n\n  /**\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\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 */\nexport type InferMiddlewareState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer S, any, any>\n    ? S extends InteropZodObject\n      ? FilterPrivateProps<InferInteropZodOutput<S>>\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 */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> =\n  T extends AgentMiddleware<infer S, any, any>\n    ? S extends InteropZodObject\n      ? FilterPrivateProps<InferInteropZodInput<S>>\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 = AgentMiddleware[]> = 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 C, any>\n    ? C extends InteropZodObject\n      ? InferInteropZodInput<C>\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 C, any>\n    ? C extends InteropZodOptional<infer Inner>\n      ? InferInteropZodInput<Inner> | undefined\n      : C extends InteropZodObject\n      ? InferInteropZodInput<C>\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 AnyAnnotationRoot | InteropZodObject> =\n  A extends AnyAnnotationRoot\n    ? A\n    : A extends InteropZodObject\n    ? AnnotationRoot<InteropZodToStateDefinition<A>>\n    : never;\n\nexport type InferSchemaInput<\n  A extends AnyAnnotationRoot | InteropZodObject | undefined\n> = A extends AnyAnnotationRoot | InteropZodObject\n  ? ToAnnotationRoot<A>[\"State\"]\n  : {};\n"],"mappings":";;;;;;;AA+PA,MAAaA,mBAAkC,OAAO,kBAAkB"}

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

@@ -153,10 +153,21 @@ type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TC
   hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
+/**
+ * Unique symbol used to brand middleware instances.
+ * This prevents functions from being accidentally assignable to AgentMiddleware
+ * since functions have a 'name' property that would otherwise make them structurally compatible.
+ */
+declare const MIDDLEWARE_BRAND: unique symbol;
 /**
  * Base middleware interface.
  */
 interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {
+  /**
+   * Brand property to distinguish middleware instances from plain objects or functions.
+   * This is required and prevents accidental assignment of functions to middleware arrays.
+   */
+  readonly [MIDDLEWARE_BRAND]: true;
   /**
    * The name of the middleware.
    */
@@ -368,5 +379,5 @@ type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObjec
 type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;
 type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>["State"] : {};
 //#endregion
-export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContexts, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareState, InferMiddlewareStates, InferSchemaInput, MiddlewareResult, NormalizedSchemaInput, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
+export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContexts, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareState, InferMiddlewareStates, InferSchemaInput, MIDDLEWARE_BRAND, MiddlewareResult, NormalizedSchemaInput, 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":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","InteropZodToStateDefinition","AnnotationRoot","AIMessage","SystemMessage","ToolMessage","ToolCall","Command","ClientTool","ServerTool","JumpToTarget","Runtime","AgentBuiltInState","ModelRequest","PromiseOrValue","T","Promise","AnyAnnotationRoot","NormalizedSchemaInput","TSchema","Record","MiddlewareResult","TState","ToolCallRequest","TContext","ToolCallHandler","WrapToolCallHook","WrapModelCallHandler","Omit","WrapModelCallHook","BeforeAgentHandler","Partial","BeforeAgentHook","BeforeModelHandler","BeforeModelHook","AfterModelHandler","AfterModelHook","AfterAgentHandler","AfterAgentHook","AgentMiddleware","TContextSchema","TFullContext","FilterPrivateProps","K","InferChannelType","ToAnnotationRoot","InferMiddlewareState","S","InferMiddlewareInputState","InferMiddlewareStates","First","Rest","InferMiddlewareInputStates","InferMergedState","InferMergedInputState","InferMiddlewareContext","C","InferMiddlewareContextInput","Inner","InferMiddlewareContexts","MergeContextTypes","A","B","InferMiddlewareContextInputs","InferContextInput","ContextSchema","InferSchemaInput"],"sources":["../../../src/agents/middleware/types.d.ts"],"sourcesContent":["import type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot } from \"@langchain/langgraph\";\nimport type { AIMessage, SystemMessage, ToolMessage } from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command } from \"@langchain/langgraph\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\ntype PromiseOrValue<T> = T | Promise<T>;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\nexport type NormalizedSchemaInput<TSchema extends InteropZodObject | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends Record<string, unknown> ? TSchema & AgentBuiltInState : AgentBuiltInState;\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> = (TState & {\n    jumpTo?: JumpToTarget;\n}) | void;\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<TState extends Record<string, unknown> = Record<string, unknown>, TContext = unknown> {\n    /**\n     * The tool call to be executed\n     */\n    toolCall: ToolCall;\n    /**\n     * The BaseTool instance being invoked.\n     * Provides access to tool metadata like name, description, schema, etc.\n     */\n    tool: ClientTool | ServerTool;\n    /**\n     * The current agent state (includes both middleware state and built-in state).\n     */\n    state: TState & AgentBuiltInState;\n    /**\n     * The runtime context containing metadata, signal, writer, interrupt, etc.\n     */\n    runtime: Runtime<TContext>;\n}\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<TSchema extends Record<string, unknown> = AgentBuiltInState, TContext = unknown> = (request: ToolCallRequest<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: Omit<ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, \n/**\n * allow to reset the system prompt or system message\n */\n\"systemPrompt\" | \"systemMessage\"> & {\n    systemPrompt?: string;\n    systemMessage?: SystemMessage;\n}) => PromiseOrValue<AIMessage>;\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Base middleware interface.\n */\nexport interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {\n    /**\n     * The name of the middleware.\n     */\n    name: string;\n    /**\n     * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:\n     * - A Zod object\n     * - A Zod optional object\n     * - A Zod default object\n     * - Undefined\n     */\n    stateSchema?: TSchema;\n    /**\n     * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:\n     * - A Zod object\n     * - A Zod optional object\n     * - A Zod default object\n     * - Undefined\n     */\n    contextSchema?: TContextSchema;\n    /**\n     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * Wraps tool execution with custom logic. This allows you to:\n     * - Modify tool call parameters before execution\n     * - Handle errors and retry with different parameters\n     * - Post-process tool results\n     * - Implement caching, logging, authentication, or other cross-cutting concerns\n     * - Return Command objects for advanced control flow\n     *\n     * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n     * along with a handler function to execute the actual tool.\n     *\n     * @param request - The tool call request containing toolCall, state, and runtime.\n     * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n     * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n     *\n     * @example\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   console.log(`Calling tool: ${request.tool.name}`);\n     *   console.log(`Tool description: ${request.tool.description}`);\n     *\n     *   try {\n     *     // Execute the tool\n     *     const result = await handler(request);\n     *     console.log(`Tool ${request.tool.name} succeeded`);\n     *     return result;\n     *   } catch (error) {\n     *     console.error(`Tool ${request.tool.name} failed:`, error);\n     *     // Could return a custom error message or retry\n     *     throw error;\n     *   }\n     * }\n     * ```\n     *\n     * @example Authentication\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   // Check if user is authorized for this tool\n     *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n     *     return new ToolMessage({\n     *       content: \"Unauthorized to call this tool\",\n     *       tool_call_id: request.toolCall.id,\n     *     });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     *\n     * @example Caching\n     * ```ts\n     * const cache = new Map();\n     * wrapToolCall: async (request, handler) => {\n     *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n     *   if (cache.has(cacheKey)) {\n     *     return cache.get(cacheKey);\n     *   }\n     *   const result = await handler(request);\n     *   cache.set(cacheKey, result);\n     *   return result;\n     * }\n     * ```\n     */\n    wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n    /**\n     * Wraps the model invocation with custom logic. This allows you to:\n     * - Modify the request before calling the model\n     * - Handle errors and retry with different parameters\n     * - Post-process the response\n     * - Implement custom caching, logging, or other cross-cutting concerns\n     *\n     * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n     * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n     * @returns The response from the model (or a modified version).\n     *\n     * @example\n     * ```ts\n     * wrapModelCall: async (request, handler) => {\n     *   // Modify request before calling\n     *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n     *\n     *   try {\n     *     // Call the model\n     *     return await handler(modifiedRequest);\n     *   } catch (error) {\n     *     // Handle errors and retry with fallback\n     *     const fallbackRequest = { ...request, model: fallbackModel };\n     *     return await handler(fallbackRequest);\n     *   }\n     * }\n     * ```\n     */\n    wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n     * It allows to modify the state of the agent before any model calls or tool executions.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n     * It allows to modify the state of the agent.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n     * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterModel?: AfterModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n     * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n    [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>[\"State\"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<S>> : {} : {};\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<S>> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T = AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined] ? undefined : B | undefined : [B] extends [undefined] ? A | undefined : [A] extends [B] ? A : [B] extends [A] ? B : A & B;\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>[\"State\"] : {};\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>[\"State\"] : {};\nexport {};\n//# sourceMappingURL=types.d.ts.map"],"mappings":";;;;;;;;;;;KAUKkB,oBAAoBC,IAAIC,QAAQD;AAAhCD,KACOG,iBAAAA,GAAoBf,cADb,CAAA,GAAA,CAAA;AAAMa,KAEbG,qBAFaH,CAAAA,gBAEyBnB,gBAFzBmB,GAAAA,SAAAA,GAAAA,KAAAA,GAAAA,GAAAA,CAAAA,GAAAA,CAEwEI,OAFxEJ,CAAAA,SAAAA,CAAAA,KAAAA,CAAAA,GAEmGH,iBAFnGG,GAEuHI,OAFvHJ,SAEuInB,gBAFvImB,GAE0Jf,qBAF1Je,CAEgLI,OAFhLJ,CAAAA,GAE2LH,iBAF3LG,GAE+MI,OAF/MJ,SAE+NK,MAF/NL,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAEyPI,OAFzPJ,GAEmQH,iBAFnQG,GAEuRH,iBAFvRG;;;AAAW;AACxBE,KAKAI,gBALiB,CAAA,MAAA,CAAGnB,GAAAA,CAKQoB,MALRpB,GAAAA;EACpBgB,MAAAA,CAAAA,EAKCR,YALDQ;CAAsCtB,CAAAA,GAAAA,IAAAA;;;;;AAAuJuB,UAWxLI,eAXwLJ,CAAAA,eAWzJC,MAXyJD,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAW/HC,MAX+HD,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,WAAAA,OAAAA,CAAAA,CAAAA;EAAtBnB;;;EAAqEoB,QAAAA,EAe1Od,UAf0Oc;EAA0BD;;;AAA+C;EAIrTE,IAAAA,EAgBFb,UAhBEa,GAgBWZ,UAhBKa;EAOXC;;;EAIHjB,KAAAA,EASHgB,MATGhB,GASMM,iBATNN;EAKJE;;;EAIUI,OAAAA,EAIPD,OAJOC,CAICY,QAJDZ,CAAAA;;;AAIA;AAMpB;;AAAsEA,KAA1Da,eAA0Db,CAAAA,gBAA1BQ,MAA0BR,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAAAA,iBAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAAmDW,eAAnDX,CAAmEO,OAAnEP,EAA4EY,QAA5EZ,CAAAA,EAAAA,GAA0FE,cAA1FF,CAAyGP,WAAzGO,GAAuHL,OAAvHK,CAAAA;;;;;AAAuHL,KAKjLmB,gBALiLnB,CAAAA,gBAKhJX,gBALgJW,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAKtEgB,eALsEhB,CAKtDW,qBALsDX,CAKhCY,OALgCZ,CAAAA,EAKtBiB,QALsBjB,CAAAA,EAAAA,OAAAA,EAKFkB,eALElB,CAKcW,qBALdX,CAKoCY,OALpCZ,CAAAA,EAK8CiB,QAL9CjB,CAAAA,EAAAA,GAK4DO,cAL5DP,CAK2EF,WAL3EE,GAKyFA,OALzFA,CAAAA;;AAAf;AAK9K;;;;;AAAuHgB,KAQ3GI,oBAR2GJ,CAAAA,gBAQtE3B,gBARsE2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAQIK,IARJL,CAQSV,YARTU,CAQsBL,qBARtBK,CAQ4CJ,OAR5CI,CAAAA,EAQsDC,QARtDD,CAAAA;;;;cAAoEE,GAAAA,eAAAA,CAAAA,GAAAA;EAA6EpB,YAAAA,CAAAA,EAAAA,MAAAA;EAAcE,aAAAA,CAAAA,EAclQH,aAdkQG;CAA7BO,EAAAA,GAenPA,cAfmPA,CAepOX,SAfoOW,CAAAA;AAAc;AAQvQ;;;;;;;;;;AAOoB;AAcpB;AAA8ClB,KAAlCiC,iBAAkCjC,CAAAA,gBAAAA,gBAAAA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAA0EiB,YAA1EjB,CAAuFsB,qBAAvFtB,CAA6GuB,OAA7GvB,CAAAA,EAAuH4B,QAAvH5B,CAAAA,EAAAA,OAAAA,EAA2I+B,oBAA3I/B,CAAgKuB,OAAhKvB,EAAyK4B,QAAzK5B,CAAAA,EAAAA,GAAuLkB,cAAvLlB,CAAsMO,SAAtMP,CAAAA;;;;;;;;;KASzCkC,kBATgOhB,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAShLK,OATgLL,EAAAA,OAAAA,EAS9JH,OAT8JG,CAStJU,QATsJV,CAAAA,EAAAA,GASxIA,cATwIA,CASzHO,gBATyHP,CASxGiB,OATwGjB,CAShGK,OATgGL,CAAAA,CAAAA,CAAAA;AAAc;AAAY;;;;AAS1HK,KAMzHa,eANyHb,CAAAA,gBAMzFvB,gBANyFuB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAMzBW,kBANyBX,CAMND,qBANMC,CAMgBA,OANhBA,CAAAA,EAM0BK,QAN1BL,CAAAA,GAAAA;EAARY,IAAAA,EAOnHD,kBAPmHC,CAOhGb,qBAPgGa,CAO1EZ,OAP0EY,CAAAA,EAOhEP,QAPgEO,CAAAA;EAAjBV,SAAAA,CAAAA,EAQ5FX,YAR4FW,EAAAA;CAAfP;AAAc;AAM3G;;;;;;;KAYKmB,kBAXwBf,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAWwBC,OAXxBD,EAAAA,OAAAA,EAW0CP,OAX1CO,CAWkDM,QAXlDN,CAAAA,EAAAA,GAWgEJ,cAXhEI,CAW+EG,gBAX/EH,CAWgGa,OAXhGb,CAWwGC,OAXxGD,CAAAA,CAAAA,CAAAA;;;;AACD;AAC1B;AASmDC,KAMzCe,eANyCf,CAAAA,gBAMTvB,gBANSuB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAMuDc,kBANvDd,CAM0ED,qBAN1EC,CAMgGA,OANhGA,CAAAA,EAM0GK,QAN1GL,CAAAA,GAAAA;EAA0BK,IAAAA,EAOrES,kBAPqET,CAOlDN,qBAPkDM,CAO5BL,OAP4BK,CAAAA,EAOlBA,QAPkBA,CAAAA;EAARb,SAAAA,CAAAA,EAQvDD,YARuDC,EAAAA;CAA8DQ;;;;AAA1B;AAM3G;;;;;KAaKgB,iBAbuGF,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAaxDd,OAbwDc,EAAAA,OAAAA,EAatCtB,OAbsCsB,CAa9BT,QAb8BS,CAAAA,EAAAA,GAahBnB,cAbgBmB,CAaDZ,gBAbCY,CAagBF,OAbhBE,CAawBd,OAbxBc,CAAAA,CAAAA,CAAAA;;;;;;AAEhF,KAiBhBG,cAjBgB,CAAA,gBAiBexC,gBAjBf,GAAA,SAAA,GAAA,SAAA,EAAA,WAAA,OAAA,CAAA,GAiB+EuC,iBAjB/E,CAiBiGjB,qBAjBjG,CAiBuHC,OAjBvH,CAAA,EAiBiIK,QAjBjI,CAAA,GAAA;EAWvBW,IAAAA,EAOKA,iBAPY,CAOMjB,qBAPNM,CAO4BL,OAP5B,CAAA,EAOsCK,QAPtC,CAAA;EAA8BL,SAAAA,CAAAA,EAQpCT,YARoCS,EAAAA;CAA0BK;;;;;;AAA4B;AAM1G;;KAYKa,iBAZ8IlB,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAY/FA,OAZ+FA,EAAAA,OAAAA,EAY7ER,OAZ6EQ,CAYrEK,QAZqEL,CAAAA,EAAAA,GAYvDL,cAZuDK,CAYxCE,gBAZwCF,CAYvBY,OAZuBZ,CAYfA,OAZeA,CAAAA,CAAAA,CAAAA;;;;;;AACvFK,KAiBhDc,cAjBgDd,CAAAA,gBAiBjB5B,gBAjBiB4B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAiB+Ca,iBAjB/Cb,CAiBiEN,qBAjBjEM,CAiBuFL,OAjBvFK,CAAAA,EAiBiGA,QAjBjGA,CAAAA,GAAAA;EAAlDW,IAAAA,EAkBAE,iBAlBAF,CAkBkBjB,qBAlBlBiB,CAkBwChB,OAlBxCgB,CAAAA,EAkBkDX,QAlBlDW,CAAAA;EACMzB,SAAAA,CAAAA,EAkBAA,YAlBAA,EAAAA;AAAY,CAAA;AAC1B;;;AASoEC,UAarD4B,eAbqD5B,CAAAA,gBAarBf,gBAbqBe,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,uBAasCf,gBAbtCe,GAayDd,iBAbzDc,CAa2Ef,gBAb3Ee,CAAAA,GAa+Fb,kBAb/Fa,CAakHf,gBAblHe,CAAAA,GAAAA,SAAAA,GAAAA,GAAAA,EAAAA,eAAAA,GAAAA,CAAAA,CAAAA;EAA8DQ;;;EAAxCL,IAAAA,EAAAA,MAAAA;EAAc;AAM1G;;;;;;EACkDK,WAAAA,CAAAA,EAkBhCA,OAlBgCA;EAAtBD;;;;AACA;AAK5B;;EAA4GtB,aAAAA,CAAAA,EAoBxF4C,cApBwF5C;EAAqCA;;;EAAoBE,KAAAA,CAAAA,EAAAA,CAwBxJU,UAxBwJV,GAwB3IW,UAxB2IX,CAAAA,EAAAA;EAYnJqB;;;;;;;;;;;;;;;;;;;;;;AA4Ia;AAC9B;;;;;;AAK4D;AAE7D;;;;;;;;;;;AAA6L;AAK7L;;;;;;;;;AAA4J;AAK5J;;;;;;;;;AAAiK;EAIrJ8B,YAAAA,CAAAA,EAvFOvB,gBAuFcX,CAvFGI,OAuFH,EAvFYsB,YAuFZ,CAAA;EAAKF;;;;;;;;;;;;;AAA8P;AAIpS;;;;;;;;;;;;;;EAAuU,aAAA,CAAA,EA9DnTV,iBA8DmT,CA9DjSV,OA8DiS,EA9DxRsB,YA8DwR,CAAA;EAI3TY;;;;;AAAqG;AAIjH;;EAAqGtC,WAAAA,CAAAA,EA7DnFiB,eA6DmFjB,CA7DnEI,OA6DmEJ,EA7D1D0B,YA6D0D1B,CAAAA;EAA3BqC;;AAAiD;AAI3H;;;;;EAAyHxD,WAAAA,CAAAA,EAxDvGsC,eAwDuGtC,CAxDvFuB,OAwDuFvB,EAxD9E6C,YAwD8E7C,CAAAA;EAAwC4D;;AAAD;AAIhK;;;;;EAA8H1D,UAAAA,CAAAA,EAnD7GsC,cAmD6GtC,CAnD9FqB,OAmD8FrB,EAnDrF2C,YAmDqF3C,CAAAA;EAAuD4D;;;;;;AAAsE;AAI3P;EAAuDnB,UAAAA,CAAAA,EA9CtCD,cA8CsCC,CA9CvBpB,OA8CuBoB,EA9CdE,YA8CcF,CAAAA;;;;;KAzClDG,kBAyCsLS,CAAAA,CAAAA,CAAAA,GAAAA,QAAsBZ,MAxCjMxB,CAwCiMwB,IAxC5LI,CAwC4LJ,SAAAA,IAAAA,MAAAA,EAAAA,GAAAA,KAAAA,GAxC3JI,CAwC2JJ,GAxCvJxB,CAwCuJwB,CAxCrJI,CAwCqJJ,CAAAA,EAA2CW;AAAvBK,KAtCzNX,gBAsCyNW,CAAAA,UAtC9LtC,iBAsC8LsC,GAtC1K3D,gBAsC0K2D,CAAAA,GAtCtJxC,CAsCsJwC,SAtC5ItC,iBAsC4IsC,GAtCxHV,gBAsCwHU,CAtCvGxC,CAsCuGwC,CAAAA,CAAAA,OAAAA,CAAAA,GAtCzFxC,CAsCyFwC,SAtC/E3D,gBAsC+E2D,GAtC5DxD,oBAsC4DwD,CAtCvCxC,CAsCuCwC,CAAAA,GAAAA,CAAAA,CAAAA;;;;;AAAsF,KAjC/ST,oBAiC+S,CAAA,UAjChRP,eAiCgR,CAAA,GAjC7PxB,CAiC6P,SAjCnPwB,eAiCmP,CAAA,KAAA,EAAA,EAAA,GAAA,EAAA,GAAA,CAAA,GAjC9MQ,CAiC8M,SAjCpMnD,gBAiCoM,GAjCjL8C,kBAiCiL,CAjC9J1C,qBAiC8J,CAjCxI+C,CAiCwI,CAAA,CAAA,GAAA,CAAA,CAAA,GAAA,CAAA,CAAA;AAAkB;;;;AAI7Ne,KAhCpGd,yBAgCoGc,CAAAA,UAhChEvB,eAgCgEuB,CAAAA,GAhC7C/C,CAgC6C+C,SAhCnCvB,eAgCmCuB,CAAAA,KAAAA,EAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GAhCEf,CAgCFe,SAhCYlE,gBAgCZkE,GAhC+BpB,kBAgC/BoB,CAhCkD/D,oBAgClD+D,CAhCuEf,CAgCvEe,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA2DD,KA5B/JZ,qBA4B+JY,CAAAA,IA5BrItB,eA4BqIsB,EAAAA,CAAAA,GA5BhH9C,CA4BgH8C,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GA5BnF9C,CA4BmF8C,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GA5BjCX,KA4BiCW,SA5BnBtB,eA4BmBsB,GA5BDV,IA4BCU,SAAAA,SA5BqBtB,eA4BrBsB,EAAAA,GA5ByCf,oBA4BzCe,CA5B8DX,KA4B9DW,CAAAA,GA5BuEZ,qBA4BvEY,CA5B6FV,IA4B7FU,CAAAA,GA5BqGf,oBA4BrGe,CA5B0HX,KA4B1HW,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA0BA,KAxBzLT,0BAwByLS,CAAAA,UAAAA,SAxB3ItB,eAwB2IsB,EAAAA,CAAAA,GAxBtH9C,CAwBsH8C,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAxBzF9C,CAwByF8C,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAxBvCX,KAwBuCW,SAxBzBtB,eAwByBsB,GAxBPV,IAwBOU,SAAAA,SAxBetB,eAwBfsB,EAAAA,GAxBmCb,yBAwBnCa,CAxB6DX,KAwB7DW,CAAAA,GAxBsET,0BAwBtES,CAxBiGV,IAwBjGU,CAAAA,GAxByGb,yBAwBzGa,CAxBmIX,KAwBnIW,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;AAAK;AAI1M;AAA4DtB,KAxBhDc,gBAwBgDd,CAAAA,UAAAA,SAxBZA,eAwBYA,EAAAA,CAAAA,GAxBSU,qBAwBTV,CAxB+BxB,CAwB/BwB,CAAAA,GAxBoC3B,iBAwBpC2B;;;;AAAkHA,KApBlKe,qBAoBkKf,CAAAA,UAAAA,SApBzHA,eAoByHA,EAAAA,CAAAA,GApBpGa,0BAoBoGb,CApBzExB,CAoByEwB,CAAAA,GApBpE3B,iBAoBoE2B;;;;AAA8EkB,KAhBhPF,sBAgBgPE,CAAAA,UAhB/MlB,eAgB+MkB,CAAAA,GAhB5L1C,CAgB4L0C,SAhBlLlB,eAgBkLkB,CAAAA,GAAAA,EAAAA,KAAAA,EAAAA,EAAAA,GAAAA,CAAAA,GAhB7ID,CAgB6IC,SAhBnI7D,gBAgBmI6D,GAhBhH1D,oBAgBgH0D,CAhB3FD,CAgB2FC,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAAsGP,KAZtVO,2BAYsVP,CAAAA,UAZhTX,eAYgTW,CAAAA,GAZ7RnC,CAY6RmC,SAZnRX,eAYmRW,CAAAA,GAAAA,EAAAA,KAAAA,EAAAA,EAAAA,GAAAA,CAAAA,GAZ9OM,CAY8ON,SAZpOpD,kBAYoOoD,CAAAA,KAAAA,MAAAA,CAAAA,GAZlMnD,oBAYkMmD,CAZ7KQ,KAY6KR,CAAAA,GAAAA,SAAAA,GAZxJM,CAYwJN,SAZ9ItD,gBAY8IsD,GAZ3HnD,oBAY2HmD,CAZtGM,CAYsGN,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;AAAD;AAIjW;AAAoDjC,KAZxC0C,uBAYwC1C,CAAAA,UAAAA,SAZGsB,eAYHtB,EAAAA,CAAAA,GAZwBF,CAYxBE,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAZqDF,CAYrDE,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAZuGiC,KAYvGjC,SAZqHsB,eAYrHtB,GAZuIkC,IAYvIlC,SAAAA,SAZ6JsB,eAY7JtB,EAAAA,GAZiLsC,sBAYjLtC,CAZwMiC,KAYxMjC,CAAAA,GAZiN0C,uBAYjN1C,CAZyOkC,IAYzOlC,CAAAA,GAZiPsC,sBAYjPtC,CAZwQiC,KAYxQjC,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;KAR/C2C,iBAQqJK,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CAR1HJ,CAQ0HI,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,CARhGH,CAQgGG,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,SAAAA,GAR3DH,CAQ2DG,GAAAA,SAAAA,GAAAA,CAR1CH,CAQ0CG,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GARjBJ,CAQiBI,GAAAA,SAAAA,GAAAA,CARAJ,CAQAI,CAAAA,SAAAA,CARYH,CAQZG,CAAAA,GARiBJ,CAQjBI,GAAAA,CARsBH,CAQtBG,CAAAA,SAAAA,CARkCJ,CAQlCI,CAAAA,GARuCH,CAQvCG,GAR2CJ,CAQ3CI,GAR+CH,CAQ/CG;;;;AAA4EA,KAJ1NF,4BAI0NE,CAAAA,UAAAA,SAJ1K1B,eAI0K0B,EAAAA,CAAAA,GAJrJlD,CAIqJkD,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAJxHlD,CAIwHkD,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAJtEf,KAIsEe,SAJxD1B,eAIwD0B,GAJtCd,IAIsCc,SAAAA,SAJhB1B,eAIgB0B,EAAAA,GAJIL,iBAIJK,CAJsBR,2BAItBQ,CAJkDf,KAIlDe,CAAAA,EAJ0DF,4BAI1DE,CAJuFd,IAIvFc,CAAAA,CAAAA,GAJgGR,2BAIhGQ,CAJ4Hf,KAI5He,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;AAAD;AACrO;AAAuChD,KAD3B+C,iBAC2B/C,CAAAA,sBADaA,iBACbA,GADiCrB,gBACjCqB,CAAAA,GADqDgD,aACrDhD,SAD2ErB,gBAC3EqB,GAD8FlB,oBAC9FkB,CADmHgD,aACnHhD,CAAAA,GADoIgD,aACpIhD,SAD0JA,iBAC1JA,GAD8K4B,gBAC9K5B,CAD+LgD,aAC/LhD,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAoBrB,KAA/CiD,gBAA+CjD,CAAAA,UAApBqB,iBAAoBrB,GAAAA,gBAAAA,CAAAA,GAAoBiE,CAApBjE,SAA8BqB,iBAA9BrB,GAAkDiE,CAAlDjE,GAAsDiE,CAAtDjE,SAAgEA,gBAAhEA,GAAmFM,cAAnFN,CAAkGK,2BAAlGL,CAA8HiE,CAA9HjE,CAAAA,CAAAA,GAAAA,KAAAA;AAAoBiE,KACnEK,gBADmEL,CAAAA,UACxC5C,iBADwC4C,GACpBjE,gBADoBiE,GAAAA,SAAAA,CAAAA,GACYA,CADZA,SACsB5C,iBADtB4C,GAC0CjE,gBAD1CiE,GAC6DhB,gBAD7DgB,CAC8EA,CAD9EA,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA"}
+{"version":3,"file":"types.d.cts","names":["InteropZodObject","InteropZodDefault","InteropZodOptional","InferInteropZodInput","InferInteropZodOutput","InteropZodToStateDefinition","AnnotationRoot","AIMessage","SystemMessage","ToolMessage","ToolCall","Command","ClientTool","ServerTool","JumpToTarget","Runtime","AgentBuiltInState","ModelRequest","PromiseOrValue","T","Promise","AnyAnnotationRoot","NormalizedSchemaInput","TSchema","Record","MiddlewareResult","TState","ToolCallRequest","TContext","ToolCallHandler","WrapToolCallHook","WrapModelCallHandler","Omit","WrapModelCallHook","BeforeAgentHandler","Partial","BeforeAgentHook","BeforeModelHandler","BeforeModelHook","AfterModelHandler","AfterModelHook","AfterAgentHandler","AfterAgentHook","MIDDLEWARE_BRAND","AgentMiddleware","TContextSchema","TFullContext","FilterPrivateProps","K","InferChannelType","ToAnnotationRoot","InferMiddlewareState","S","InferMiddlewareInputState","InferMiddlewareStates","First","Rest","InferMiddlewareInputStates","InferMergedState","InferMergedInputState","InferMiddlewareContext","C","InferMiddlewareContextInput","Inner","InferMiddlewareContexts","MergeContextTypes","A","B","InferMiddlewareContextInputs","InferContextInput","ContextSchema","InferSchemaInput"],"sources":["../../../src/agents/middleware/types.d.ts"],"sourcesContent":["import type { InteropZodObject, InteropZodDefault, InteropZodOptional, InferInteropZodInput, InferInteropZodOutput } from \"@langchain/core/utils/types\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport type { AnnotationRoot } from \"@langchain/langgraph\";\nimport type { AIMessage, SystemMessage, ToolMessage } from \"@langchain/core/messages\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\nimport type { Command } from \"@langchain/langgraph\";\nimport type { ClientTool, ServerTool } from \"@langchain/core/tools\";\nimport type { JumpToTarget } from \"../constants.js\";\nimport type { Runtime, AgentBuiltInState } from \"../runtime.js\";\nimport type { ModelRequest } from \"../nodes/types.js\";\ntype PromiseOrValue<T> = T | Promise<T>;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\nexport type NormalizedSchemaInput<TSchema extends InteropZodObject | undefined | never = any> = [TSchema] extends [never] ? AgentBuiltInState : TSchema extends InteropZodObject ? InferInteropZodOutput<TSchema> & AgentBuiltInState : TSchema extends Record<string, unknown> ? TSchema & AgentBuiltInState : AgentBuiltInState;\n/**\n * Result type for middleware functions.\n */\nexport type MiddlewareResult<TState> = (TState & {\n    jumpTo?: JumpToTarget;\n}) | void;\n/**\n * Represents a tool call request for the wrapToolCall hook.\n * Contains the tool call information along with the agent's current state and runtime.\n */\nexport interface ToolCallRequest<TState extends Record<string, unknown> = Record<string, unknown>, TContext = unknown> {\n    /**\n     * The tool call to be executed\n     */\n    toolCall: ToolCall;\n    /**\n     * The BaseTool instance being invoked.\n     * Provides access to tool metadata like name, description, schema, etc.\n     */\n    tool: ClientTool | ServerTool;\n    /**\n     * The current agent state (includes both middleware state and built-in state).\n     */\n    state: TState & AgentBuiltInState;\n    /**\n     * The runtime context containing metadata, signal, writer, interrupt, etc.\n     */\n    runtime: Runtime<TContext>;\n}\n/**\n * Handler function type for wrapping tool calls.\n * Takes a tool call request and returns the tool result or a command.\n */\nexport type ToolCallHandler<TSchema extends Record<string, unknown> = AgentBuiltInState, TContext = unknown> = (request: ToolCallRequest<TSchema, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Wrapper function type for the wrapToolCall hook.\n * Allows middleware to intercept and modify tool execution.\n */\nexport type WrapToolCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ToolCallRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: ToolCallHandler<NormalizedSchemaInput<TSchema>, TContext>) => PromiseOrValue<ToolMessage | Command>;\n/**\n * Handler function type for wrapping model calls.\n * Takes a model request and returns the AI message response.\n *\n * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime\n * @returns The AI message response from the model\n */\nexport type WrapModelCallHandler<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: Omit<ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, \n/**\n * allow to reset the system prompt or system message\n */\n\"systemPrompt\" | \"systemMessage\"> & {\n    systemPrompt?: string;\n    systemMessage?: SystemMessage;\n}) => PromiseOrValue<AIMessage>;\n/**\n * Wrapper function type for the wrapModelCall hook.\n * Allows middleware to intercept and modify model execution.\n * This enables you to:\n * - Modify the request before calling the model (e.g., change system prompt, add/remove tools)\n * - Handle errors and retry with different parameters\n * - Post-process the response\n * - Implement custom caching, logging, or other cross-cutting concerns\n *\n * @param request - The model request containing all parameters needed for the model call\n * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response\n * @returns The AI message response from the model (or a modified version)\n */\nexport type WrapModelCallHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = (request: ModelRequest<NormalizedSchemaInput<TSchema>, TContext>, handler: WrapModelCallHandler<TSchema, TContext>) => PromiseOrValue<AIMessage>;\n/**\n * Handler function type for the beforeAgent hook.\n * Called once at the start of agent invocation before any model calls or tool executions.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the start of the agent invocation.\n */\nexport type BeforeAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the beforeModel hook.\n * Called before the model is invoked and before the wrapModelCall hook.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype BeforeModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the beforeModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called before each model invocation.\n */\nexport type BeforeModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: BeforeModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterModel hook.\n * Called after the model is invoked and before any tools are called.\n * Allows modifying the agent state after model invocation, e.g., to update tool call parameters.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterModelHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterModel lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called after each model invocation.\n */\nexport type AfterModelHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterModelHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Handler function type for the afterAgent hook.\n * Called once at the end of agent invocation after all model calls and tool executions are complete.\n *\n * @param state - The current agent state (includes both middleware state and built-in state)\n * @param runtime - The runtime context containing metadata, signal, writer, interrupt, etc.\n * @returns A middleware result containing partial state updates or undefined to pass through\n */\ntype AfterAgentHandler<TSchema, TContext> = (state: TSchema, runtime: Runtime<TContext>) => PromiseOrValue<MiddlewareResult<Partial<TSchema>>>;\n/**\n * Hook type for the afterAgent lifecycle event.\n * Can be either a handler function or an object with a handler and optional jump targets.\n * This hook is called once at the end of the agent invocation.\n */\nexport type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TContext = unknown> = AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext> | {\n    hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;\n    canJumpTo?: JumpToTarget[];\n};\n/**\n * Unique symbol used to brand middleware instances.\n * This prevents functions from being accidentally assignable to AgentMiddleware\n * since functions have a 'name' property that would otherwise make them structurally compatible.\n */\nexport declare const MIDDLEWARE_BRAND: unique symbol;\n/**\n * Base middleware interface.\n */\nexport interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {\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     * The name of the middleware.\n     */\n    name: string;\n    /**\n     * The schema of the middleware state. Middleware state is persisted between multiple invocations. It can be either:\n     * - A Zod object\n     * - A Zod optional object\n     * - A Zod default object\n     * - Undefined\n     */\n    stateSchema?: TSchema;\n    /**\n     * The schema of the middleware context. Middleware context is read-only and not persisted between multiple invocations. It can be either:\n     * - A Zod object\n     * - A Zod optional object\n     * - A Zod default object\n     * - Undefined\n     */\n    contextSchema?: TContextSchema;\n    /**\n     * Additional tools registered by the middleware.\n     */\n    tools?: (ClientTool | ServerTool)[];\n    /**\n     * Wraps tool execution with custom logic. This allows you to:\n     * - Modify tool call parameters before execution\n     * - Handle errors and retry with different parameters\n     * - Post-process tool results\n     * - Implement caching, logging, authentication, or other cross-cutting concerns\n     * - Return Command objects for advanced control flow\n     *\n     * The handler receives a ToolCallRequest containing the tool call, state, and runtime,\n     * along with a handler function to execute the actual tool.\n     *\n     * @param request - The tool call request containing toolCall, state, and runtime.\n     * @param handler - The function that executes the tool. Call this with a ToolCallRequest to get the result.\n     * @returns The tool result as a ToolMessage or a Command for advanced control flow.\n     *\n     * @example\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   console.log(`Calling tool: ${request.tool.name}`);\n     *   console.log(`Tool description: ${request.tool.description}`);\n     *\n     *   try {\n     *     // Execute the tool\n     *     const result = await handler(request);\n     *     console.log(`Tool ${request.tool.name} succeeded`);\n     *     return result;\n     *   } catch (error) {\n     *     console.error(`Tool ${request.tool.name} failed:`, error);\n     *     // Could return a custom error message or retry\n     *     throw error;\n     *   }\n     * }\n     * ```\n     *\n     * @example Authentication\n     * ```ts\n     * wrapToolCall: async (request, handler) => {\n     *   // Check if user is authorized for this tool\n     *   if (!request.runtime.context.isAuthorized(request.tool.name)) {\n     *     return new ToolMessage({\n     *       content: \"Unauthorized to call this tool\",\n     *       tool_call_id: request.toolCall.id,\n     *     });\n     *   }\n     *   return handler(request);\n     * }\n     * ```\n     *\n     * @example Caching\n     * ```ts\n     * const cache = new Map();\n     * wrapToolCall: async (request, handler) => {\n     *   const cacheKey = `${request.tool.name}:${JSON.stringify(request.toolCall.args)}`;\n     *   if (cache.has(cacheKey)) {\n     *     return cache.get(cacheKey);\n     *   }\n     *   const result = await handler(request);\n     *   cache.set(cacheKey, result);\n     *   return result;\n     * }\n     * ```\n     */\n    wrapToolCall?: WrapToolCallHook<TSchema, TFullContext>;\n    /**\n     * Wraps the model invocation with custom logic. This allows you to:\n     * - Modify the request before calling the model\n     * - Handle errors and retry with different parameters\n     * - Post-process the response\n     * - Implement custom caching, logging, or other cross-cutting concerns\n     *\n     * @param request - The model request containing model, messages, systemPrompt, tools, state, and runtime.\n     * @param handler - The function that invokes the model. Call this with a ModelRequest to get the response.\n     * @returns The response from the model (or a modified version).\n     *\n     * @example\n     * ```ts\n     * wrapModelCall: async (request, handler) => {\n     *   // Modify request before calling\n     *   const modifiedRequest = { ...request, systemPrompt: \"You are helpful\" };\n     *\n     *   try {\n     *     // Call the model\n     *     return await handler(modifiedRequest);\n     *   } catch (error) {\n     *     // Handle errors and retry with fallback\n     *     const fallbackRequest = { ...request, model: fallbackModel };\n     *     return await handler(fallbackRequest);\n     *   }\n     * }\n     * ```\n     */\n    wrapModelCall?: WrapModelCallHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the agent execution starts. This function is called once at the start of the agent invocation.\n     * It allows to modify the state of the agent before any model calls or tool executions.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeAgent?: BeforeAgentHook<TSchema, TFullContext>;\n    /**\n     * The function to run before the model call. This function is called before the model is invoked and before the `wrapModelCall` hook.\n     * It allows to modify the state of the agent.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    beforeModel?: BeforeModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the model call. This function is called after the model is invoked and before any tools are called.\n     * It allows to modify the state of the agent after the model is invoked, e.g. to update tool call parameters.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterModel?: AfterModelHook<TSchema, TFullContext>;\n    /**\n     * The function to run after the agent execution completes. This function is called once at the end of the agent invocation.\n     * It allows to modify the final state of the agent after all model calls and tool executions are complete.\n     *\n     * @param state - The middleware state\n     * @param runtime - The middleware runtime\n     * @returns The modified middleware state or undefined to pass through\n     */\n    afterAgent?: AfterAgentHook<TSchema, TFullContext>;\n}\n/**\n * Helper type to filter out properties that start with underscore (private properties)\n */\ntype FilterPrivateProps<T> = {\n    [K in keyof T as K extends `_${string}` ? never : K]: T[K];\n};\nexport type InferChannelType<T extends AnyAnnotationRoot | InteropZodObject> = T extends AnyAnnotationRoot ? ToAnnotationRoot<T>[\"State\"] : T extends InteropZodObject ? InferInteropZodInput<T> : {};\n/**\n * Helper type to infer the state schema type from a middleware\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<S>> : {} : {};\n/**\n * Helper type to infer the input state schema type from a middleware (all properties optional)\n * This filters out private properties (those starting with underscore)\n */\nexport type InferMiddlewareInputState<T extends AgentMiddleware> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<S>> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (just the middleware states)\n */\nexport type InferMiddlewareStates<T = AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareState<First> & InferMiddlewareStates<Rest> : InferMiddlewareState<First> : {} : {};\n/**\n * Helper type to infer merged input state from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareInputStates<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareInputState<First> & InferMiddlewareInputStates<Rest> : InferMiddlewareInputState<First> : {} : {};\n/**\n * Helper type to infer merged state from an array of middleware (includes built-in state)\n */\nexport type InferMergedState<T extends readonly AgentMiddleware[]> = InferMiddlewareStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer merged input state from an array of middleware (includes built-in state)\n */\nexport type InferMergedInputState<T extends readonly AgentMiddleware[]> = InferMiddlewareInputStates<T> & AgentBuiltInState;\n/**\n * Helper type to infer the context schema type from a middleware\n */\nexport type InferMiddlewareContext<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer the input context schema type from a middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInput<T extends AgentMiddleware> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};\n/**\n * Helper type to infer merged context from an array of middleware\n */\nexport type InferMiddlewareContexts<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? InferMiddlewareContext<First> & InferMiddlewareContexts<Rest> : InferMiddlewareContext<First> : {} : {};\n/**\n * Helper to merge two context types, preserving undefined unions\n */\ntype MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined] ? undefined : B | undefined : [B] extends [undefined] ? A | undefined : [A] extends [B] ? A : [B] extends [A] ? B : A & B;\n/**\n * Helper type to infer merged input context from an array of middleware (with optional defaults)\n */\nexport type InferMiddlewareContextInputs<T extends readonly AgentMiddleware[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware ? Rest extends readonly AgentMiddleware[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};\n/**\n * Helper type to extract input type from context schema (with optional defaults)\n */\nexport type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>[\"State\"] : {};\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>[\"State\"] : {};\nexport {};\n//# sourceMappingURL=types.d.ts.map"],"mappings":";;;;;;;;;;;KAUKkB,oBAAoBC,IAAIC,QAAQD;AAAhCD,KACOG,iBAAAA,GAAoBf,cADb,CAAA,GAAA,CAAA;AAAMa,KAEbG,qBAFaH,CAAAA,gBAEyBnB,gBAFzBmB,GAAAA,SAAAA,GAAAA,KAAAA,GAAAA,GAAAA,CAAAA,GAAAA,CAEwEI,OAFxEJ,CAAAA,SAAAA,CAAAA,KAAAA,CAAAA,GAEmGH,iBAFnGG,GAEuHI,OAFvHJ,SAEuInB,gBAFvImB,GAE0Jf,qBAF1Je,CAEgLI,OAFhLJ,CAAAA,GAE2LH,iBAF3LG,GAE+MI,OAF/MJ,SAE+NK,MAF/NL,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAEyPI,OAFzPJ,GAEmQH,iBAFnQG,GAEuRH,iBAFvRG;;;AAAW;AACxBE,KAKAI,gBALiB,CAAA,MAAA,CAAGnB,GAAAA,CAKQoB,MALRpB,GAAAA;EACpBgB,MAAAA,CAAAA,EAKCR,YALDQ;CAAsCtB,CAAAA,GAAAA,IAAAA;;;;;AAAuJuB,UAWxLI,eAXwLJ,CAAAA,eAWzJC,MAXyJD,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAW/HC,MAX+HD,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,WAAAA,OAAAA,CAAAA,CAAAA;EAAtBnB;;;EAAqEoB,QAAAA,EAe1Od,UAf0Oc;EAA0BD;;;AAA+C;EAIrTE,IAAAA,EAgBFb,UAhBEa,GAgBWZ,UAhBKa;EAOXC;;;EAIHjB,KAAAA,EASHgB,MATGhB,GASMM,iBATNN;EAKJE;;;EAIUI,OAAAA,EAIPD,OAJOC,CAICY,QAJDZ,CAAAA;;;AAIA;AAMpB;;AAAsEA,KAA1Da,eAA0Db,CAAAA,gBAA1BQ,MAA0BR,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,GAAAA,iBAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAAmDW,eAAnDX,CAAmEO,OAAnEP,EAA4EY,QAA5EZ,CAAAA,EAAAA,GAA0FE,cAA1FF,CAAyGP,WAAzGO,GAAuHL,OAAvHK,CAAAA;;;;;AAAuHL,KAKjLmB,gBALiLnB,CAAAA,gBAKhJX,gBALgJW,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAKtEgB,eALsEhB,CAKtDW,qBALsDX,CAKhCY,OALgCZ,CAAAA,EAKtBiB,QALsBjB,CAAAA,EAAAA,OAAAA,EAKFkB,eALElB,CAKcW,qBALdX,CAKoCY,OALpCZ,CAAAA,EAK8CiB,QAL9CjB,CAAAA,EAAAA,GAK4DO,cAL5DP,CAK2EF,WAL3EE,GAKyFA,OALzFA,CAAAA;;AAAf;AAK9K;;;;;AAAuHgB,KAQ3GI,oBAR2GJ,CAAAA,gBAQtE3B,gBARsE2B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAQIK,IARJL,CAQSV,YARTU,CAQsBL,qBARtBK,CAQ4CJ,OAR5CI,CAAAA,EAQsDC,QARtDD,CAAAA;;;;cAAoEE,GAAAA,eAAAA,CAAAA,GAAAA;EAA6EpB,YAAAA,CAAAA,EAAAA,MAAAA;EAAcE,aAAAA,CAAAA,EAclQH,aAdkQG;CAA7BO,EAAAA,GAenPA,cAfmPA,CAepOX,SAfoOW,CAAAA;AAAc;AAQvQ;;;;;;;;;;AAOoB;AAcpB;AAA8ClB,KAAlCiC,iBAAkCjC,CAAAA,gBAAAA,gBAAAA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAAAA,CAAAA,OAAAA,EAA0EiB,YAA1EjB,CAAuFsB,qBAAvFtB,CAA6GuB,OAA7GvB,CAAAA,EAAuH4B,QAAvH5B,CAAAA,EAAAA,OAAAA,EAA2I+B,oBAA3I/B,CAAgKuB,OAAhKvB,EAAyK4B,QAAzK5B,CAAAA,EAAAA,GAAuLkB,cAAvLlB,CAAsMO,SAAtMP,CAAAA;;;;;;;;;KASzCkC,kBATgOhB,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAShLK,OATgLL,EAAAA,OAAAA,EAS9JH,OAT8JG,CAStJU,QATsJV,CAAAA,EAAAA,GASxIA,cATwIA,CASzHO,gBATyHP,CASxGiB,OATwGjB,CAShGK,OATgGL,CAAAA,CAAAA,CAAAA;AAAc;AAAY;;;;AAS1HK,KAMzHa,eANyHb,CAAAA,gBAMzFvB,gBANyFuB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAMzBW,kBANyBX,CAMND,qBANMC,CAMgBA,OANhBA,CAAAA,EAM0BK,QAN1BL,CAAAA,GAAAA;EAARY,IAAAA,EAOnHD,kBAPmHC,CAOhGb,qBAPgGa,CAO1EZ,OAP0EY,CAAAA,EAOhEP,QAPgEO,CAAAA;EAAjBV,SAAAA,CAAAA,EAQ5FX,YAR4FW,EAAAA;CAAfP;AAAc;AAM3G;;;;;;;KAYKmB,kBAXwBf,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAWwBC,OAXxBD,EAAAA,OAAAA,EAW0CP,OAX1CO,CAWkDM,QAXlDN,CAAAA,EAAAA,GAWgEJ,cAXhEI,CAW+EG,gBAX/EH,CAWgGa,OAXhGb,CAWwGC,OAXxGD,CAAAA,CAAAA,CAAAA;;;;AACD;AAC1B;AASmDC,KAMzCe,eANyCf,CAAAA,gBAMTvB,gBANSuB,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAMuDc,kBANvDd,CAM0ED,qBAN1EC,CAMgGA,OANhGA,CAAAA,EAM0GK,QAN1GL,CAAAA,GAAAA;EAA0BK,IAAAA,EAOrES,kBAPqET,CAOlDN,qBAPkDM,CAO5BL,OAP4BK,CAAAA,EAOlBA,QAPkBA,CAAAA;EAARb,SAAAA,CAAAA,EAQvDD,YARuDC,EAAAA;CAA8DQ;;;;AAA1B;AAM3G;;;;;KAaKgB,iBAbuGF,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAaxDd,OAbwDc,EAAAA,OAAAA,EAatCtB,OAbsCsB,CAa9BT,QAb8BS,CAAAA,EAAAA,GAahBnB,cAbgBmB,CAaDZ,gBAbCY,CAagBF,OAbhBE,CAawBd,OAbxBc,CAAAA,CAAAA,CAAAA;;;;;;AAEhF,KAiBhBG,cAjBgB,CAAA,gBAiBexC,gBAjBf,GAAA,SAAA,GAAA,SAAA,EAAA,WAAA,OAAA,CAAA,GAiB+EuC,iBAjB/E,CAiBiGjB,qBAjBjG,CAiBuHC,OAjBvH,CAAA,EAiBiIK,QAjBjI,CAAA,GAAA;EAWvBW,IAAAA,EAOKA,iBAPY,CAOMjB,qBAPNM,CAO4BL,OAP5B,CAAA,EAOsCK,QAPtC,CAAA;EAA8BL,SAAAA,CAAAA,EAQpCT,YARoCS,EAAAA;CAA0BK;;;;;;AAA4B;AAM1G;;KAYKa,iBAZ8IlB,CAAAA,OAAAA,EAAAA,QAAAA,CAAAA,GAAAA,CAAAA,KAAAA,EAY/FA,OAZ+FA,EAAAA,OAAAA,EAY7ER,OAZ6EQ,CAYrEK,QAZqEL,CAAAA,EAAAA,GAYvDL,cAZuDK,CAYxCE,gBAZwCF,CAYvBY,OAZuBZ,CAYfA,OAZeA,CAAAA,CAAAA,CAAAA;;;;;;AACvFK,KAiBhDc,cAjBgDd,CAAAA,gBAiBjB5B,gBAjBiB4B,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,WAAAA,OAAAA,CAAAA,GAiB+Ca,iBAjB/Cb,CAiBiEN,qBAjBjEM,CAiBuFL,OAjBvFK,CAAAA,EAiBiGA,QAjBjGA,CAAAA,GAAAA;EAAlDW,IAAAA,EAkBAE,iBAlBAF,CAkBkBjB,qBAlBlBiB,CAkBwChB,OAlBxCgB,CAAAA,EAkBkDX,QAlBlDW,CAAAA;EACMzB,SAAAA,CAAAA,EAkBAA,YAlBAA,EAAAA;AAAY,CAAA;AAC1B;;;;;AAS0HqB,cAevGQ,gBAfuGR,EAAAA,OAAAA,MAAAA;;;AAAlB;AAM9FO,UAaKE,eAbSrB,CAAAA,gBAauBvB,gBAbvB,GAAA,SAAA,GAAA,GAAA,EAAA,uBAakFA,gBAblF,GAaqGC,iBAbrG,CAauHD,gBAbvH,CAAA,GAa2IE,kBAb3I,CAa8JF,gBAb9J,CAAA,GAAA,SAAA,GAAA,GAAA,EAAA,eAAA,GAAA,CAAA,CAAA;EAAiBA;;;;EAAgEyC,UAkB7FE,gBAAAA,CAlB6FF,EAAAA,IAAAA;EACzDlB;;;EAAxCkB,IAAAA,EAAAA,MAAAA;EACM3B;AAAY;AAO5B;AAIA;;;;EAA+Hb,WAAAA,CAAAA,EAiB7GsB,OAjB6GtB;EAAyDD;;;;;;;EA4FpJuB,aAAAA,CAAAA,EAnEhBsB,cAmEgBtB;EAASuB;;;EA6BEA,KAAAA,CAAAA,EAAAA,CA5FlClC,UA4FkCkC,GA5FrBjC,UA4FqBiC,CAAAA,EAAAA;EAA3Bb;;;;;;;;;;;;;AAoCW;AAC9B;;;;;;AAK4D;AAE7D;;;;;;;;;;;AAA6L;AAK7L;;;;;;;;;AAA4J;AAK5J;;;;;;;;;AAAiK;AAIjK;;;;;;;;;EAAoNkB,YAAAA,CAAAA,EAvFjMrB,gBAuFiMqB,CAvFhL5B,OAuFgL4B,EAvFvKL,YAuFuKK,CAAAA;EAAoDK;;;;AAA4B;AAIpS;;;;;;;;;;;;;;AAAuU;AAIvU;;;;;AAAiH;AAIjH;;EAAqGrC,aAAAA,CAAAA,EAtEjFc,iBAsEiFd,CAtE/DI,OAsE+DJ,EAtEtD2B,YAsEsD3B,CAAAA;EAA3BsC;;AAAiD;AAI3H;;;;;EAAyHzD,WAAAA,CAAAA,EAjEvGoC,eAiEuGpC,CAjEvFuB,OAiEuFvB,EAjE9E8C,YAiE8E9C,CAAAA;EAAwC6D;;AAAD;AAIhK;;;;;EAA8H3D,WAAAA,CAAAA,EA5D5GoC,eA4D4GpC,CA5D5FqB,OA4D4FrB,EA5DnF4C,YA4DmF5C,CAAAA;EAAuD6D;;;;;;AAAsE;AAI3P;EAAuDnB,UAAAA,CAAAA,EAvDtCJ,cAuDsCI,CAvDvBrB,OAuDuBqB,EAvDdE,YAuDcF,CAAAA;EAAqBzB;;;;;;;;EAAiNqC,UAAAA,CAAAA,EA9C5Qd,cA8C4Qc,CA9C7PjC,OA8C6PiC,EA9CpPV,YA8CoPU,CAAAA;;;;AAA8B;AAAkB,KAzCxUT,kBA6CAkB,CAAiB,CAAA,CAAA,GAAA,QAAUC,MA5ChB/C,CA4CgB+C,IA5CXlB,CA4CWkB,SAAAA,IAAAA,MAAAA,EAAAA,GAAAA,KAAAA,GA5CsBlB,CA4CtBkB,GA5C0B/C,CA4C1B+C,CA5C4BlB,CA4C5BkB,CAAAA,EAA0BC;AAAqCA,KA1CnFlB,gBA0CmFkB,CAAAA,UA1CxD9C,iBA0CwD8C,GA1CpCnE,gBA0CoCmE,CAAAA,GA1ChBhD,CA0CgBgD,SA1CN9C,iBA0CM8C,GA1CcjB,gBA0CdiB,CA1C+BhD,CA0C/BgD,CAAAA,CAAAA,OAAAA,CAAAA,GA1C6ChD,CA0C7CgD,SA1CuDnE,gBA0CvDmE,GA1C0EhE,oBA0C1EgE,CA1C+FhD,CA0C/FgD,CAAAA,GAAAA,CAAAA,CAAAA;;;;;AAA4ED,KArC/Jf,oBAqC+Je,CAAAA,UArChItB,eAqCgIsB,CAAAA,GArC7G/C,CAqC6G+C,SArCnGtB,eAqCmGsB,CAAAA,KAAAA,EAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GArC9Dd,CAqC8Dc,SArCpDlE,gBAqCoDkE,GArCjCnB,kBAqCiCmB,CArCd9D,qBAqCc8D,CArCQd,CAqCRc,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;;AAA8BC,KAhC7Ld,yBAgC6Lc,CAAAA,UAhCzJvB,eAgCyJuB,CAAAA,GAhCtIhD,CAgCsIgD,SAhC5HvB,eAgC4HuB,CAAAA,KAAAA,EAAAA,EAAAA,GAAAA,EAAAA,GAAAA,CAAAA,GAhCvFf,CAgCuFe,SAhC7EnE,gBAgC6EmE,GAhC1DpB,kBAgC0DoB,CAhCvChE,oBAgCuCgE,CAhClBf,CAgCkBe,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAC;AAI1M;;AAAiFhD,KAhCrEmC,qBAgCqEnC,CAAAA,IAhC3CyB,eAgC2CzB,EAAAA,CAAAA,GAhCtBA,CAgCsBA,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAhCOA,CAgCPA,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAhCyDoC,KAgCzDpC,SAhCuEyB,eAgCvEzB,GAhCyFqC,IAgCzFrC,SAAAA,SAhC+GyB,eAgC/GzB,EAAAA,GAhCmIgC,oBAgCnIhC,CAhCwJoC,KAgCxJpC,CAAAA,GAhCiKmC,qBAgCjKnC,CAhCuLqC,IAgCvLrC,CAAAA,GAhC+LgC,oBAgC/LhC,CAhCoNoC,KAgCpNpC,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA+GqC,KA5BpLC,0BA4BoLD,CAAAA,UAAAA,SA5BtIZ,eA4BsIY,EAAAA,CAAAA,GA5BjHrC,CA4BiHqC,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GA5BpFrC,CA4BoFqC,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GA5BlCD,KA4BkCC,SA5BpBZ,eA4BoBY,GA5BFA,IA4BEA,SAAAA,SA5BoBZ,eA4BpBY,EAAAA,GA5BwCH,yBA4BxCG,CA5BkED,KA4BlEC,CAAAA,GA5B2EC,0BA4B3ED,CA5BsGA,IA4BtGA,CAAAA,GA5B8GH,yBA4B9GG,CA5BwID,KA4BxIC,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA6HA,KAxBjTE,gBAwBiTF,CAAAA,UAAAA,SAxB7QZ,eAwB6QY,EAAAA,CAAAA,GAxBxPF,qBAwBwPE,CAxBlOrC,CAwBkOqC,CAAAA,GAxB7NxC,iBAwB6NwC;;;;AAASM,KApB1TH,qBAoB0TG,CAAAA,UAAAA,SApBjRlB,eAoBiRkB,EAAAA,CAAAA,GApB5PL,0BAoB4PK,CApBjO3C,CAoBiO2C,CAAAA,GApB5N9C,iBAoB4N8C;AAA2B;AAIjW;;AAAwE9D,KApB5D4D,sBAoB4D5D,CAAAA,UApB3B4C,eAoB2B5C,CAAAA,GApBRmB,CAoBQnB,SApBE4C,eAoBF5C,CAAAA,GAAAA,EAAAA,KAAAA,EAAAA,EAAAA,GAAAA,CAAAA,GApBuC6D,CAoBvC7D,SApBiDA,gBAoBjDA,GApBoEG,oBAoBpEH,CApByF6D,CAoBzF7D,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA6DG,KAhBzH2D,2BAgByH3D,CAAAA,UAhBnFyC,eAgBmFzC,CAAAA,GAhBhEgB,CAgBgEhB,SAhBtDyC,eAgBsDzC,CAAAA,GAAAA,EAAAA,KAAAA,EAAAA,EAAAA,GAAAA,CAAAA,GAhBjB0D,CAgBiB1D,SAhBPD,kBAgBOC,CAAAA,KAAAA,MAAAA,CAAAA,GAhB2BA,oBAgB3BA,CAhBgD4D,KAgBhD5D,CAAAA,GAAAA,SAAAA,GAhBqE0D,CAgBrE1D,SAhB+EH,gBAgB/EG,GAhBkGA,oBAgBlGA,CAhBuH0D,CAgBvH1D,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAAgF+C,KAZzMc,uBAYyMd,CAAAA,UAAAA,SAZ9JN,eAY8JM,EAAAA,CAAAA,GAZzI/B,CAYyI+B,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAZ5G/B,CAY4G+B,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAZ1DK,KAY0DL,SAZ5CN,eAY4CM,GAZ1BM,IAY0BN,SAAAA,SAZJN,eAYIM,EAAAA,GAZgBU,sBAYhBV,CAZuCK,KAYvCL,CAAAA,GAZgDc,uBAYhDd,CAZwEM,IAYxEN,CAAAA,GAZgFU,sBAYhFV,CAZuGK,KAYvGL,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAgB;AACrO;;KATKe,iBASsDjE,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CAT3BkE,CAS2BlE,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,CATDmE,CASCnE,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAAAA,SAAAA,GAToCmE,CASpCnE,GAAAA,SAAAA,GAAAA,CATqDmE,CASrDnE,CAAAA,SAAAA,CAAAA,SAAAA,CAAAA,GAT8EkE,CAS9ElE,GAAAA,SAAAA,GAAAA,CAT+FkE,CAS/FlE,CAAAA,SAAAA,CAT2GmE,CAS3GnE,CAAAA,GATgHkE,CAShHlE,GAAAA,CATqHmE,CASrHnE,CAAAA,SAAAA,CATiIkE,CASjIlE,CAAAA,GATsImE,CAStInE,GAT0IkE,CAS1IlE,GAT8ImE,CAS9InE;;;;AAAsDkE,KALrGE,4BAKqGF,CAAAA,UAAAA,SALrDtB,eAKqDsB,EAAAA,CAAAA,GALhC/C,CAKgC+C,SAAAA,SAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GALH/C,CAKG+C,SAAAA,SAAAA,CAAAA,KAAAA,MAAAA,EAAAA,GAAAA,KAAAA,KAAAA,CAAAA,GAL+CX,KAK/CW,SAL6DtB,eAK7DsB,GAL+EV,IAK/EU,SAAAA,SALqGtB,eAKrGsB,EAAAA,GALyHD,iBAKzHC,CAL2IJ,2BAK3II,CALuKX,KAKvKW,CAAAA,EAL+KE,4BAK/KF,CAL4MV,IAK5MU,CAAAA,CAAAA,GALqNJ,2BAKrNI,CALiPX,KAKjPW,CAAAA,GAAAA,CAAAA,CAAAA,GAAAA,CAAAA,CAAAA;;;;AAA6B5D,KADlI+D,iBACkI/D,CAAAA,sBAD1Fe,iBAC0Ff,GADtEN,gBACsEM,CAAAA,GADlDgE,aACkDhE,SAD5BN,gBAC4BM,GADTH,oBACSG,CADYgE,aACZhE,CAAAA,GAD6BgE,aAC7BhE,SADmDe,iBACnDf,GADuE4C,gBACvE5C,CADwFgE,aACxFhE,CAAAA,CAAAA,OAAAA,CAAAA,GAAAA,CAAAA,CAAAA;AAAc,KAAhJ4C,gBAAgJ,CAAA,UAArH7B,iBAAqH,GAAjGrB,gBAAiG,CAAA,GAA7EkE,CAA6E,SAAnE7C,iBAAmE,GAA/C6C,CAA+C,GAA3CA,CAA2C,SAAjClE,gBAAiC,GAAdM,cAAc,CAACD,2BAAD,CAA6B6D,CAA7B,CAAA,CAAA,GAAA,KAAA;AAChJK,KAAAA,gBAAgB,CAAAL,UAAW7C,iBAAX,GAA+BrB,gBAA/B,GAAA,SAAA,CAAA,GAA+DkE,CAA/D,SAAyE7C,iBAAzE,GAA6FrB,gBAA7F,GAAgHkD,gBAAhH,CAAiIgB,CAAjI,CAAA,CAAA,OAAA,CAAA,GAAA,CAAA,CAAA"}

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

@@ -153,10 +153,21 @@ type AfterAgentHook<TSchema extends InteropZodObject | undefined = undefined, TC
   hook: AfterAgentHandler<NormalizedSchemaInput<TSchema>, TContext>;
   canJumpTo?: JumpToTarget[];
 };
+/**
+ * Unique symbol used to brand middleware instances.
+ * This prevents functions from being accidentally assignable to AgentMiddleware
+ * since functions have a 'name' property that would otherwise make them structurally compatible.
+ */
+declare const MIDDLEWARE_BRAND: unique symbol;
 /**
  * Base middleware interface.
  */
 interface AgentMiddleware<TSchema extends InteropZodObject | undefined = any, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = any, TFullContext = any> {
+  /**
+   * Brand property to distinguish middleware instances from plain objects or functions.
+   * This is required and prevents accidental assignment of functions to middleware arrays.
+   */
+  readonly [MIDDLEWARE_BRAND]: true;
   /**
    * The name of the middleware.
    */
@@ -368,5 +379,5 @@ type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObjec
 type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;
 type InferSchemaInput<A extends AnyAnnotationRoot | InteropZodObject | undefined> = A extends AnyAnnotationRoot | InteropZodObject ? ToAnnotationRoot<A>["State"] : {};
 //#endregion
-export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContexts, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareState, InferMiddlewareStates, InferSchemaInput, MiddlewareResult, NormalizedSchemaInput, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
+export { AfterAgentHook, AfterModelHook, AgentMiddleware, AnyAnnotationRoot, BeforeAgentHook, BeforeModelHook, InferChannelType, InferContextInput, InferMergedInputState, InferMergedState, InferMiddlewareContext, InferMiddlewareContextInput, InferMiddlewareContextInputs, InferMiddlewareContexts, InferMiddlewareInputState, InferMiddlewareInputStates, InferMiddlewareState, InferMiddlewareStates, InferSchemaInput, MIDDLEWARE_BRAND, MiddlewareResult, NormalizedSchemaInput, ToAnnotationRoot, ToolCallHandler, ToolCallRequest, WrapModelCallHandler, WrapModelCallHook, WrapToolCallHook };
 //# sourceMappingURL=types.d.ts.map