langchain 1.5.4 → 1.5.6

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

@@ -1 +1 @@
-{"version":3,"file":"summarization.js","names":["z","z4","uuid"],"sources":["../../../src/agents/middleware/summarization.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { z as z4 } from \"zod/v4\";\nimport { v4 as uuid } from \"@langchain/core/utils/uuid\";\nimport {\n  BaseMessage,\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n  RemoveMessage,\n  trimMessages,\n  HumanMessage,\n  getBufferString,\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 {\n  mergeConfigs,\n  pickRunnableConfigKeys,\n  type RunnableConfig,\n} from \"@langchain/core/runnables\";\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\";\nimport type { Runtime } from \"../runtime.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        runtime\n      );\n\n      const summaryMessage = new HumanMessage({\n        content: `${summaryPrefix}\\n\\n${summary}`,\n        id: uuid(),\n        additional_kwargs: { lc_source: \"summarization\" },\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 AI/Tool pairs.\n   * If cutoff lands on ToolMessage, move backward to include the AIMessage.\n   */\n  const safeCutoff = findSafeCutoffPoint(messages, cutoffCandidate);\n\n  /**\n   * If findSafeCutoffPoint moved forward (fallback case), verify it's safe.\n   * If it moved backward, we already have a safe point.\n   */\n  if (safeCutoff <= cutoffCandidate) {\n    return safeCutoff;\n  }\n\n  /**\n   * Fallback: iterate backward to find a safe cutoff\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  /**\n   * First, try to find a safe cutoff point using findSafeCutoffPoint.\n   * This handles the case where cutoff lands on a ToolMessage by moving\n   * backward to include the corresponding AIMessage.\n   */\n  const safeCutoff = findSafeCutoffPoint(messages, targetCutoff);\n\n  /**\n   * If findSafeCutoffPoint moved backward (found matching AIMessage), use it.\n   */\n  if (safeCutoff <= targetCutoff) {\n    return safeCutoff;\n  }\n\n  /**\n   * Fallback: iterate backward to find a safe cutoff\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 * Find a safe cutoff point that doesn't split AI/Tool message pairs.\n *\n * If the message at `cutoffIndex` is a `ToolMessage`, search backward for the\n * `AIMessage` containing the corresponding `tool_calls` and adjust the cutoff to\n * include it. This ensures tool call requests and responses stay together.\n *\n * Falls back to advancing forward past `ToolMessage` objects only if no matching\n * `AIMessage` is found (edge case).\n */\nfunction findSafeCutoffPoint(\n  messages: BaseMessage[],\n  cutoffIndex: number\n): number {\n  if (\n    cutoffIndex >= messages.length ||\n    !ToolMessage.isInstance(messages[cutoffIndex])\n  ) {\n    return cutoffIndex;\n  }\n\n  // Collect tool_call_ids from consecutive ToolMessages at/after cutoff\n  const toolCallIds = new Set<string>();\n  let idx = cutoffIndex;\n  while (idx < messages.length && ToolMessage.isInstance(messages[idx])) {\n    const toolMsg = messages[idx] as ToolMessage;\n    if (toolMsg.tool_call_id) {\n      toolCallIds.add(toolMsg.tool_call_id);\n    }\n    idx++;\n  }\n\n  // Search backward for AIMessage with matching tool_calls\n  for (let i = cutoffIndex - 1; i >= 0; i--) {\n    const msg = messages[i];\n    if (AIMessage.isInstance(msg) && hasToolCalls(msg)) {\n      const aiToolCallIds = extractToolCallIds(msg as AIMessage);\n      // Check if there's any overlap between the tool_call_ids\n      for (const id of toolCallIds) {\n        if (aiToolCallIds.has(id)) {\n          // Found the AIMessage - move cutoff to include it\n          return i;\n        }\n      }\n    }\n  }\n\n  // Fallback: no matching AIMessage found, advance past ToolMessages to avoid\n  // orphaned tool responses\n  return idx;\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 *\n * @param messagesToSummarize - Messages to summarize.\n * @param model - The language model to use for summarization.\n * @param summaryPrompt - The prompt template for summarization.\n * @param tokenCounter - Function to count tokens.\n * @param trimTokensToSummarize - Optional token limit for trimming messages.\n * @param runtime - The runtime environment, used to inherit config so that\n *   LangGraph's handlers can properly track and tag the summarization model call.\n */\nasync function createSummary(\n  messagesToSummarize: BaseMessage[],\n  model: BaseLanguageModel,\n  summaryPrompt: string,\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined,\n  runtime: Runtime<unknown>\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  /**\n   * Format messages using getBufferString to avoid token inflation from metadata\n   * when str() / JSON.stringify is called on message objects.\n   * This produces compact output like:\n   * ```\n   * Human: What's the weather?\n   * AI: Let me check...[tool_calls]\n   * Tool: 72°F and sunny\n   * ```\n   */\n  const formattedMessages = getBufferString(trimmedMessages);\n\n  try {\n    const formattedPrompt = summaryPrompt.replace(\n      \"{messages}\",\n      formattedMessages\n    );\n    /**\n     * Merge parent runnable config with summarization metadata so LangGraph's\n     * stream handlers (and other callback-based consumers) can properly track\n     * and tag the summarization model call.\n     */\n    const baseConfig: RunnableConfig = pickRunnableConfigKeys(runtime) ?? {};\n    const config = mergeConfigs(baseConfig, {\n      metadata: { lc_source: \"summarization\" },\n    });\n    const response = await model.invoke(formattedPrompt, config);\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":";;;;;;;;;;;;;AAkCA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BtC,MAAM,yBAAyB;AAC/B,MAAM,2BAA2B;AACjC,MAAM,2BAA2B;AACjC,MAAM,iCAAiC;AACvC,MAAM,8BAA8B;AAEpC,MAAM,qBAAqBA,IACxB,UAAU,CACV,KAAKA,IAAE,MAAMA,IAAE,QAAqB,CAAC,CAAC,CACtC,QAAQA,IAAE,MAAM,CAACA,IAAE,QAAQ,EAAEA,IAAE,QAAQA,IAAE,QAAQ,CAAC,CAAC,CAAC,CAAC;AAKxD,MAAa,oBAAoBA,IAC9B,OAAO;;;;CAIN,UAAUA,IACP,QAAQ,CACR,GAAG,GAAG,kCAAkC,CACxC,IAAI,GAAG,2CAA2C,CAClD,UAAU;;;;CAIb,QAAQA,IAAE,QAAQ,CAAC,SAAS,gCAAgC,CAAC,UAAU;;;;CAIvE,UAAUA,IACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,SAAS,kCAAkC,CAC3C,UAAU;CACd,CAAC,CACD,QACE,SAAS;AAIR,QAHc;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;EAAS,CAAC,QACvD,MAAM,MAAM,KAAA,EACd,CAAC,UACc;GAElB,EACE,SAAS,kEACV,CACF;AAGH,MAAa,aAAaA,IACvB,OAAO;;;;CAIN,UAAUA,IACP,QAAQ,CACR,IAAI,GAAG,gCAAgC,CACvC,IAAI,GAAG,2CAA2C,CAClD,UAAU;;;;CAIb,QAAQA,IACL,QAAQ,CACR,IAAI,GAAG,4CAA4C,CACnD,UAAU;CACb,UAAUA,IACP,QAAQ,CACR,IAAI,8BAA8B,CAClC,IAAI,GAAG,gCAAgC,CACvC,UAAU;CACd,CAAC,CACD,QACE,SAAS;AAIR,QAHc;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;EAAS,CAAC,QACvD,MAAM,MAAM,KAAA,EACd,CAAC,WACe;GAEnB,EACE,SAAS,iEACV,CACF;AAGH,MAAM,gBAAgBA,IAAE,OAAO;;;;CAI7B,OAAOA,IAAE,QAAoC;;;;;;;;;;;;;;;;;CAiB7C,SAASA,IAAE,MAAM,CAAC,mBAAmBA,IAAE,MAAM,kBAAkB,CAAC,CAAC,CAAC,UAAU;;;;CAI5E,MAAM,WAAW,UAAU;;;;CAI3B,cAAc,mBAAmB,UAAU;;;;;CAK3C,eAAeA,IAAE,QAAQ,CAAC,QAAQ,uBAAuB;;;;CAIzD,uBAAuBA,IAAE,QAAQ,CAAC,UAAU;;;;CAI5C,eAAeA,IAAE,QAAQ,CAAC,UAAU;;;;CAIpC,wBAAwBA,IAAE,QAAQ,CAAC,UAAU;;;;CAI7C,gBAAgBA,IAAE,QAAQ,CAAC,UAAU;CACtC,CAAC;;;;AASF,SAAgB,iBAAiB,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,KAAA;AAIzC,KAAI,WAAW,SAAS,OAAO,MAAM,UAAU,SAC7C,QAAO,oBAAoB,MAAM,MAAM;AAEzC,KAAI,eAAe,SAAS,OAAO,MAAM,cAAc,SACrD,QAAO,oBAAoB,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAoD/C,SAAgB,wBACd,SACA;;;;CAIA,MAAM,EAAE,MAAM,aAAa,UAAU,iBAAiB,eAAe,QAAQ;AAC7E,KAAI,MACF,OAAM,IAAI,MACR,6CAA6CC,EAAG,cAAc,MAAM,GACrE;AAGH,QAAO,iBAAiB;EACtB,MAAM;EACN,eAAe,cAAc,OAAO;;;;;AAKlC,OAAOD,IAAE,QAA2B,CAAC,UAAU,EAChD,CAAC;EACF,aAAa,OAAO,OAAO,YAAY;GACrC,IAAI,UACF,YAAY;GACd,IAAI,OAAoB,YAAY;;;;AAOpC,OAAI,YAAY,2BAA2B,KAAA,GAAW;AACpD,YAAQ,KACN,kFACD;AACD,QAAI,YAAY,KAAA,EACd,WAAU,EAAE,QAAQ,YAAY,wBAAwB;;;;;AAO5D,OAAI,YAAY,mBAAmB,KAAA,GAAW;AAC5C,YAAQ,KACN,yEACD;AACD,QACE,CAAC,QACA,QACC,cAAc,QACd,KAAK,aAAa,yBAEpB,QAAO,EAAE,UAAU,YAAY,gBAAgB;;;;;GAOnD,MAAM,kBACJ,QAAQ,SAAS,YAAY,KAAA,IACzB,QAAQ,QAAQ,UAChB;GACN,MAAM,eACJ,QAAQ,SAAS,SAAS,KAAA,IACtB,QAAQ,QAAQ,OACf,QAAQ,EAAE,UAAU,0BAA0B;GAErD,MAAM,gBAAgB,WAAW,MAAM,aAAa;;;;GAKpD,IAAI,oBAAmC,EAAE;AACzC,OAAI,oBAAoB,KAAA,EACtB,qBAAoB,EAAE;YACb,MAAM,QAAQ,gBAAgB;;;;AAIvC,uBAAqB,gBAAkC,KAAK,MAC1D,kBAAkB,MAAM,EAAE,CAC3B;;;;;AAKD,uBAAoB,CAAC,kBAAkB,MAAM,gBAAgB,CAAC;;;;GAMhE,MAAM,kBACJ,kBAAkB,MAAM,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,2GAED;GAGH,MAAM,gBACJ,QAAQ,SAAS,kBAAkB,yBAC9B,YAAY,iBAAiB,yBAC7B,QAAQ,SAAS,iBAClB,YAAY,iBACZ;GACN,MAAM,gBACJ,QAAQ,QAAQ,iBAChB,YAAY,iBACZ;GACF,MAAM,wBACJ,QAAQ,SAAS,0BAA0B,KAAA,IACvC,QAAQ,QAAQ,wBACf,YAAY,yBAAyB;;;;AAK5C,oBAAiB,MAAM,SAAS;GAEhC,MAAM,eACJ,QAAQ,SAAS,iBAAiB,KAAA,IAC9B,QAAQ,QAAQ,eACf,YAAY,gBAAgB;GACnC,MAAM,cAAc,MAAM,aAAa,MAAM,SAAS;AAQtD,OAAI,CAAC,MAPqB,gBACxB,MAAM,UACN,aACA,mBACA,MACD,CAGC;GAGF,MAAM,EAAE,cAAc,yBAAyB,mBAC7C,MAAM,SACP;GACD,MAAM,cAAc,MAAM,qBACxB,sBACA,eACA,cACA,MACD;AAED,OAAI,eAAe,EACjB;GAGF,MAAM,EAAE,qBAAqB,sBAAsB,kBACjD,cACA,sBACA,YACD;GAWD,MAAM,iBAAiB,IAAI,aAAa;IACtC,SAAS,GAAG,cAAc,MAAM,MAVZ,cACpB,qBACA,OACA,eACA,cACA,uBACA,QACD;IAIC,IAAIE,IAAM;IACV,mBAAmB,EAAE,WAAW,iBAAiB;IAClD,CAAC;AAEF,UAAO,EACL,UAAU;IACR,IAAI,cAAc,EAAE,IAAI,qBAAqB,CAAC;IAC9C;IACA,GAAG;IACJ,EACF;;EAEJ,CAAC;;;;;AAMJ,SAAS,iBAAiB,UAA+B;AACvD,MAAK,MAAM,OAAO,SAChB,KAAI,CAAC,IAAI,GACP,KAAI,KAAKA,IAAM;;;;;AAQrB,SAAS,mBAAmB,UAG1B;AACA,KAAI,SAAS,SAAS,KAAK,cAAc,WAAW,SAAS,GAAG,CAC9D,QAAO;EACL,cAAc,SAAS;EACvB,sBAAsB,SAAS,MAAM,EAAE;EACxC;AAEH,QAAO,EACL,sBAAsB,UACvB;;;;;AAMH,SAAS,kBACP,cACA,sBACA,aAC0E;CAC1E,MAAM,sBAAsB,qBAAqB,MAAM,GAAG,YAAY;CACtE,MAAM,oBAAoB,qBAAqB,MAAM,YAAY;AAGjE,KAAI,aACF,qBAAoB,QAAQ,aAAa;AAG3C,QAAO;EAAE;EAAqB;EAAmB;;;;;;;;;;;;AAanD,eAAe,gBACb,UACA,aACA,mBACA,OACkB;AAClB,KAAI,kBAAkB,WAAW,EAC/B,QAAO;;;;AAMT,MAAK,MAAM,WAAW,mBAAmB;;;;EAIvC,IAAI,eAAe;EACnB,IAAI,iBAAiB;AAErB,MAAI,QAAQ,aAAa,KAAA,GAAW;AAClC,oBAAiB;AACjB,OAAI,SAAS,SAAS,QAAQ,SAC5B,gBAAe;;AAInB,MAAI,QAAQ,WAAW,KAAA,GAAW;AAChC,oBAAiB;AACjB,OAAI,cAAc,QAAQ,OACxB,gBAAe;;AAInB,MAAI,QAAQ,aAAa,KAAA,GAAW;AAClC,oBAAiB;GACjB,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,OAAI,OAAO,mBAAmB;QAExB,cADc,KAAK,MAAM,iBAAiB,QAAQ,SAC3B,CACzB,gBAAe;;;;;AAMjB,kBAAe;;;;;AAOnB,MAAI,kBAAkB,aACpB,QAAO;;AAIX,QAAO;;;;;AAMT,eAAe,qBACb,UACA,MACA,cACA,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;;;;;AAK3D,QAAO,eAAe,UAAU,KAAK,YAAY,yBAAyB;;;;;AAM5E,eAAe,qBACb,UACA,MACA,cACA,OAC6B;AAC7B,KAAI,SAAS,WAAW,EACtB,QAAO;CAGT,IAAI;AAEJ,KAAI,cAAc,QAAQ,KAAK,aAAa,KAAA,GAAW;EACrD,MAAM,iBAAiB,iBAAiB,MAAM;AAC9C,MAAI,OAAO,mBAAmB,SAC5B;AAEF,qBAAmB,KAAK,MAAM,iBAAiB,KAAK,SAAS;YACpD,YAAY,QAAQ,KAAK,WAAW,KAAA,EAC7C,oBAAmB,KAAK,MAAM,KAAK,OAAO;KAE1C;AAGF,KAAI,oBAAoB,EACtB,oBAAmB;AAIrB,KAAI,MADsB,aAAa,SAAS,IAC7B,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;AAE1C,MAAI,MADuB,aAAa,SAAS,MAAM,IAAI,CAAC,IACxC,kBAAkB;AACpC,qBAAkB;AAClB,WAAQ;QAER,QAAO,MAAM;;AAIjB,KAAI,oBAAoB,SAAS,OAC/B,mBAAkB;AAGpB,KAAI,mBAAmB,SAAS,QAAQ;AACtC,MAAI,SAAS,WAAW,EACtB,QAAO;AAET,oBAAkB,SAAS,SAAS;;;;;;CAOtC,MAAM,aAAa,oBAAoB,UAAU,gBAAgB;;;;;AAMjE,KAAI,cAAc,gBAChB,QAAO;;;;AAMT,MAAK,IAAI,IAAI,iBAAiB,KAAK,GAAG,IACpC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;;;;;AAMT,SAAS,eACP,UACA,gBACQ;AACR,KAAI,SAAS,UAAU,eACrB,QAAO;CAGT,MAAM,eAAe,SAAS,SAAS;;;;;;CAOvC,MAAM,aAAa,oBAAoB,UAAU,aAAa;;;;AAK9D,KAAI,cAAc,aAChB,QAAO;;;;AAMT,MAAK,IAAI,IAAI,cAAc,KAAK,GAAG,IACjC,KAAI,kBAAkB,UAAU,EAAE,CAChC,QAAO;AAIX,QAAO;;;;;AAMT,SAAS,kBACP,UACA,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;;AAIX,QAAO;;;;;AAMT,SAAS,mBAAmB,WAAmC;CAC7D,MAAM,8BAAc,IAAI,KAAa;AACrC,KAAI,UAAU,WACZ,MAAK,MAAM,YAAY,UAAU,YAAY;EAC3C,MAAM,KACJ,OAAO,aAAa,YAAY,QAAQ,WAAW,SAAS,KAAK;AACnE,MAAI,GACF,aAAY,IAAI,GAAG;;AAIzB,QAAO;;;;;;;;;;;;AAaT,SAAS,oBACP,UACA,aACQ;AACR,KACE,eAAe,SAAS,UACxB,CAAC,YAAY,WAAW,SAAS,aAAa,CAE9C,QAAO;CAIT,MAAM,8BAAc,IAAI,KAAa;CACrC,IAAI,MAAM;AACV,QAAO,MAAM,SAAS,UAAU,YAAY,WAAW,SAAS,KAAK,EAAE;EACrE,MAAM,UAAU,SAAS;AACzB,MAAI,QAAQ,aACV,aAAY,IAAI,QAAQ,aAAa;AAEvC;;AAIF,MAAK,IAAI,IAAI,cAAc,GAAG,KAAK,GAAG,KAAK;EACzC,MAAM,MAAM,SAAS;AACrB,MAAI,UAAU,WAAW,IAAI,IAAI,aAAa,IAAI,EAAE;GAClD,MAAM,gBAAgB,mBAAmB,IAAiB;AAE1D,QAAK,MAAM,MAAM,YACf,KAAI,cAAc,IAAI,GAAG,CAEvB,QAAO;;;AAQf,QAAO;;;;;AAMT,SAAS,wBACP,UACA,gBACA,aACA,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;OAEd,iBAAiB,gBACf,IAAI,YAE3B,QAAO;;;AAIb,QAAO;;;;;;;;;;;;;AAcT,eAAe,cACb,qBACA,OACA,eACA,cACA,uBACA,SACiB;AACjB,KAAI,CAAC,oBAAoB,OACvB,QAAO;CAGT,MAAM,kBAAkB,MAAM,uBAC5B,qBACA,cACA,sBACD;AAED,KAAI,CAAC,gBAAgB,OACnB,QAAO;;;;;;;;;;;CAaT,MAAM,oBAAoB,gBAAgB,gBAAgB;AAE1D,KAAI;EACF,MAAM,kBAAkB,cAAc,QACpC,cACA,kBACD;EAOD,MAAM,SAAS,aADoB,uBAAuB,QAAQ,IAAI,EAAE,EAChC,EACtC,UAAU,EAAE,WAAW,iBAAiB,EACzC,CAAC;EAEF,MAAM,WAAU,MADO,MAAM,OAAO,iBAAiB,OAAO,EACnC;;;;AAIzB,MAAI,OAAO,YAAY,SACrB,QAAO,QAAQ,MAAM;WACZ,MAAM,QAAQ,QAAQ,CAa/B,QAToB,QACjB,KAAK,SAAS;AACb,OAAI,OAAO,SAAS,SAAU,QAAO;AACrC,OAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,KACzD,QAAQ,KAA0B;AAEpC,UAAO;IACP,CACD,KAAK,GACU,CAAC,MAAM;AAE3B,SAAO;UACA,GAAG;AACV,SAAO,6BAA6B;;;;;;AAOxC,eAAe,uBACb,UACA,cACA,uBACwB;AACxB,KAAI,0BAA0B,KAAA,EAC5B,QAAO;AAGT,KAAI;AACF,SAAO,MAAM,aAAa,UAAU;GAClC,WAAW;GACX,cAAc,OAAO,SAAS,aAAa,KAAK;GAChD,UAAU;GACV,cAAc;GACd,eAAe;GAChB,CAAC;SACI;;;;AAIN,SAAO,SAAS,MAAM,CAAC,+BAA+B"}
+{"version":3,"file":"summarization.js","names":["z","z4","uuid"],"sources":["../../../src/agents/middleware/summarization.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { z as z4 } from \"zod/v4\";\nimport { v4 as uuid } from \"@langchain/core/utils/uuid\";\nimport {\n  BaseMessage,\n  AIMessage,\n  SystemMessage,\n  ToolMessage,\n  RemoveMessage,\n  trimMessages,\n  HumanMessage,\n  getBufferString,\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 {\n  mergeConfigs,\n  pickRunnableConfigKeys,\n  type RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { REMOVE_ALL_MESSAGES } from \"@langchain/langgraph\";\nimport { createMiddleware } from \"../middleware.js\";\nimport { INTERNAL_CALL_TAG } from \"./constants.js\";\nimport { countTokensApproximately } from \"./utils.js\";\nimport { hasToolCalls } from \"../utils.js\";\nimport { initChatModel } from \"../../chat_models/universal.js\";\nimport type { Runtime } from \"../runtime.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        runtime\n      );\n\n      // Reuse a replaced message's id so the summary projects as an update, not a new message.\n      const summaryMessage = new HumanMessage({\n        content: `${summaryPrefix}\\n\\n${summary}`,\n        id: conversationMessages[0].id,\n        additional_kwargs: { lc_source: \"summarization\" },\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 AI/Tool pairs.\n   * If cutoff lands on ToolMessage, move backward to include the AIMessage.\n   */\n  const safeCutoff = findSafeCutoffPoint(messages, cutoffCandidate);\n\n  /**\n   * If findSafeCutoffPoint moved forward (fallback case), verify it's safe.\n   * If it moved backward, we already have a safe point.\n   */\n  if (safeCutoff <= cutoffCandidate) {\n    return safeCutoff;\n  }\n\n  /**\n   * Fallback: iterate backward to find a safe cutoff\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  /**\n   * First, try to find a safe cutoff point using findSafeCutoffPoint.\n   * This handles the case where cutoff lands on a ToolMessage by moving\n   * backward to include the corresponding AIMessage.\n   */\n  const safeCutoff = findSafeCutoffPoint(messages, targetCutoff);\n\n  /**\n   * If findSafeCutoffPoint moved backward (found matching AIMessage), use it.\n   */\n  if (safeCutoff <= targetCutoff) {\n    return safeCutoff;\n  }\n\n  /**\n   * Fallback: iterate backward to find a safe cutoff\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 * Find a safe cutoff point that doesn't split AI/Tool message pairs.\n *\n * If the message at `cutoffIndex` is a `ToolMessage`, search backward for the\n * `AIMessage` containing the corresponding `tool_calls` and adjust the cutoff to\n * include it. This ensures tool call requests and responses stay together.\n *\n * Falls back to advancing forward past `ToolMessage` objects only if no matching\n * `AIMessage` is found (edge case).\n */\nfunction findSafeCutoffPoint(\n  messages: BaseMessage[],\n  cutoffIndex: number\n): number {\n  if (\n    cutoffIndex >= messages.length ||\n    !ToolMessage.isInstance(messages[cutoffIndex])\n  ) {\n    return cutoffIndex;\n  }\n\n  // Collect tool_call_ids from consecutive ToolMessages at/after cutoff\n  const toolCallIds = new Set<string>();\n  let idx = cutoffIndex;\n  while (idx < messages.length && ToolMessage.isInstance(messages[idx])) {\n    const toolMsg = messages[idx] as ToolMessage;\n    if (toolMsg.tool_call_id) {\n      toolCallIds.add(toolMsg.tool_call_id);\n    }\n    idx++;\n  }\n\n  // Search backward for AIMessage with matching tool_calls\n  for (let i = cutoffIndex - 1; i >= 0; i--) {\n    const msg = messages[i];\n    if (AIMessage.isInstance(msg) && hasToolCalls(msg)) {\n      const aiToolCallIds = extractToolCallIds(msg as AIMessage);\n      // Check if there's any overlap between the tool_call_ids\n      for (const id of toolCallIds) {\n        if (aiToolCallIds.has(id)) {\n          // Found the AIMessage - move cutoff to include it\n          return i;\n        }\n      }\n    }\n  }\n\n  // Fallback: no matching AIMessage found, advance past ToolMessages to avoid\n  // orphaned tool responses\n  return idx;\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 *\n * @param messagesToSummarize - Messages to summarize.\n * @param model - The language model to use for summarization.\n * @param summaryPrompt - The prompt template for summarization.\n * @param tokenCounter - Function to count tokens.\n * @param trimTokensToSummarize - Optional token limit for trimming messages.\n * @param runtime - The runtime environment, used to inherit config so that\n *   LangGraph's handlers can properly track and tag the summarization model call.\n */\nasync function createSummary(\n  messagesToSummarize: BaseMessage[],\n  model: BaseLanguageModel,\n  summaryPrompt: string,\n  tokenCounter: TokenCounter,\n  trimTokensToSummarize: number | undefined,\n  runtime: Runtime<unknown>\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  /**\n   * Format messages using getBufferString to avoid token inflation from metadata\n   * when str() / JSON.stringify is called on message objects.\n   * This produces compact output like:\n   * ```\n   * Human: What's the weather?\n   * AI: Let me check...[tool_calls]\n   * Tool: 72°F and sunny\n   * ```\n   */\n  const formattedMessages = getBufferString(trimmedMessages);\n\n  try {\n    const formattedPrompt = summaryPrompt.replace(\n      \"{messages}\",\n      formattedMessages\n    );\n    /**\n     * Merge parent runnable config with summarization metadata so LangGraph's\n     * stream handlers (and other callback-based consumers) can properly track\n     * and tag the summarization model call.\n     */\n    const baseConfig: RunnableConfig = pickRunnableConfigKeys(runtime) ?? {};\n    const config = mergeConfigs(baseConfig, {\n      metadata: { lc_source: \"summarization\" },\n      tags: [INTERNAL_CALL_TAG],\n    });\n    const response = await model.invoke(formattedPrompt, config);\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":";;;;;;;;;;;;;;AAmCA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BtC,MAAM,yBAAyB;AAC/B,MAAM,2BAA2B;AACjC,MAAM,2BAA2B;AAEjC,MAAM,8BAA8B;AAEpC,MAAM,qBAAqBA,IACxB,SAAS,CAAC,CACV,KAAKA,IAAE,MAAMA,IAAE,OAAoB,CAAC,CAAC,CAAC,CACtC,QAAQA,IAAE,MAAM,CAACA,IAAE,OAAO,GAAGA,IAAE,QAAQA,IAAE,OAAO,CAAC,CAAC,CAAC,CAAC;AAKvD,MAAa,oBAAoBA,IAC9B,OAAO;;;;CAIN,UAAUA,IACP,OAAO,CAAC,CACR,GAAG,GAAG,iCAAiC,CAAC,CACxC,IAAI,GAAG,0CAA0C,CAAC,CAClD,SAAS;;;;CAIZ,QAAQA,IAAE,OAAO,CAAC,CAAC,SAAS,+BAA+B,CAAC,CAAC,SAAS;;;;CAItE,UAAUA,IACP,OAAO,CAAC,CACR,IAAI,6BAA6B,CAAC,CAClC,SAAS,iCAAiC,CAAC,CAC3C,SAAS;AACd,CAAC,CAAC,CACD,QACE,SAAS;CAIR,OAHc;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAQ,CAAC,CAAC,QACvD,MAAM,MAAM,KAAA,CACf,CAAC,CAAC,UACc;AAClB,GACA,EACE,SAAS,iEACX,CACF;AAGF,MAAa,aAAaA,IACvB,OAAO;;;;CAIN,UAAUA,IACP,OAAO,CAAC,CACR,IAAI,GAAG,+BAA+B,CAAC,CACvC,IAAI,GAAG,0CAA0C,CAAC,CAClD,SAAS;;;;CAIZ,QAAQA,IACL,OAAO,CAAC,CACR,IAAI,GAAG,2CAA2C,CAAC,CACnD,SAAS;CACZ,UAAUA,IACP,OAAO,CAAC,CACR,IAAI,6BAA6B,CAAC,CAClC,IAAI,GAAG,+BAA+B,CAAC,CACvC,SAAS;AACd,CAAC,CAAC,CACD,QACE,SAAS;CAIR,OAHc;EAAC,KAAK;EAAU,KAAK;EAAQ,KAAK;CAAQ,CAAC,CAAC,QACvD,MAAM,MAAM,KAAA,CACf,CAAC,CAAC,WACe;AACnB,GACA,EACE,SAAS,gEACX,CACF;AAGF,MAAM,gBAAgBA,IAAE,OAAO;;;;CAI7B,OAAOA,IAAE,OAAmC;;;;;;;;;;;;;;;;;CAiB5C,SAASA,IAAE,MAAM,CAAC,mBAAmBA,IAAE,MAAM,iBAAiB,CAAC,CAAC,CAAC,CAAC,SAAS;;;;CAI3E,MAAM,WAAW,SAAS;;;;CAI1B,cAAc,mBAAmB,SAAS;;;;;CAK1C,eAAeA,IAAE,OAAO,CAAC,CAAC,QAAQ,sBAAsB;;;;CAIxD,uBAAuBA,IAAE,OAAO,CAAC,CAAC,SAAS;;;;CAI3C,eAAeA,IAAE,OAAO,CAAC,CAAC,SAAS;;;;CAInC,wBAAwBA,IAAE,OAAO,CAAC,CAAC,SAAS;;;;CAI5C,gBAAgBA,IAAE,OAAO,CAAC,CAAC,SAAS;AACtC,CAAC;;;;AASD,SAAgB,iBAAiB,OAA8C;CAE7E,IACE,aAAa,SACb,OAAO,MAAM,YAAY,YACzB,MAAM,WACN,oBAAoB,MAAM,YACzB,OAAO,MAAM,QAAQ,mBAAmB,YACvC,MAAM,QAAQ,kBAAkB,OAElC,OAAO,MAAM,QAAQ,kBAAkB,KAAA;CAIzC,IAAI,WAAW,SAAS,OAAO,MAAM,UAAU,UAC7C,OAAO,oBAAoB,MAAM,KAAK;CAExC,IAAI,eAAe,SAAS,OAAO,MAAM,cAAc,UACrD,OAAO,oBAAoB,MAAM,SAAS;AAI9C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDA,SAAgB,wBACd,SACA;;;;CAIA,MAAM,EAAE,MAAM,aAAa,UAAU,iBAAiB,eAAe,OAAO;CAC5E,IAAI,OACF,MAAM,IAAI,MACR,6CAA6CC,EAAG,cAAc,KAAK,GACrE;CAGF,OAAO,iBAAiB;EACtB,MAAM;EACN,eAAe,cAAc,OAAO;;;;;AAKlC,OAAOD,IAAE,OAA0B,CAAC,CAAC,SAAS,EAChD,CAAC;EACD,aAAa,OAAO,OAAO,YAAY;GACrC,IAAI,UACF,YAAY;GACd,IAAI,OAAoB,YAAY;;;;GAOpC,IAAI,YAAY,2BAA2B,KAAA,GAAW;IACpD,QAAQ,KACN,iFACF;IACA,IAAI,YAAY,KAAA,GACd,UAAU,EAAE,QAAQ,YAAY,uBAAuB;GAE3D;;;;GAKA,IAAI,YAAY,mBAAmB,KAAA,GAAW;IAC5C,QAAQ,KACN,wEACF;IACA,IACE,CAAC,QACA,QACC,cAAc,QACd,KAAK,aAAa,0BAEpB,OAAO,EAAE,UAAU,YAAY,eAAe;GAElD;;;;GAKA,MAAM,kBACJ,QAAQ,SAAS,YAAY,KAAA,IACzB,QAAQ,QAAQ,UAChB;GACN,MAAM,eACJ,QAAQ,SAAS,SAAS,KAAA,IACtB,QAAQ,QAAQ,OACf,QAAQ,EAAE,UAAU,yBAAyB;GAEpD,MAAM,gBAAgB,WAAW,MAAM,YAAY;;;;GAKnD,IAAI,oBAAmC,CAAC;GACxC,IAAI,oBAAoB,KAAA,GACtB,oBAAoB,CAAC;QAChB,IAAI,MAAM,QAAQ,eAAe;;;;GAItC,oBAAqB,gBAAkC,KAAK,MAC1D,kBAAkB,MAAM,CAAC,CAC3B;;;;;GAKA,oBAAoB,CAAC,kBAAkB,MAAM,eAAe,CAAC;;;;GAM/D,MAAM,kBACJ,kBAAkB,MAAM,MAAM,cAAc,CAAC,KAC7C,cAAc;GAEhB,MAAM,QACJ,OAAO,YAAY,UAAU,WACzB,MAAM,cAAc,YAAY,KAAK,IACrC,YAAY;GAElB,IAAI,mBAAmB,CAAC,iBAAiB,KAAK,GAC5C,MAAM,IAAI,MACR,0GAEF;GAGF,MAAM,gBACJ,QAAQ,SAAS,kBAAkB,yBAC9B,YAAY,iBAAiB,yBAC7B,QAAQ,SAAS,iBAClB,YAAY,iBACZ;GACN,MAAM,gBACJ,QAAQ,QAAQ,iBAChB,YAAY,iBACZ;GACF,MAAM,wBACJ,QAAQ,SAAS,0BAA0B,KAAA,IACvC,QAAQ,QAAQ,wBACf,YAAY,yBAAyB;;;;GAK5C,iBAAiB,MAAM,QAAQ;GAE/B,MAAM,eACJ,QAAQ,SAAS,iBAAiB,KAAA,IAC9B,QAAQ,QAAQ,eACf,YAAY,gBAAgB;GACnC,MAAM,cAAc,MAAM,aAAa,MAAM,QAAQ;GAQrD,IAAI,CAAC,MAPqB,gBACxB,MAAM,UACN,aACA,mBACA,KACF,GAGE;GAGF,MAAM,EAAE,cAAc,yBAAyB,mBAC7C,MAAM,QACR;GACA,MAAM,cAAc,MAAM,qBACxB,sBACA,eACA,cACA,KACF;GAEA,IAAI,eAAe,GACjB;GAGF,MAAM,EAAE,qBAAqB,sBAAsB,kBACjD,cACA,sBACA,WACF;GAYA,MAAM,iBAAiB,IAAI,aAAa;IACtC,SAAS,GAAG,cAAc,MAAM,MAXZ,cACpB,qBACA,OACA,eACA,cACA,uBACA,OACF;IAKE,IAAI,qBAAqB,EAAE,CAAC;IAC5B,mBAAmB,EAAE,WAAW,gBAAgB;GAClD,CAAC;GAED,OAAO,EACL,UAAU;IACR,IAAI,cAAc,EAAE,IAAI,oBAAoB,CAAC;IAC7C;IACA,GAAG;GACL,EACF;EACF;CACF,CAAC;AACH;;;;AAKA,SAAS,iBAAiB,UAA+B;CACvD,KAAK,MAAM,OAAO,UAChB,IAAI,CAAC,IAAI,IACP,IAAI,KAAKE,GAAK;AAGpB;;;;AAKA,SAAS,mBAAmB,UAG1B;CACA,IAAI,SAAS,SAAS,KAAK,cAAc,WAAW,SAAS,EAAE,GAC7D,OAAO;EACL,cAAc,SAAS;EACvB,sBAAsB,SAAS,MAAM,CAAC;CACxC;CAEF,OAAO,EACL,sBAAsB,SACxB;AACF;;;;AAKA,SAAS,kBACP,cACA,sBACA,aAC0E;CAC1E,MAAM,sBAAsB,qBAAqB,MAAM,GAAG,WAAW;CACrE,MAAM,oBAAoB,qBAAqB,MAAM,WAAW;CAGhE,IAAI,cACF,oBAAoB,QAAQ,YAAY;CAG1C,OAAO;EAAE;EAAqB;CAAkB;AAClD;;;;;;;;;;;AAYA,eAAe,gBACb,UACA,aACA,mBACA,OACkB;CAClB,IAAI,kBAAkB,WAAW,GAC/B,OAAO;;;;CAMT,KAAK,MAAM,WAAW,mBAAmB;;;;EAIvC,IAAI,eAAe;EACnB,IAAI,iBAAiB;EAErB,IAAI,QAAQ,aAAa,KAAA,GAAW;GAClC,iBAAiB;GACjB,IAAI,SAAS,SAAS,QAAQ,UAC5B,eAAe;EAEnB;EAEA,IAAI,QAAQ,WAAW,KAAA,GAAW;GAChC,iBAAiB;GACjB,IAAI,cAAc,QAAQ,QACxB,eAAe;EAEnB;EAEA,IAAI,QAAQ,aAAa,KAAA,GAAW;GAClC,iBAAiB;GACjB,MAAM,iBAAiB,iBAAiB,KAAK;GAC7C,IAAI,OAAO,mBAAmB,UAExB;QAAA,cADc,KAAK,MAAM,iBAAiB,QAAQ,QAC5B,GACxB,eAAe;GAAA;;;;GAMjB,eAAe;EAEnB;;;;EAKA,IAAI,kBAAkB,cACpB,OAAO;CAEX;CAEA,OAAO;AACT;;;;AAKA,eAAe,qBACb,UACA,MACA,cACA,OACiB;CACjB,IAAI,YAAY,QAAQ,cAAc,MAAM;EAC1C,MAAM,mBAAmB,MAAM,qBAC7B,UACA,MACA,cACA,KACF;EACA,IAAI,OAAO,qBAAqB,UAC9B,OAAO;;;;EAKT,OAAO,eAAe,UAAU,wBAAwB;CAC1D;;;;CAIA,OAAO,eAAe,UAAU,KAAK,YAAY,wBAAwB;AAC3E;;;;AAKA,eAAe,qBACb,UACA,MACA,cACA,OAC6B;CAC7B,IAAI,SAAS,WAAW,GACtB,OAAO;CAGT,IAAI;CAEJ,IAAI,cAAc,QAAQ,KAAK,aAAa,KAAA,GAAW;EACrD,MAAM,iBAAiB,iBAAiB,KAAK;EAC7C,IAAI,OAAO,mBAAmB,UAC5B;EAEF,mBAAmB,KAAK,MAAM,iBAAiB,KAAK,QAAQ;CAC9D,OAAO,IAAI,YAAY,QAAQ,KAAK,WAAW,KAAA,GAC7C,mBAAmB,KAAK,MAAM,KAAK,MAAM;MAEzC;CAGF,IAAI,oBAAoB,GACtB,mBAAmB;CAIrB,IAAI,MADsB,aAAa,QAAQ,KAC5B,kBACjB,OAAO;;;;;CAOT,IAAI,OAAO;CACX,IAAI,QAAQ,SAAS;CACrB,IAAI,kBAAkB,SAAS;CAC/B,MAAM,gBAAgB,KAAK,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC,IAAI;CAE/D,KAAK,IAAI,IAAI,GAAG,IAAI,eAAe,KAAK;EACtC,IAAI,QAAQ,OACV;EAGF,MAAM,MAAM,KAAK,OAAO,OAAO,SAAS,CAAC;EAEzC,IAAI,MADuB,aAAa,SAAS,MAAM,GAAG,CAAC,KACvC,kBAAkB;GACpC,kBAAkB;GAClB,QAAQ;EACV,OACE,OAAO,MAAM;CAEjB;CAEA,IAAI,oBAAoB,SAAS,QAC/B,kBAAkB;CAGpB,IAAI,mBAAmB,SAAS,QAAQ;EACtC,IAAI,SAAS,WAAW,GACtB,OAAO;EAET,kBAAkB,SAAS,SAAS;CACtC;;;;;CAMA,MAAM,aAAa,oBAAoB,UAAU,eAAe;;;;;CAMhE,IAAI,cAAc,iBAChB,OAAO;;;;CAMT,KAAK,IAAI,IAAI,iBAAiB,KAAK,GAAG,KACpC,IAAI,kBAAkB,UAAU,CAAC,GAC/B,OAAO;CAIX,OAAO;AACT;;;;AAKA,SAAS,eACP,UACA,gBACQ;CACR,IAAI,SAAS,UAAU,gBACrB,OAAO;CAGT,MAAM,eAAe,SAAS,SAAS;;;;;;CAOvC,MAAM,aAAa,oBAAoB,UAAU,YAAY;;;;CAK7D,IAAI,cAAc,cAChB,OAAO;;;;CAMT,KAAK,IAAI,IAAI,cAAc,KAAK,GAAG,KACjC,IAAI,kBAAkB,UAAU,CAAC,GAC/B,OAAO;CAIX,OAAO;AACT;;;;AAKA,SAAS,kBACP,UACA,aACS;CACT,IAAI,eAAe,SAAS,QAC1B,OAAO;;;;CAMT,IACE,cAAc,SAAS,UACvB,UAAU,WAAW,SAAS,YAAY,KAC1C,aAAa,SAAS,YAAY,GAElC,OAAO;CAGT,MAAM,cAAc,KAAK,IAAI,GAAG,cAAc,2BAA2B;CACzE,MAAM,YAAY,KAAK,IACrB,SAAS,QACT,cAAc,2BAChB;CAEA,KAAK,IAAI,IAAI,aAAa,IAAI,WAAW,KAAK;EAC5C,IAAI,CAAC,aAAa,SAAS,EAAE,GAC3B;EAGF,MAAM,cAAc,mBAAmB,SAAS,EAAe;EAC/D,IAAI,wBAAwB,UAAU,GAAG,aAAa,WAAW,GAC/D,OAAO;CAEX;CAEA,OAAO;AACT;;;;AAKA,SAAS,mBAAmB,WAAmC;CAC7D,MAAM,8BAAc,IAAI,IAAY;CACpC,IAAI,UAAU,YACZ,KAAK,MAAM,YAAY,UAAU,YAAY;EAC3C,MAAM,KACJ,OAAO,aAAa,YAAY,QAAQ,WAAW,SAAS,KAAK;EACnE,IAAI,IACF,YAAY,IAAI,EAAE;CAEtB;CAEF,OAAO;AACT;;;;;;;;;;;AAYA,SAAS,oBACP,UACA,aACQ;CACR,IACE,eAAe,SAAS,UACxB,CAAC,YAAY,WAAW,SAAS,YAAY,GAE7C,OAAO;CAIT,MAAM,8BAAc,IAAI,IAAY;CACpC,IAAI,MAAM;CACV,OAAO,MAAM,SAAS,UAAU,YAAY,WAAW,SAAS,IAAI,GAAG;EACrE,MAAM,UAAU,SAAS;EACzB,IAAI,QAAQ,cACV,YAAY,IAAI,QAAQ,YAAY;EAEtC;CACF;CAGA,KAAK,IAAI,IAAI,cAAc,GAAG,KAAK,GAAG,KAAK;EACzC,MAAM,MAAM,SAAS;EACrB,IAAI,UAAU,WAAW,GAAG,KAAK,aAAa,GAAG,GAAG;GAClD,MAAM,gBAAgB,mBAAmB,GAAgB;GAEzD,KAAK,MAAM,MAAM,aACf,IAAI,cAAc,IAAI,EAAE,GAEtB,OAAO;EAGb;CACF;CAIA,OAAO;AACT;;;;AAKA,SAAS,wBACP,UACA,gBACA,aACA,aACS;CACT,KAAK,IAAI,IAAI,iBAAiB,GAAG,IAAI,SAAS,QAAQ,KAAK;EACzD,MAAM,UAAU,SAAS;EACzB,IACE,YAAY,WAAW,OAAO,KAC9B,YAAY,IAAI,QAAQ,YAAY,GAEb;OAAA,iBAAiB,gBACf,IAAI,aAE3B,OAAO;EAAA;CAGb;CACA,OAAO;AACT;;;;;;;;;;;;AAaA,eAAe,cACb,qBACA,OACA,eACA,cACA,uBACA,SACiB;CACjB,IAAI,CAAC,oBAAoB,QACvB,OAAO;CAGT,MAAM,kBAAkB,MAAM,uBAC5B,qBACA,cACA,qBACF;CAEA,IAAI,CAAC,gBAAgB,QACnB,OAAO;;;;;;;;;;;CAaT,MAAM,oBAAoB,gBAAgB,eAAe;CAEzD,IAAI;EACF,MAAM,kBAAkB,cAAc,QACpC,cACA,iBACF;EAOA,MAAM,SAAS,aADoB,uBAAuB,OAAO,KAAK,CAAC,GAC/B;GACtC,UAAU,EAAE,WAAW,gBAAgB;GACvC,MAAM,CAAC,iBAAiB;EAC1B,CAAC;EAED,MAAM,WAAU,MADO,MAAM,OAAO,iBAAiB,MAAM,EAAA,CAClC;;;;EAIzB,IAAI,OAAO,YAAY,UACrB,OAAO,QAAQ,KAAK;OACf,IAAI,MAAM,QAAQ,OAAO,GAa9B,OAToB,QACjB,KAAK,SAAS;GACb,IAAI,OAAO,SAAS,UAAU,OAAO;GACrC,IAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,UAAU,MACzD,OAAQ,KAA0B;GAEpC,OAAO;EACT,CAAC,CAAC,CACD,KAAK,EACS,CAAC,CAAC,KAAK;EAE1B,OAAO;CACT,SAAS,GAAG;EACV,OAAO,6BAA6B;CACtC;AACF;;;;AAKA,eAAe,uBACb,UACA,cACA,uBACwB;CACxB,IAAI,0BAA0B,KAAA,GAC5B,OAAO;CAGT,IAAI;EACF,OAAO,MAAM,aAAa,UAAU;GAClC,WAAW;GACX,cAAc,OAAO,SAAS,aAAa,IAAI;GAC/C,UAAU;GACV,cAAc;GACd,eAAe;EACjB,CAAC;CACH,QAAQ;;;;EAIN,OAAO,SAAS,MAAM,GAA+B;CACvD;AACF"}

package/dist/agents/middleware/todoListMiddleware.cjs

@@ -1,4 +1,3 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 const require_middleware = require("../middleware.cjs");
 require("../index.cjs");
 let _langchain_core_messages = require("@langchain/core/messages");

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

@@ -1 +1 @@
-{"version":3,"file":"todoListMiddleware.cjs","names":["z","Command","ToolMessage","createMiddleware","AIMessage"],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { Command } from \"@langchain/langgraph\";\nimport { tool } from \"@langchain/core/tools\";\nimport { AIMessage, ToolMessage } from \"@langchain/core/messages\";\n\nimport { createMiddleware } from \"../index.js\";\n\n/**\n * Description for the write_todos tool\n * Ported exactly from Python WRITE_TODOS_DESCRIPTION\n */\nconst WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.\nIt also helps the user understand the progress of the task and overall progress of their requests.\nOnly use this tool if you think it will be helpful in staying organized. If the user's request is trivial and takes less than 3 steps, it is better to NOT use this tool and just do the task directly.\n\n## When to Use This Tool\nUse this tool in these scenarios:\n\n1. Complex multi-step tasks - When a task requires 3 or more distinct steps or actions\n2. Non-trivial and complex tasks - Tasks that require careful planning or multiple operations\n3. User explicitly requests todo list - When the user directly asks you to use the todo list\n4. User provides multiple tasks - When users provide a list of things to be done (numbered or comma-separated)\n5. The plan may need future revisions or updates based on results from the first few steps. Keeping track of this in a list is helpful.\n\n## How to Use This Tool\n1. When you start working on a task - Mark it as in_progress BEFORE beginning work.\n2. After completing a task - Mark it as completed and add any new follow-up tasks discovered during implementation.\n3. You can also update future tasks, such as deleting them if they are no longer necessary, or adding new tasks that are necessary. Don't change previously completed tasks.\n4. You can make several updates to the todo list at once. For example, when you complete a task, you can mark the next task you need to start as in_progress.\n\n## When NOT to Use This Tool\nIt is important to skip using this tool when:\n1. There is only a single, straightforward task\n2. The task is trivial and tracking it provides no benefit\n3. The task can be completed in less than 3 trivial steps\n4. The task is purely conversational or informational\n\n## Examples of When to Use the Todo List\n\n<example>\nUser: I want to add a dark mode toggle to the application settings. Make sure you run the tests and build when you're done!\nAssistant: I'll help add a dark mode toggle to your application settings. Let me create a todo list to track this implementation.\n*Creates todo list with the following items:*\n1. Create dark mode toggle component in Settings page\n2. Add dark mode state management (context/store)\n3. Implement CSS-in-JS styles for dark theme\n4. Update existing components to support theme switching\n5. Run tests and build process, addressing any failures or errors that occur\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Adding dark mode in it of itself is a multi-step feature requiring UI, state management, and styling changes\n2. The assistant inferred that tests and build need to pass by adding \"Ensure tests and build succeed\" as the final task\n3. Both of the user's requests are complex and require multiple steps to complete.\n</reasoning>\n</example>\n\n<example>\nUser: Help me plan a comprehensive marketing campaign for our new product launch, including social media, email marketing, and press releases.\nAssistant: I'll help you plan a comprehensive marketing campaign for your product launch. Let me create a todo list to organize all the components.\n*Creates todo list with the following items:*\n1. Define target audience and key messaging\n2. Create social media content calendar and posting schedule\n3. Design email marketing sequence and automation flows\n4. Draft press release and identify media contacts\n5. Set up tracking and analytics for campaign performance\n6. Create visual assets and brand materials\n7. Coordinate launch timeline and deliverables\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Marketing campaign planning involves multiple distinct channels and activities\n2. Each component requires careful coordination and planning\n3. The systematic approach ensures all aspects of the launch are covered\n</reasoning>\n</example>\n\n<example>\nUser: Help me rename the function getCwd to getCurrentWorkingDirectory across my project\nAssistant: Let me first search through your codebase to find all occurrences of 'getCwd'.\n*Uses search tools to locate all instances of getCwd in the codebase*\nAssistant: I've found 15 instances of 'getCwd' across 8 different files. Let me create a todo list to track these changes.\n*Creates todo list with specific items for each file that needs updating*\n\n<reasoning>\nThe assistant used the todo list because:\n1. The assistant searched to understand the scope of the task\n2. Upon finding multiple occurrences across different files, it determined this was a complex task with multiple steps (>3)\n3. The todo list helps ensure every instance is tracked and updated systematically\n4. This approach prevents missing any occurrences and maintains consistency.\n</reasoning>\n</example>\n\n<example>\nUser: I need help organizing my home office: decluttering the desk, setting up a filing system, organizing digital files, and creating a daily workflow.\nAssistant: I'll help you organize your home office systematically. Let me create a todo list to track each area of organization.\n*Creates todo list with items like: 1) Sort and declutter desk items, 2) Set up physical filing system for documents, 3) Organize digital files and folders, 4) Create daily workflow and productivity systems, 5) Establish maintenance routines*\nLet's start with decluttering your desk space to create a clean foundation for the other organizational systems.\n\n<reasoning>\nThe assistant used the todo list because:\n1. The user provided multiple distinct organizational tasks\n2. Each area requires different approaches and careful planning\n3. Home organization benefits from systematic, step-by-step completion\n4. Progress tracking ensures all areas are addressed thoroughly\n</reasoning>\n</example>\n\n## Examples of When NOT to Use the Todo List\n\n<example>\nUser: How do I print 'Hello World' in Python?\nAssistant: In Python, you can print \"Hello World\" with this simple code:\n\n\\`\\`\\`python\nprint(\"Hello World\")\n\\`\\`\\`\n\nThis will output the text \"Hello World\" to the console when executed.</assistant>\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, trivial task that can be completed in one step. There's no need to track multiple tasks or steps for such a straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: What's the best way to water houseplants?\nAssistant: The best way to water houseplants depends on the specific plant type, but here are some general guidelines:\n\n- Check soil moisture by inserting your finger about 1-2 inches into the soil\n- Water thoroughly when the top inch feels dry, allowing water to drain from the bottom\n- Use room temperature water and water in the morning when possible\n- Avoid overwatering, which is more harmful than underwatering for most plants\n\nDifferent plants have different needs, so it's important to research your specific varieties for optimal care.\n\n<reasoning>\nThe assistant did not use the todo list because this is an informational request with no actual tasks to complete. The user is simply asking for advice and explanation, not for the assistant to perform multiple steps or activities.\n</reasoning>\n</example>\n\n<example>\nUser: Can you add a comment to the calculateTotal function to explain what it does?\nAssistant: Sure, let me add a comment to the calculateTotal function to explain what it does.\n* Uses the Edit tool to add a comment to the calculateTotal function *\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, straightforward task confined to one location. Adding a comment doesn't require tracking multiple steps or systematic organization.\n</reasoning>\n</example>\n\n<example>\nUser: What time is it in Tokyo right now?\nAssistant: I'll check the current time in Tokyo for you.\n\n*Searches for current time in Tokyo*\n\nThe current time in Tokyo, Japan is [current time]. Tokyo is in the Japan Standard Time (JST) zone, which is UTC+9.\n\n<reasoning>\nThe assistant did not use the todo list because this is a single information lookup with immediate results. There are no multiple steps to track or organize, making the todo list unnecessary for this straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: I need to write a function that checks if a number is prime and then test it out.\nAssistant: I'll help you write a function that checks if a number is prime and then test it out.\n*Writes function that checks if a number is prime*\n*Tests the function*\n\n<reasoning>\nEven though this is a multi-step task, it is very straightforward and can be completed in two trivial steps (which is less than 3 steps!). Using the todo list here is overkill and wastes time and tokens.\n</reasoning>\n</example>\n\n<example>\nUser: I want you to order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\nAssistant: I'll help you order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\n*Orders a pizza from Dominos*\n*Orders a burger from McDonald's*\n*Orders a salad from Subway*\n\n<reasoning>\nEven though this is a multi-step task, assuming the assistant has the ability to order from these restaurants, it is very straightforward and can be completed in three trivial tool calls. \nUsing the todo list here is overkill and wastes time and tokens. These three tool calls should be made in parallel, in fact.\n</reasoning>\n</example>\n\n\n## Task States and Management\n\n1. **Task States**: Use these states to track progress:\n   - pending: Task not yet started\n   - in_progress: Currently working on (you can have multiple tasks in_progress at a time if they are not related to each other and can be run in parallel)\n   - completed: Task finished successfully\n\n2. **Task Management**:\n   - Update task status in real-time as you work\n   - Mark tasks complete IMMEDIATELY after finishing (don't batch completions)\n   - Complete current tasks before starting new ones\n   - Remove tasks that are no longer relevant from the list entirely\n   - IMPORTANT: When you write this todo list, you should mark your first task (or tasks) as in_progress immediately!.\n   - IMPORTANT: Unless all tasks are completed, you should always have at least one task in_progress to show the user that you are working on something.\n\n3. **Task Completion Requirements**:\n   - ONLY mark a task as completed when you have FULLY accomplished it\n   - If you encounter errors, blockers, or cannot finish, keep the task as in_progress\n   - When blocked, create a new task describing what needs to be resolved\n   - Never mark a task as completed if:\n     - There are unresolved issues or errors\n     - Work is partial or incomplete\n     - You encountered blockers that prevent completion\n     - You couldn't find necessary resources or dependencies\n     - Quality standards haven't been met\n\n4. **Task Breakdown**:\n   - Create specific, actionable items\n   - Break complex tasks into smaller, manageable steps\n   - Use clear, descriptive task names\n\nBeing proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully\nRemember: If you only need to make a few tool calls to complete a task, and it is clear what you need to do, it is better to just do the task directly and NOT call this tool at all.`;\n\nexport const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = `## \\`write_todos\\`\n\nYou have access to the \\`write_todos\\` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The \\`write_todos\\` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;\n\nconst TodoStatus = z\n  .enum([\"pending\", \"in_progress\", \"completed\"])\n  .describe(\"Status of the todo\");\nconst TodoSchema = z.object({\n  content: z.string().describe(\"Content of the todo item\"),\n  status: TodoStatus,\n});\nexport type Todo = z.infer<typeof TodoSchema>;\n\nconst stateSchema = z.object({\n  todos: z.array(TodoSchema).default([]),\n});\nexport type TodoMiddlewareState = z.infer<typeof stateSchema>;\n\nexport interface TodoListMiddlewareOptions {\n  /**\n   * Custom system prompt to guide the agent on using the todo tool.\n   * If not provided, uses the default {@link PLANNING_MIDDLEWARE_SYSTEM_PROMPT}.\n   */\n  systemPrompt?: string;\n  /**\n   * Custom description for the {@link writeTodos} tool.\n   * If not provided, uses the default {@link WRITE_TODOS_DESCRIPTION}.\n   */\n  toolDescription?: string;\n}\n\n/**\n * Creates a middleware that provides todo list management capabilities to agents.\n *\n * This middleware adds a `write_todos` tool that allows agents to create and manage\n * structured task lists for complex multi-step operations. It's designed to help\n * agents track progress, organize complex tasks, and provide users with visibility\n * into task completion status.\n *\n * The middleware automatically injects system prompts that guide the agent on when\n * and how to use the todo functionality effectively. It also enforces that the\n * `write_todos` tool is called at most once per model turn, since the tool replaces\n * the entire todo list and parallel calls would create ambiguity about precedence.\n *\n * @example\n * ```typescript\n * import { todoListMiddleware, createAgent } from 'langchain';\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [todoListMiddleware()],\n * });\n *\n * // Agent now has access to write_todos tool and todo state tracking\n * const result = await agent.invoke({\n *   messages: [new HumanMessage(\"Help me refactor my codebase\")]\n * });\n *\n * console.log(result.todos); // Array of todo items with status tracking\n * ```\n *\n * @returns A configured middleware instance that provides todo management capabilities\n *\n * @see {@link TodoMiddlewareState} for the state schema\n * @see {@link writeTodos} for the tool implementation\n */\nexport function todoListMiddleware(options?: TodoListMiddlewareOptions) {\n  /**\n   * Write todos tool - manages todo list with Command return\n   */\n  const writeTodos = tool(\n    ({ todos }, config) => {\n      return new Command({\n        update: {\n          todos,\n          messages: [\n            new ToolMessage({\n              content: `Updated todo list to ${JSON.stringify(todos)}`,\n              tool_call_id: config.toolCall?.id as string,\n              name: \"write_todos\",\n            }),\n          ],\n        },\n      });\n    },\n    {\n      name: \"write_todos\",\n      description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,\n      schema: z.object({\n        todos: z.array(TodoSchema).describe(\"List of todo items to update\"),\n      }),\n    }\n  );\n\n  return createMiddleware({\n    name: \"todoListMiddleware\",\n    stateSchema,\n    tools: [writeTodos],\n    wrapModelCall: (request, handler) =>\n      handler({\n        ...request,\n        systemMessage: request.systemMessage.concat(\n          `\\n\\n${options?.systemPrompt ?? TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT}`\n        ),\n      }),\n    afterModel: (state) => {\n      /**\n       * Check for parallel write_todos tool calls and return errors if detected.\n       *\n       * The todo list is designed to be updated at most once per model turn. Since\n       * the `write_todos` tool replaces the entire todo list with each call, making\n       * multiple parallel calls would create ambiguity about which update should take\n       * precedence. This method prevents such conflicts by rejecting any response that\n       * contains multiple write_todos tool calls.\n       */\n      const messages = state.messages;\n      if (!messages || messages.length === 0) {\n        return undefined;\n      }\n\n      /**\n       * Find the last AI message\n       */\n      const lastAiMsg = [...messages]\n        .reverse()\n        .find((msg) => AIMessage.isInstance(msg));\n      if (\n        !lastAiMsg ||\n        !lastAiMsg.tool_calls ||\n        lastAiMsg.tool_calls.length === 0\n      ) {\n        return undefined;\n      }\n\n      /**\n       * Count write_todos tool calls\n       */\n      const writeTodosCalls = lastAiMsg.tool_calls.filter(\n        (tc) => tc.name === writeTodos.name\n      );\n\n      if (writeTodosCalls.length > 1) {\n        /**\n         * Create error tool messages for all write_todos calls\n         */\n        const errorMessages = writeTodosCalls.map(\n          (tc) =>\n            new ToolMessage({\n              content:\n                \"Error: The `write_todos` tool should never be called multiple times \" +\n                \"in parallel. Please call it only once per model invocation to update \" +\n                \"the todo list.\",\n              tool_call_id: tc.id as string,\n              name: \"write_todos\",\n              status: \"error\",\n            })\n        );\n\n        /**\n         * Keep the tool calls in the AI message but return error messages\n         * This follows the same pattern as HumanInTheLoopMiddleware\n         */\n        return { messages: errorMessages };\n      }\n\n      return undefined;\n    },\n  });\n}\n"],"mappings":";;;;;;;;;;;;AAWA,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsNhC,MAAa,qCAAqC;;;;;;;;;;;;;AAclD,MAAM,aAAaA,OAAAA,EAChB,KAAK;CAAC;CAAW;CAAe;CAAY,CAAC,CAC7C,SAAS,qBAAqB;AACjC,MAAM,aAAaA,OAAAA,EAAE,OAAO;CAC1B,SAASA,OAAAA,EAAE,QAAQ,CAAC,SAAS,2BAA2B;CACxD,QAAQ;CACT,CAAC;AAGF,MAAM,cAAcA,OAAAA,EAAE,OAAO,EAC3B,OAAOA,OAAAA,EAAE,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC,EACvC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDF,SAAgB,mBAAmB,SAAqC;;;;CAItE,MAAM,cAAA,GAAA,sBAAA,OACH,EAAE,SAAS,WAAW;AACrB,SAAO,IAAIC,qBAAAA,QAAQ,EACjB,QAAQ;GACN;GACA,UAAU,CACR,IAAIC,yBAAAA,YAAY;IACd,SAAS,wBAAwB,KAAK,UAAU,MAAM;IACtD,cAAc,OAAO,UAAU;IAC/B,MAAM;IACP,CAAC,CACH;GACF,EACF,CAAC;IAEJ;EACE,MAAM;EACN,aAAa,SAAS,mBAAmB;EACzC,QAAQF,OAAAA,EAAE,OAAO,EACf,OAAOA,OAAAA,EAAE,MAAM,WAAW,CAAC,SAAS,+BAA+B,EACpE,CAAC;EACH,CACF;AAED,QAAOG,mBAAAA,iBAAiB;EACtB,MAAM;EACN;EACA,OAAO,CAAC,WAAW;EACnB,gBAAgB,SAAS,YACvB,QAAQ;GACN,GAAG;GACH,eAAe,QAAQ,cAAc,OACnC,OAAO,SAAS,gBAAgB,qCACjC;GACF,CAAC;EACJ,aAAa,UAAU;;;;;;;;;;GAUrB,MAAM,WAAW,MAAM;AACvB,OAAI,CAAC,YAAY,SAAS,WAAW,EACnC;;;;GAMF,MAAM,YAAY,CAAC,GAAG,SAAS,CAC5B,SAAS,CACT,MAAM,QAAQC,yBAAAA,UAAU,WAAW,IAAI,CAAC;AAC3C,OACE,CAAC,aACD,CAAC,UAAU,cACX,UAAU,WAAW,WAAW,EAEhC;;;;GAMF,MAAM,kBAAkB,UAAU,WAAW,QAC1C,OAAO,GAAG,SAAS,WAAW,KAChC;AAED,OAAI,gBAAgB,SAAS;;;;;AAqB3B,UAAO,EAAE,UAjBa,gBAAgB,KACnC,OACC,IAAIF,yBAAAA,YAAY;IACd,SACE;IAGF,cAAc,GAAG;IACjB,MAAM;IACN,QAAQ;IACT,CAAC,CAO0B,EAAE;;EAKvC,CAAC"}
+{"version":3,"file":"todoListMiddleware.cjs","names":["z","Command","ToolMessage","createMiddleware","AIMessage"],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { Command } from \"@langchain/langgraph\";\nimport { tool } from \"@langchain/core/tools\";\nimport { AIMessage, ToolMessage } from \"@langchain/core/messages\";\n\nimport { createMiddleware } from \"../index.js\";\n\n/**\n * Description for the write_todos tool\n * Ported exactly from Python WRITE_TODOS_DESCRIPTION\n */\nconst WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.\nIt also helps the user understand the progress of the task and overall progress of their requests.\nOnly use this tool if you think it will be helpful in staying organized. If the user's request is trivial and takes less than 3 steps, it is better to NOT use this tool and just do the task directly.\n\n## When to Use This Tool\nUse this tool in these scenarios:\n\n1. Complex multi-step tasks - When a task requires 3 or more distinct steps or actions\n2. Non-trivial and complex tasks - Tasks that require careful planning or multiple operations\n3. User explicitly requests todo list - When the user directly asks you to use the todo list\n4. User provides multiple tasks - When users provide a list of things to be done (numbered or comma-separated)\n5. The plan may need future revisions or updates based on results from the first few steps. Keeping track of this in a list is helpful.\n\n## How to Use This Tool\n1. When you start working on a task - Mark it as in_progress BEFORE beginning work.\n2. After completing a task - Mark it as completed and add any new follow-up tasks discovered during implementation.\n3. You can also update future tasks, such as deleting them if they are no longer necessary, or adding new tasks that are necessary. Don't change previously completed tasks.\n4. You can make several updates to the todo list at once. For example, when you complete a task, you can mark the next task you need to start as in_progress.\n\n## When NOT to Use This Tool\nIt is important to skip using this tool when:\n1. There is only a single, straightforward task\n2. The task is trivial and tracking it provides no benefit\n3. The task can be completed in less than 3 trivial steps\n4. The task is purely conversational or informational\n\n## Examples of When to Use the Todo List\n\n<example>\nUser: I want to add a dark mode toggle to the application settings. Make sure you run the tests and build when you're done!\nAssistant: I'll help add a dark mode toggle to your application settings. Let me create a todo list to track this implementation.\n*Creates todo list with the following items:*\n1. Create dark mode toggle component in Settings page\n2. Add dark mode state management (context/store)\n3. Implement CSS-in-JS styles for dark theme\n4. Update existing components to support theme switching\n5. Run tests and build process, addressing any failures or errors that occur\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Adding dark mode in it of itself is a multi-step feature requiring UI, state management, and styling changes\n2. The assistant inferred that tests and build need to pass by adding \"Ensure tests and build succeed\" as the final task\n3. Both of the user's requests are complex and require multiple steps to complete.\n</reasoning>\n</example>\n\n<example>\nUser: Help me plan a comprehensive marketing campaign for our new product launch, including social media, email marketing, and press releases.\nAssistant: I'll help you plan a comprehensive marketing campaign for your product launch. Let me create a todo list to organize all the components.\n*Creates todo list with the following items:*\n1. Define target audience and key messaging\n2. Create social media content calendar and posting schedule\n3. Design email marketing sequence and automation flows\n4. Draft press release and identify media contacts\n5. Set up tracking and analytics for campaign performance\n6. Create visual assets and brand materials\n7. Coordinate launch timeline and deliverables\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Marketing campaign planning involves multiple distinct channels and activities\n2. Each component requires careful coordination and planning\n3. The systematic approach ensures all aspects of the launch are covered\n</reasoning>\n</example>\n\n<example>\nUser: Help me rename the function getCwd to getCurrentWorkingDirectory across my project\nAssistant: Let me first search through your codebase to find all occurrences of 'getCwd'.\n*Uses search tools to locate all instances of getCwd in the codebase*\nAssistant: I've found 15 instances of 'getCwd' across 8 different files. Let me create a todo list to track these changes.\n*Creates todo list with specific items for each file that needs updating*\n\n<reasoning>\nThe assistant used the todo list because:\n1. The assistant searched to understand the scope of the task\n2. Upon finding multiple occurrences across different files, it determined this was a complex task with multiple steps (>3)\n3. The todo list helps ensure every instance is tracked and updated systematically\n4. This approach prevents missing any occurrences and maintains consistency.\n</reasoning>\n</example>\n\n<example>\nUser: I need help organizing my home office: decluttering the desk, setting up a filing system, organizing digital files, and creating a daily workflow.\nAssistant: I'll help you organize your home office systematically. Let me create a todo list to track each area of organization.\n*Creates todo list with items like: 1) Sort and declutter desk items, 2) Set up physical filing system for documents, 3) Organize digital files and folders, 4) Create daily workflow and productivity systems, 5) Establish maintenance routines*\nLet's start with decluttering your desk space to create a clean foundation for the other organizational systems.\n\n<reasoning>\nThe assistant used the todo list because:\n1. The user provided multiple distinct organizational tasks\n2. Each area requires different approaches and careful planning\n3. Home organization benefits from systematic, step-by-step completion\n4. Progress tracking ensures all areas are addressed thoroughly\n</reasoning>\n</example>\n\n## Examples of When NOT to Use the Todo List\n\n<example>\nUser: How do I print 'Hello World' in Python?\nAssistant: In Python, you can print \"Hello World\" with this simple code:\n\n\\`\\`\\`python\nprint(\"Hello World\")\n\\`\\`\\`\n\nThis will output the text \"Hello World\" to the console when executed.</assistant>\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, trivial task that can be completed in one step. There's no need to track multiple tasks or steps for such a straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: What's the best way to water houseplants?\nAssistant: The best way to water houseplants depends on the specific plant type, but here are some general guidelines:\n\n- Check soil moisture by inserting your finger about 1-2 inches into the soil\n- Water thoroughly when the top inch feels dry, allowing water to drain from the bottom\n- Use room temperature water and water in the morning when possible\n- Avoid overwatering, which is more harmful than underwatering for most plants\n\nDifferent plants have different needs, so it's important to research your specific varieties for optimal care.\n\n<reasoning>\nThe assistant did not use the todo list because this is an informational request with no actual tasks to complete. The user is simply asking for advice and explanation, not for the assistant to perform multiple steps or activities.\n</reasoning>\n</example>\n\n<example>\nUser: Can you add a comment to the calculateTotal function to explain what it does?\nAssistant: Sure, let me add a comment to the calculateTotal function to explain what it does.\n* Uses the Edit tool to add a comment to the calculateTotal function *\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, straightforward task confined to one location. Adding a comment doesn't require tracking multiple steps or systematic organization.\n</reasoning>\n</example>\n\n<example>\nUser: What time is it in Tokyo right now?\nAssistant: I'll check the current time in Tokyo for you.\n\n*Searches for current time in Tokyo*\n\nThe current time in Tokyo, Japan is [current time]. Tokyo is in the Japan Standard Time (JST) zone, which is UTC+9.\n\n<reasoning>\nThe assistant did not use the todo list because this is a single information lookup with immediate results. There are no multiple steps to track or organize, making the todo list unnecessary for this straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: I need to write a function that checks if a number is prime and then test it out.\nAssistant: I'll help you write a function that checks if a number is prime and then test it out.\n*Writes function that checks if a number is prime*\n*Tests the function*\n\n<reasoning>\nEven though this is a multi-step task, it is very straightforward and can be completed in two trivial steps (which is less than 3 steps!). Using the todo list here is overkill and wastes time and tokens.\n</reasoning>\n</example>\n\n<example>\nUser: I want you to order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\nAssistant: I'll help you order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\n*Orders a pizza from Dominos*\n*Orders a burger from McDonald's*\n*Orders a salad from Subway*\n\n<reasoning>\nEven though this is a multi-step task, assuming the assistant has the ability to order from these restaurants, it is very straightforward and can be completed in three trivial tool calls. \nUsing the todo list here is overkill and wastes time and tokens. These three tool calls should be made in parallel, in fact.\n</reasoning>\n</example>\n\n\n## Task States and Management\n\n1. **Task States**: Use these states to track progress:\n   - pending: Task not yet started\n   - in_progress: Currently working on (you can have multiple tasks in_progress at a time if they are not related to each other and can be run in parallel)\n   - completed: Task finished successfully\n\n2. **Task Management**:\n   - Update task status in real-time as you work\n   - Mark tasks complete IMMEDIATELY after finishing (don't batch completions)\n   - Complete current tasks before starting new ones\n   - Remove tasks that are no longer relevant from the list entirely\n   - IMPORTANT: When you write this todo list, you should mark your first task (or tasks) as in_progress immediately!.\n   - IMPORTANT: Unless all tasks are completed, you should always have at least one task in_progress to show the user that you are working on something.\n\n3. **Task Completion Requirements**:\n   - ONLY mark a task as completed when you have FULLY accomplished it\n   - If you encounter errors, blockers, or cannot finish, keep the task as in_progress\n   - When blocked, create a new task describing what needs to be resolved\n   - Never mark a task as completed if:\n     - There are unresolved issues or errors\n     - Work is partial or incomplete\n     - You encountered blockers that prevent completion\n     - You couldn't find necessary resources or dependencies\n     - Quality standards haven't been met\n\n4. **Task Breakdown**:\n   - Create specific, actionable items\n   - Break complex tasks into smaller, manageable steps\n   - Use clear, descriptive task names\n\nBeing proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully\nRemember: If you only need to make a few tool calls to complete a task, and it is clear what you need to do, it is better to just do the task directly and NOT call this tool at all.`;\n\nexport const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = `## \\`write_todos\\`\n\nYou have access to the \\`write_todos\\` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The \\`write_todos\\` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;\n\nconst TodoStatus = z\n  .enum([\"pending\", \"in_progress\", \"completed\"])\n  .describe(\"Status of the todo\");\nconst TodoSchema = z.object({\n  content: z.string().describe(\"Content of the todo item\"),\n  status: TodoStatus,\n});\nexport type Todo = z.infer<typeof TodoSchema>;\n\nconst stateSchema = z.object({\n  todos: z.array(TodoSchema).default([]),\n});\nexport type TodoMiddlewareState = z.infer<typeof stateSchema>;\n\nexport interface TodoListMiddlewareOptions {\n  /**\n   * Custom system prompt to guide the agent on using the todo tool.\n   * If not provided, uses the default {@link PLANNING_MIDDLEWARE_SYSTEM_PROMPT}.\n   */\n  systemPrompt?: string;\n  /**\n   * Custom description for the {@link writeTodos} tool.\n   * If not provided, uses the default {@link WRITE_TODOS_DESCRIPTION}.\n   */\n  toolDescription?: string;\n}\n\n/**\n * Creates a middleware that provides todo list management capabilities to agents.\n *\n * This middleware adds a `write_todos` tool that allows agents to create and manage\n * structured task lists for complex multi-step operations. It's designed to help\n * agents track progress, organize complex tasks, and provide users with visibility\n * into task completion status.\n *\n * The middleware automatically injects system prompts that guide the agent on when\n * and how to use the todo functionality effectively. It also enforces that the\n * `write_todos` tool is called at most once per model turn, since the tool replaces\n * the entire todo list and parallel calls would create ambiguity about precedence.\n *\n * @example\n * ```typescript\n * import { todoListMiddleware, createAgent } from 'langchain';\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [todoListMiddleware()],\n * });\n *\n * // Agent now has access to write_todos tool and todo state tracking\n * const result = await agent.invoke({\n *   messages: [new HumanMessage(\"Help me refactor my codebase\")]\n * });\n *\n * console.log(result.todos); // Array of todo items with status tracking\n * ```\n *\n * @returns A configured middleware instance that provides todo management capabilities\n *\n * @see {@link TodoMiddlewareState} for the state schema\n * @see {@link writeTodos} for the tool implementation\n */\nexport function todoListMiddleware(options?: TodoListMiddlewareOptions) {\n  /**\n   * Write todos tool - manages todo list with Command return\n   */\n  const writeTodos = tool(\n    ({ todos }, config) => {\n      return new Command({\n        update: {\n          todos,\n          messages: [\n            new ToolMessage({\n              content: `Updated todo list to ${JSON.stringify(todos)}`,\n              tool_call_id: config.toolCall?.id as string,\n              name: \"write_todos\",\n            }),\n          ],\n        },\n      });\n    },\n    {\n      name: \"write_todos\",\n      description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,\n      schema: z.object({\n        todos: z.array(TodoSchema).describe(\"List of todo items to update\"),\n      }),\n    }\n  );\n\n  return createMiddleware({\n    name: \"todoListMiddleware\",\n    stateSchema,\n    tools: [writeTodos],\n    wrapModelCall: (request, handler) =>\n      handler({\n        ...request,\n        systemMessage: request.systemMessage.concat(\n          `\\n\\n${options?.systemPrompt ?? TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT}`\n        ),\n      }),\n    afterModel: (state) => {\n      /**\n       * Check for parallel write_todos tool calls and return errors if detected.\n       *\n       * The todo list is designed to be updated at most once per model turn. Since\n       * the `write_todos` tool replaces the entire todo list with each call, making\n       * multiple parallel calls would create ambiguity about which update should take\n       * precedence. This method prevents such conflicts by rejecting any response that\n       * contains multiple write_todos tool calls.\n       */\n      const messages = state.messages;\n      if (!messages || messages.length === 0) {\n        return undefined;\n      }\n\n      /**\n       * Find the last AI message\n       */\n      const lastAiMsg = [...messages]\n        .reverse()\n        .find((msg) => AIMessage.isInstance(msg));\n      if (\n        !lastAiMsg ||\n        !lastAiMsg.tool_calls ||\n        lastAiMsg.tool_calls.length === 0\n      ) {\n        return undefined;\n      }\n\n      /**\n       * Count write_todos tool calls\n       */\n      const writeTodosCalls = lastAiMsg.tool_calls.filter(\n        (tc) => tc.name === writeTodos.name\n      );\n\n      if (writeTodosCalls.length > 1) {\n        /**\n         * Create error tool messages for all write_todos calls\n         */\n        const errorMessages = writeTodosCalls.map(\n          (tc) =>\n            new ToolMessage({\n              content:\n                \"Error: The `write_todos` tool should never be called multiple times \" +\n                \"in parallel. Please call it only once per model invocation to update \" +\n                \"the todo list.\",\n              tool_call_id: tc.id as string,\n              name: \"write_todos\",\n              status: \"error\",\n            })\n        );\n\n        /**\n         * Keep the tool calls in the AI message but return error messages\n         * This follows the same pattern as HumanInTheLoopMiddleware\n         */\n        return { messages: errorMessages };\n      }\n\n      return undefined;\n    },\n  });\n}\n"],"mappings":";;;;;;;;;;;AAWA,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsNhC,MAAa,qCAAqC;;;;;;;;;;;;;AAclD,MAAM,aAAaA,OAAAA,EAChB,KAAK;CAAC;CAAW;CAAe;AAAW,CAAC,CAAC,CAC7C,SAAS,oBAAoB;AAChC,MAAM,aAAaA,OAAAA,EAAE,OAAO;CAC1B,SAASA,OAAAA,EAAE,OAAO,CAAC,CAAC,SAAS,0BAA0B;CACvD,QAAQ;AACV,CAAC;AAGD,MAAM,cAAcA,OAAAA,EAAE,OAAO,EAC3B,OAAOA,OAAAA,EAAE,MAAM,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,EACvC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDD,SAAgB,mBAAmB,SAAqC;;;;CAItE,MAAM,cAAA,GAAA,sBAAA,KAAA,EACH,EAAE,SAAS,WAAW;EACrB,OAAO,IAAIC,qBAAAA,QAAQ,EACjB,QAAQ;GACN;GACA,UAAU,CACR,IAAIC,yBAAAA,YAAY;IACd,SAAS,wBAAwB,KAAK,UAAU,KAAK;IACrD,cAAc,OAAO,UAAU;IAC/B,MAAM;GACR,CAAC,CACH;EACF,EACF,CAAC;CACH,GACA;EACE,MAAM;EACN,aAAa,SAAS,mBAAmB;EACzC,QAAQF,OAAAA,EAAE,OAAO,EACf,OAAOA,OAAAA,EAAE,MAAM,UAAU,CAAC,CAAC,SAAS,8BAA8B,EACpE,CAAC;CACH,CACF;CAEA,OAAOG,mBAAAA,iBAAiB;EACtB,MAAM;EACN;EACA,OAAO,CAAC,UAAU;EAClB,gBAAgB,SAAS,YACvB,QAAQ;GACN,GAAG;GACH,eAAe,QAAQ,cAAc,OACnC,OAAO,SAAS,gBAAgB,oCAClC;EACF,CAAC;EACH,aAAa,UAAU;;;;;;;;;;GAUrB,MAAM,WAAW,MAAM;GACvB,IAAI,CAAC,YAAY,SAAS,WAAW,GACnC;;;;GAMF,MAAM,YAAY,CAAC,GAAG,QAAQ,CAAC,CAC5B,QAAQ,CAAC,CACT,MAAM,QAAQC,yBAAAA,UAAU,WAAW,GAAG,CAAC;GAC1C,IACE,CAAC,aACD,CAAC,UAAU,cACX,UAAU,WAAW,WAAW,GAEhC;;;;GAMF,MAAM,kBAAkB,UAAU,WAAW,QAC1C,OAAO,GAAG,SAAS,WAAW,IACjC;GAEA,IAAI,gBAAgB,SAAS;;;;;GAqB3B,OAAO,EAAE,UAjBa,gBAAgB,KACnC,OACC,IAAIF,yBAAAA,YAAY;IACd,SACE;IAGF,cAAc,GAAG;IACjB,MAAM;IACN,QAAQ;GACV,CAAC,CAO0B,EAAE;EAIrC;CACF,CAAC;AACH"}

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

@@ -1,10 +1,7 @@
 import { AgentMiddleware } from "./types.cjs";
-import * as _$_langchain_core_messages0 from "@langchain/core/messages";
 import { ToolMessage } from "@langchain/core/messages";
-import * as _$_langchain_core_tools0 from "@langchain/core/tools";
 import { Command } from "@langchain/langgraph";
 import { z } from "zod/v3";
-
 //#region src/agents/middleware/todoListMiddleware.d.ts
 declare const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = "## `write_todos`\n\nYou have access to the `write_todos` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The `write_todos` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.";
 declare const TodoSchema: z.ZodObject<{
@@ -86,7 +83,7 @@ declare function todoListMiddleware(options?: TodoListMiddlewareOptions): AgentM
     content: string;
     status: "completed" | "in_progress" | "pending";
   }[] | undefined;
-}>, undefined, unknown, readonly [_$_langchain_core_tools0.DynamicStructuredTool<z.ZodObject<{
+}>, undefined, unknown, readonly [import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{
   todos: z.ZodArray<z.ZodObject<{
     content: z.ZodString;
     status: z.ZodEnum<["pending", "in_progress", "completed"]>;
@@ -122,7 +119,7 @@ declare function todoListMiddleware(options?: TodoListMiddlewareOptions): AgentM
     content: string;
     status: "completed" | "in_progress" | "pending";
   }[];
-  messages: ToolMessage<_$_langchain_core_messages0.MessageStructure<_$_langchain_core_messages0.MessageToolSet>>[];
+  messages: ToolMessage<import("@langchain/core/messages").MessageStructure<import("@langchain/core/messages").MessageToolSet>>[];
 }, string>, unknown, "write_todos">], readonly []>;
 //#endregion
 export { TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT, Todo, TodoListMiddlewareOptions, todoListMiddleware };

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

@@ -1 +1 @@
-{"version":3,"file":"todoListMiddleware.d.cts","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"mappings":";;;;;;;;cAiOa,kCAAA;AAAA,cAiBP,UAAA,EAAU,CAAA,CAAA,SAAA;;;;;;;;;;KAIJ,IAAA,GAAO,CAAA,CAAE,KAAA,QAAa,UAAA;AAAA,UAOjB,yBAAA;EAPD;;;;EAYd,YAAA;EAZgC;;;AAOlC;EAUE,eAAA;AAAA;;;AAsCF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAAgB,kBAAA,CAAmB,OAAA,GAAU,yBAAA,mBAAyB,CAAA,CAAA,SAAA"}
+{"version":3,"file":"todoListMiddleware.d.cts","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"mappings":";;;;;cAiOa;cAiBP,YAAU,EAAA;;;;;;;;;;KAIJ,OAAO,EAAE,aAAa;UAOjB;;;;;EAKf;;;;;EAKA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsCc,mBAAmB,UAAU,4CAAyB,EAAA"}

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

@@ -1,10 +1,7 @@
 import { AgentMiddleware } from "./types.js";
-import * as _$_langchain_core_messages0 from "@langchain/core/messages";
 import { ToolMessage } from "@langchain/core/messages";
-import * as _$_langchain_core_tools0 from "@langchain/core/tools";
 import { Command } from "@langchain/langgraph";
 import { z } from "zod/v3";
-
 //#region src/agents/middleware/todoListMiddleware.d.ts
 declare const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = "## `write_todos`\n\nYou have access to the `write_todos` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The `write_todos` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.";
 declare const TodoSchema: z.ZodObject<{
@@ -86,7 +83,7 @@ declare function todoListMiddleware(options?: TodoListMiddlewareOptions): AgentM
     content: string;
     status: "completed" | "in_progress" | "pending";
   }[] | undefined;
-}>, undefined, unknown, readonly [_$_langchain_core_tools0.DynamicStructuredTool<z.ZodObject<{
+}>, undefined, unknown, readonly [import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{
   todos: z.ZodArray<z.ZodObject<{
     content: z.ZodString;
     status: z.ZodEnum<["pending", "in_progress", "completed"]>;
@@ -122,7 +119,7 @@ declare function todoListMiddleware(options?: TodoListMiddlewareOptions): AgentM
     content: string;
     status: "completed" | "in_progress" | "pending";
   }[];
-  messages: ToolMessage<_$_langchain_core_messages0.MessageStructure<_$_langchain_core_messages0.MessageToolSet>>[];
+  messages: ToolMessage<import("@langchain/core/messages").MessageStructure<import("@langchain/core/messages").MessageToolSet>>[];
 }, string>, unknown, "write_todos">], readonly []>;
 //#endregion
 export { TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT, Todo, TodoListMiddlewareOptions, todoListMiddleware };

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

@@ -1 +1 @@
-{"version":3,"file":"todoListMiddleware.d.ts","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"mappings":";;;;;;;;cAiOa,kCAAA;AAAA,cAiBP,UAAA,EAAU,CAAA,CAAA,SAAA;;;;;;;;;;KAIJ,IAAA,GAAO,CAAA,CAAE,KAAA,QAAa,UAAA;AAAA,UAOjB,yBAAA;EAPD;;;;EAYd,YAAA;EAZgC;;;AAOlC;EAUE,eAAA;AAAA;;;AAsCF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAAgB,kBAAA,CAAmB,OAAA,GAAU,yBAAA,mBAAyB,CAAA,CAAA,SAAA"}
+{"version":3,"file":"todoListMiddleware.d.ts","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"mappings":";;;;;cAiOa;cAiBP,YAAU,EAAA;;;;;;;;;;KAIJ,OAAO,EAAE,aAAa;UAOjB;;;;;EAKf;;;;;EAKA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsCc,mBAAmB,UAAU,4CAAyB,EAAA"}

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

@@ -1 +1 @@
-{"version":3,"file":"todoListMiddleware.js","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { Command } from \"@langchain/langgraph\";\nimport { tool } from \"@langchain/core/tools\";\nimport { AIMessage, ToolMessage } from \"@langchain/core/messages\";\n\nimport { createMiddleware } from \"../index.js\";\n\n/**\n * Description for the write_todos tool\n * Ported exactly from Python WRITE_TODOS_DESCRIPTION\n */\nconst WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.\nIt also helps the user understand the progress of the task and overall progress of their requests.\nOnly use this tool if you think it will be helpful in staying organized. If the user's request is trivial and takes less than 3 steps, it is better to NOT use this tool and just do the task directly.\n\n## When to Use This Tool\nUse this tool in these scenarios:\n\n1. Complex multi-step tasks - When a task requires 3 or more distinct steps or actions\n2. Non-trivial and complex tasks - Tasks that require careful planning or multiple operations\n3. User explicitly requests todo list - When the user directly asks you to use the todo list\n4. User provides multiple tasks - When users provide a list of things to be done (numbered or comma-separated)\n5. The plan may need future revisions or updates based on results from the first few steps. Keeping track of this in a list is helpful.\n\n## How to Use This Tool\n1. When you start working on a task - Mark it as in_progress BEFORE beginning work.\n2. After completing a task - Mark it as completed and add any new follow-up tasks discovered during implementation.\n3. You can also update future tasks, such as deleting them if they are no longer necessary, or adding new tasks that are necessary. Don't change previously completed tasks.\n4. You can make several updates to the todo list at once. For example, when you complete a task, you can mark the next task you need to start as in_progress.\n\n## When NOT to Use This Tool\nIt is important to skip using this tool when:\n1. There is only a single, straightforward task\n2. The task is trivial and tracking it provides no benefit\n3. The task can be completed in less than 3 trivial steps\n4. The task is purely conversational or informational\n\n## Examples of When to Use the Todo List\n\n<example>\nUser: I want to add a dark mode toggle to the application settings. Make sure you run the tests and build when you're done!\nAssistant: I'll help add a dark mode toggle to your application settings. Let me create a todo list to track this implementation.\n*Creates todo list with the following items:*\n1. Create dark mode toggle component in Settings page\n2. Add dark mode state management (context/store)\n3. Implement CSS-in-JS styles for dark theme\n4. Update existing components to support theme switching\n5. Run tests and build process, addressing any failures or errors that occur\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Adding dark mode in it of itself is a multi-step feature requiring UI, state management, and styling changes\n2. The assistant inferred that tests and build need to pass by adding \"Ensure tests and build succeed\" as the final task\n3. Both of the user's requests are complex and require multiple steps to complete.\n</reasoning>\n</example>\n\n<example>\nUser: Help me plan a comprehensive marketing campaign for our new product launch, including social media, email marketing, and press releases.\nAssistant: I'll help you plan a comprehensive marketing campaign for your product launch. Let me create a todo list to organize all the components.\n*Creates todo list with the following items:*\n1. Define target audience and key messaging\n2. Create social media content calendar and posting schedule\n3. Design email marketing sequence and automation flows\n4. Draft press release and identify media contacts\n5. Set up tracking and analytics for campaign performance\n6. Create visual assets and brand materials\n7. Coordinate launch timeline and deliverables\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Marketing campaign planning involves multiple distinct channels and activities\n2. Each component requires careful coordination and planning\n3. The systematic approach ensures all aspects of the launch are covered\n</reasoning>\n</example>\n\n<example>\nUser: Help me rename the function getCwd to getCurrentWorkingDirectory across my project\nAssistant: Let me first search through your codebase to find all occurrences of 'getCwd'.\n*Uses search tools to locate all instances of getCwd in the codebase*\nAssistant: I've found 15 instances of 'getCwd' across 8 different files. Let me create a todo list to track these changes.\n*Creates todo list with specific items for each file that needs updating*\n\n<reasoning>\nThe assistant used the todo list because:\n1. The assistant searched to understand the scope of the task\n2. Upon finding multiple occurrences across different files, it determined this was a complex task with multiple steps (>3)\n3. The todo list helps ensure every instance is tracked and updated systematically\n4. This approach prevents missing any occurrences and maintains consistency.\n</reasoning>\n</example>\n\n<example>\nUser: I need help organizing my home office: decluttering the desk, setting up a filing system, organizing digital files, and creating a daily workflow.\nAssistant: I'll help you organize your home office systematically. Let me create a todo list to track each area of organization.\n*Creates todo list with items like: 1) Sort and declutter desk items, 2) Set up physical filing system for documents, 3) Organize digital files and folders, 4) Create daily workflow and productivity systems, 5) Establish maintenance routines*\nLet's start with decluttering your desk space to create a clean foundation for the other organizational systems.\n\n<reasoning>\nThe assistant used the todo list because:\n1. The user provided multiple distinct organizational tasks\n2. Each area requires different approaches and careful planning\n3. Home organization benefits from systematic, step-by-step completion\n4. Progress tracking ensures all areas are addressed thoroughly\n</reasoning>\n</example>\n\n## Examples of When NOT to Use the Todo List\n\n<example>\nUser: How do I print 'Hello World' in Python?\nAssistant: In Python, you can print \"Hello World\" with this simple code:\n\n\\`\\`\\`python\nprint(\"Hello World\")\n\\`\\`\\`\n\nThis will output the text \"Hello World\" to the console when executed.</assistant>\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, trivial task that can be completed in one step. There's no need to track multiple tasks or steps for such a straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: What's the best way to water houseplants?\nAssistant: The best way to water houseplants depends on the specific plant type, but here are some general guidelines:\n\n- Check soil moisture by inserting your finger about 1-2 inches into the soil\n- Water thoroughly when the top inch feels dry, allowing water to drain from the bottom\n- Use room temperature water and water in the morning when possible\n- Avoid overwatering, which is more harmful than underwatering for most plants\n\nDifferent plants have different needs, so it's important to research your specific varieties for optimal care.\n\n<reasoning>\nThe assistant did not use the todo list because this is an informational request with no actual tasks to complete. The user is simply asking for advice and explanation, not for the assistant to perform multiple steps or activities.\n</reasoning>\n</example>\n\n<example>\nUser: Can you add a comment to the calculateTotal function to explain what it does?\nAssistant: Sure, let me add a comment to the calculateTotal function to explain what it does.\n* Uses the Edit tool to add a comment to the calculateTotal function *\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, straightforward task confined to one location. Adding a comment doesn't require tracking multiple steps or systematic organization.\n</reasoning>\n</example>\n\n<example>\nUser: What time is it in Tokyo right now?\nAssistant: I'll check the current time in Tokyo for you.\n\n*Searches for current time in Tokyo*\n\nThe current time in Tokyo, Japan is [current time]. Tokyo is in the Japan Standard Time (JST) zone, which is UTC+9.\n\n<reasoning>\nThe assistant did not use the todo list because this is a single information lookup with immediate results. There are no multiple steps to track or organize, making the todo list unnecessary for this straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: I need to write a function that checks if a number is prime and then test it out.\nAssistant: I'll help you write a function that checks if a number is prime and then test it out.\n*Writes function that checks if a number is prime*\n*Tests the function*\n\n<reasoning>\nEven though this is a multi-step task, it is very straightforward and can be completed in two trivial steps (which is less than 3 steps!). Using the todo list here is overkill and wastes time and tokens.\n</reasoning>\n</example>\n\n<example>\nUser: I want you to order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\nAssistant: I'll help you order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\n*Orders a pizza from Dominos*\n*Orders a burger from McDonald's*\n*Orders a salad from Subway*\n\n<reasoning>\nEven though this is a multi-step task, assuming the assistant has the ability to order from these restaurants, it is very straightforward and can be completed in three trivial tool calls. \nUsing the todo list here is overkill and wastes time and tokens. These three tool calls should be made in parallel, in fact.\n</reasoning>\n</example>\n\n\n## Task States and Management\n\n1. **Task States**: Use these states to track progress:\n   - pending: Task not yet started\n   - in_progress: Currently working on (you can have multiple tasks in_progress at a time if they are not related to each other and can be run in parallel)\n   - completed: Task finished successfully\n\n2. **Task Management**:\n   - Update task status in real-time as you work\n   - Mark tasks complete IMMEDIATELY after finishing (don't batch completions)\n   - Complete current tasks before starting new ones\n   - Remove tasks that are no longer relevant from the list entirely\n   - IMPORTANT: When you write this todo list, you should mark your first task (or tasks) as in_progress immediately!.\n   - IMPORTANT: Unless all tasks are completed, you should always have at least one task in_progress to show the user that you are working on something.\n\n3. **Task Completion Requirements**:\n   - ONLY mark a task as completed when you have FULLY accomplished it\n   - If you encounter errors, blockers, or cannot finish, keep the task as in_progress\n   - When blocked, create a new task describing what needs to be resolved\n   - Never mark a task as completed if:\n     - There are unresolved issues or errors\n     - Work is partial or incomplete\n     - You encountered blockers that prevent completion\n     - You couldn't find necessary resources or dependencies\n     - Quality standards haven't been met\n\n4. **Task Breakdown**:\n   - Create specific, actionable items\n   - Break complex tasks into smaller, manageable steps\n   - Use clear, descriptive task names\n\nBeing proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully\nRemember: If you only need to make a few tool calls to complete a task, and it is clear what you need to do, it is better to just do the task directly and NOT call this tool at all.`;\n\nexport const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = `## \\`write_todos\\`\n\nYou have access to the \\`write_todos\\` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The \\`write_todos\\` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;\n\nconst TodoStatus = z\n  .enum([\"pending\", \"in_progress\", \"completed\"])\n  .describe(\"Status of the todo\");\nconst TodoSchema = z.object({\n  content: z.string().describe(\"Content of the todo item\"),\n  status: TodoStatus,\n});\nexport type Todo = z.infer<typeof TodoSchema>;\n\nconst stateSchema = z.object({\n  todos: z.array(TodoSchema).default([]),\n});\nexport type TodoMiddlewareState = z.infer<typeof stateSchema>;\n\nexport interface TodoListMiddlewareOptions {\n  /**\n   * Custom system prompt to guide the agent on using the todo tool.\n   * If not provided, uses the default {@link PLANNING_MIDDLEWARE_SYSTEM_PROMPT}.\n   */\n  systemPrompt?: string;\n  /**\n   * Custom description for the {@link writeTodos} tool.\n   * If not provided, uses the default {@link WRITE_TODOS_DESCRIPTION}.\n   */\n  toolDescription?: string;\n}\n\n/**\n * Creates a middleware that provides todo list management capabilities to agents.\n *\n * This middleware adds a `write_todos` tool that allows agents to create and manage\n * structured task lists for complex multi-step operations. It's designed to help\n * agents track progress, organize complex tasks, and provide users with visibility\n * into task completion status.\n *\n * The middleware automatically injects system prompts that guide the agent on when\n * and how to use the todo functionality effectively. It also enforces that the\n * `write_todos` tool is called at most once per model turn, since the tool replaces\n * the entire todo list and parallel calls would create ambiguity about precedence.\n *\n * @example\n * ```typescript\n * import { todoListMiddleware, createAgent } from 'langchain';\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [todoListMiddleware()],\n * });\n *\n * // Agent now has access to write_todos tool and todo state tracking\n * const result = await agent.invoke({\n *   messages: [new HumanMessage(\"Help me refactor my codebase\")]\n * });\n *\n * console.log(result.todos); // Array of todo items with status tracking\n * ```\n *\n * @returns A configured middleware instance that provides todo management capabilities\n *\n * @see {@link TodoMiddlewareState} for the state schema\n * @see {@link writeTodos} for the tool implementation\n */\nexport function todoListMiddleware(options?: TodoListMiddlewareOptions) {\n  /**\n   * Write todos tool - manages todo list with Command return\n   */\n  const writeTodos = tool(\n    ({ todos }, config) => {\n      return new Command({\n        update: {\n          todos,\n          messages: [\n            new ToolMessage({\n              content: `Updated todo list to ${JSON.stringify(todos)}`,\n              tool_call_id: config.toolCall?.id as string,\n              name: \"write_todos\",\n            }),\n          ],\n        },\n      });\n    },\n    {\n      name: \"write_todos\",\n      description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,\n      schema: z.object({\n        todos: z.array(TodoSchema).describe(\"List of todo items to update\"),\n      }),\n    }\n  );\n\n  return createMiddleware({\n    name: \"todoListMiddleware\",\n    stateSchema,\n    tools: [writeTodos],\n    wrapModelCall: (request, handler) =>\n      handler({\n        ...request,\n        systemMessage: request.systemMessage.concat(\n          `\\n\\n${options?.systemPrompt ?? TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT}`\n        ),\n      }),\n    afterModel: (state) => {\n      /**\n       * Check for parallel write_todos tool calls and return errors if detected.\n       *\n       * The todo list is designed to be updated at most once per model turn. Since\n       * the `write_todos` tool replaces the entire todo list with each call, making\n       * multiple parallel calls would create ambiguity about which update should take\n       * precedence. This method prevents such conflicts by rejecting any response that\n       * contains multiple write_todos tool calls.\n       */\n      const messages = state.messages;\n      if (!messages || messages.length === 0) {\n        return undefined;\n      }\n\n      /**\n       * Find the last AI message\n       */\n      const lastAiMsg = [...messages]\n        .reverse()\n        .find((msg) => AIMessage.isInstance(msg));\n      if (\n        !lastAiMsg ||\n        !lastAiMsg.tool_calls ||\n        lastAiMsg.tool_calls.length === 0\n      ) {\n        return undefined;\n      }\n\n      /**\n       * Count write_todos tool calls\n       */\n      const writeTodosCalls = lastAiMsg.tool_calls.filter(\n        (tc) => tc.name === writeTodos.name\n      );\n\n      if (writeTodosCalls.length > 1) {\n        /**\n         * Create error tool messages for all write_todos calls\n         */\n        const errorMessages = writeTodosCalls.map(\n          (tc) =>\n            new ToolMessage({\n              content:\n                \"Error: The `write_todos` tool should never be called multiple times \" +\n                \"in parallel. Please call it only once per model invocation to update \" +\n                \"the todo list.\",\n              tool_call_id: tc.id as string,\n              name: \"write_todos\",\n              status: \"error\",\n            })\n        );\n\n        /**\n         * Keep the tool calls in the AI message but return error messages\n         * This follows the same pattern as HumanInTheLoopMiddleware\n         */\n        return { messages: errorMessages };\n      }\n\n      return undefined;\n    },\n  });\n}\n"],"mappings":";;;;;;;;;;;AAWA,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsNhC,MAAa,qCAAqC;;;;;;;;;;;;;AAclD,MAAM,aAAa,EAChB,KAAK;CAAC;CAAW;CAAe;CAAY,CAAC,CAC7C,SAAS,qBAAqB;AACjC,MAAM,aAAa,EAAE,OAAO;CAC1B,SAAS,EAAE,QAAQ,CAAC,SAAS,2BAA2B;CACxD,QAAQ;CACT,CAAC;AAGF,MAAM,cAAc,EAAE,OAAO,EAC3B,OAAO,EAAE,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC,EACvC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDF,SAAgB,mBAAmB,SAAqC;;;;CAItE,MAAM,aAAa,MAChB,EAAE,SAAS,WAAW;AACrB,SAAO,IAAI,QAAQ,EACjB,QAAQ;GACN;GACA,UAAU,CACR,IAAI,YAAY;IACd,SAAS,wBAAwB,KAAK,UAAU,MAAM;IACtD,cAAc,OAAO,UAAU;IAC/B,MAAM;IACP,CAAC,CACH;GACF,EACF,CAAC;IAEJ;EACE,MAAM;EACN,aAAa,SAAS,mBAAmB;EACzC,QAAQ,EAAE,OAAO,EACf,OAAO,EAAE,MAAM,WAAW,CAAC,SAAS,+BAA+B,EACpE,CAAC;EACH,CACF;AAED,QAAO,iBAAiB;EACtB,MAAM;EACN;EACA,OAAO,CAAC,WAAW;EACnB,gBAAgB,SAAS,YACvB,QAAQ;GACN,GAAG;GACH,eAAe,QAAQ,cAAc,OACnC,OAAO,SAAS,gBAAgB,qCACjC;GACF,CAAC;EACJ,aAAa,UAAU;;;;;;;;;;GAUrB,MAAM,WAAW,MAAM;AACvB,OAAI,CAAC,YAAY,SAAS,WAAW,EACnC;;;;GAMF,MAAM,YAAY,CAAC,GAAG,SAAS,CAC5B,SAAS,CACT,MAAM,QAAQ,UAAU,WAAW,IAAI,CAAC;AAC3C,OACE,CAAC,aACD,CAAC,UAAU,cACX,UAAU,WAAW,WAAW,EAEhC;;;;GAMF,MAAM,kBAAkB,UAAU,WAAW,QAC1C,OAAO,GAAG,SAAS,WAAW,KAChC;AAED,OAAI,gBAAgB,SAAS;;;;;AAqB3B,UAAO,EAAE,UAjBa,gBAAgB,KACnC,OACC,IAAI,YAAY;IACd,SACE;IAGF,cAAc,GAAG;IACjB,MAAM;IACN,QAAQ;IACT,CAAC,CAO0B,EAAE;;EAKvC,CAAC"}
+{"version":3,"file":"todoListMiddleware.js","names":[],"sources":["../../../src/agents/middleware/todoListMiddleware.ts"],"sourcesContent":["import { z } from \"zod/v3\";\nimport { Command } from \"@langchain/langgraph\";\nimport { tool } from \"@langchain/core/tools\";\nimport { AIMessage, ToolMessage } from \"@langchain/core/messages\";\n\nimport { createMiddleware } from \"../index.js\";\n\n/**\n * Description for the write_todos tool\n * Ported exactly from Python WRITE_TODOS_DESCRIPTION\n */\nconst WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.\nIt also helps the user understand the progress of the task and overall progress of their requests.\nOnly use this tool if you think it will be helpful in staying organized. If the user's request is trivial and takes less than 3 steps, it is better to NOT use this tool and just do the task directly.\n\n## When to Use This Tool\nUse this tool in these scenarios:\n\n1. Complex multi-step tasks - When a task requires 3 or more distinct steps or actions\n2. Non-trivial and complex tasks - Tasks that require careful planning or multiple operations\n3. User explicitly requests todo list - When the user directly asks you to use the todo list\n4. User provides multiple tasks - When users provide a list of things to be done (numbered or comma-separated)\n5. The plan may need future revisions or updates based on results from the first few steps. Keeping track of this in a list is helpful.\n\n## How to Use This Tool\n1. When you start working on a task - Mark it as in_progress BEFORE beginning work.\n2. After completing a task - Mark it as completed and add any new follow-up tasks discovered during implementation.\n3. You can also update future tasks, such as deleting them if they are no longer necessary, or adding new tasks that are necessary. Don't change previously completed tasks.\n4. You can make several updates to the todo list at once. For example, when you complete a task, you can mark the next task you need to start as in_progress.\n\n## When NOT to Use This Tool\nIt is important to skip using this tool when:\n1. There is only a single, straightforward task\n2. The task is trivial and tracking it provides no benefit\n3. The task can be completed in less than 3 trivial steps\n4. The task is purely conversational or informational\n\n## Examples of When to Use the Todo List\n\n<example>\nUser: I want to add a dark mode toggle to the application settings. Make sure you run the tests and build when you're done!\nAssistant: I'll help add a dark mode toggle to your application settings. Let me create a todo list to track this implementation.\n*Creates todo list with the following items:*\n1. Create dark mode toggle component in Settings page\n2. Add dark mode state management (context/store)\n3. Implement CSS-in-JS styles for dark theme\n4. Update existing components to support theme switching\n5. Run tests and build process, addressing any failures or errors that occur\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Adding dark mode in it of itself is a multi-step feature requiring UI, state management, and styling changes\n2. The assistant inferred that tests and build need to pass by adding \"Ensure tests and build succeed\" as the final task\n3. Both of the user's requests are complex and require multiple steps to complete.\n</reasoning>\n</example>\n\n<example>\nUser: Help me plan a comprehensive marketing campaign for our new product launch, including social media, email marketing, and press releases.\nAssistant: I'll help you plan a comprehensive marketing campaign for your product launch. Let me create a todo list to organize all the components.\n*Creates todo list with the following items:*\n1. Define target audience and key messaging\n2. Create social media content calendar and posting schedule\n3. Design email marketing sequence and automation flows\n4. Draft press release and identify media contacts\n5. Set up tracking and analytics for campaign performance\n6. Create visual assets and brand materials\n7. Coordinate launch timeline and deliverables\n*Begins working on the first task*\n\n<reasoning>\nThe assistant used the todo list because:\n1. Marketing campaign planning involves multiple distinct channels and activities\n2. Each component requires careful coordination and planning\n3. The systematic approach ensures all aspects of the launch are covered\n</reasoning>\n</example>\n\n<example>\nUser: Help me rename the function getCwd to getCurrentWorkingDirectory across my project\nAssistant: Let me first search through your codebase to find all occurrences of 'getCwd'.\n*Uses search tools to locate all instances of getCwd in the codebase*\nAssistant: I've found 15 instances of 'getCwd' across 8 different files. Let me create a todo list to track these changes.\n*Creates todo list with specific items for each file that needs updating*\n\n<reasoning>\nThe assistant used the todo list because:\n1. The assistant searched to understand the scope of the task\n2. Upon finding multiple occurrences across different files, it determined this was a complex task with multiple steps (>3)\n3. The todo list helps ensure every instance is tracked and updated systematically\n4. This approach prevents missing any occurrences and maintains consistency.\n</reasoning>\n</example>\n\n<example>\nUser: I need help organizing my home office: decluttering the desk, setting up a filing system, organizing digital files, and creating a daily workflow.\nAssistant: I'll help you organize your home office systematically. Let me create a todo list to track each area of organization.\n*Creates todo list with items like: 1) Sort and declutter desk items, 2) Set up physical filing system for documents, 3) Organize digital files and folders, 4) Create daily workflow and productivity systems, 5) Establish maintenance routines*\nLet's start with decluttering your desk space to create a clean foundation for the other organizational systems.\n\n<reasoning>\nThe assistant used the todo list because:\n1. The user provided multiple distinct organizational tasks\n2. Each area requires different approaches and careful planning\n3. Home organization benefits from systematic, step-by-step completion\n4. Progress tracking ensures all areas are addressed thoroughly\n</reasoning>\n</example>\n\n## Examples of When NOT to Use the Todo List\n\n<example>\nUser: How do I print 'Hello World' in Python?\nAssistant: In Python, you can print \"Hello World\" with this simple code:\n\n\\`\\`\\`python\nprint(\"Hello World\")\n\\`\\`\\`\n\nThis will output the text \"Hello World\" to the console when executed.</assistant>\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, trivial task that can be completed in one step. There's no need to track multiple tasks or steps for such a straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: What's the best way to water houseplants?\nAssistant: The best way to water houseplants depends on the specific plant type, but here are some general guidelines:\n\n- Check soil moisture by inserting your finger about 1-2 inches into the soil\n- Water thoroughly when the top inch feels dry, allowing water to drain from the bottom\n- Use room temperature water and water in the morning when possible\n- Avoid overwatering, which is more harmful than underwatering for most plants\n\nDifferent plants have different needs, so it's important to research your specific varieties for optimal care.\n\n<reasoning>\nThe assistant did not use the todo list because this is an informational request with no actual tasks to complete. The user is simply asking for advice and explanation, not for the assistant to perform multiple steps or activities.\n</reasoning>\n</example>\n\n<example>\nUser: Can you add a comment to the calculateTotal function to explain what it does?\nAssistant: Sure, let me add a comment to the calculateTotal function to explain what it does.\n* Uses the Edit tool to add a comment to the calculateTotal function *\n\n<reasoning>\nThe assistant did not use the todo list because this is a single, straightforward task confined to one location. Adding a comment doesn't require tracking multiple steps or systematic organization.\n</reasoning>\n</example>\n\n<example>\nUser: What time is it in Tokyo right now?\nAssistant: I'll check the current time in Tokyo for you.\n\n*Searches for current time in Tokyo*\n\nThe current time in Tokyo, Japan is [current time]. Tokyo is in the Japan Standard Time (JST) zone, which is UTC+9.\n\n<reasoning>\nThe assistant did not use the todo list because this is a single information lookup with immediate results. There are no multiple steps to track or organize, making the todo list unnecessary for this straightforward request.\n</reasoning>\n</example>\n\n<example>\nUser: I need to write a function that checks if a number is prime and then test it out.\nAssistant: I'll help you write a function that checks if a number is prime and then test it out.\n*Writes function that checks if a number is prime*\n*Tests the function*\n\n<reasoning>\nEven though this is a multi-step task, it is very straightforward and can be completed in two trivial steps (which is less than 3 steps!). Using the todo list here is overkill and wastes time and tokens.\n</reasoning>\n</example>\n\n<example>\nUser: I want you to order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\nAssistant: I'll help you order a pizza from Dominos, order a burger from McDonald's, and order a salad from Subway.\n*Orders a pizza from Dominos*\n*Orders a burger from McDonald's*\n*Orders a salad from Subway*\n\n<reasoning>\nEven though this is a multi-step task, assuming the assistant has the ability to order from these restaurants, it is very straightforward and can be completed in three trivial tool calls. \nUsing the todo list here is overkill and wastes time and tokens. These three tool calls should be made in parallel, in fact.\n</reasoning>\n</example>\n\n\n## Task States and Management\n\n1. **Task States**: Use these states to track progress:\n   - pending: Task not yet started\n   - in_progress: Currently working on (you can have multiple tasks in_progress at a time if they are not related to each other and can be run in parallel)\n   - completed: Task finished successfully\n\n2. **Task Management**:\n   - Update task status in real-time as you work\n   - Mark tasks complete IMMEDIATELY after finishing (don't batch completions)\n   - Complete current tasks before starting new ones\n   - Remove tasks that are no longer relevant from the list entirely\n   - IMPORTANT: When you write this todo list, you should mark your first task (or tasks) as in_progress immediately!.\n   - IMPORTANT: Unless all tasks are completed, you should always have at least one task in_progress to show the user that you are working on something.\n\n3. **Task Completion Requirements**:\n   - ONLY mark a task as completed when you have FULLY accomplished it\n   - If you encounter errors, blockers, or cannot finish, keep the task as in_progress\n   - When blocked, create a new task describing what needs to be resolved\n   - Never mark a task as completed if:\n     - There are unresolved issues or errors\n     - Work is partial or incomplete\n     - You encountered blockers that prevent completion\n     - You couldn't find necessary resources or dependencies\n     - Quality standards haven't been met\n\n4. **Task Breakdown**:\n   - Create specific, actionable items\n   - Break complex tasks into smaller, manageable steps\n   - Use clear, descriptive task names\n\nBeing proactive with task management demonstrates attentiveness and ensures you complete all requirements successfully\nRemember: If you only need to make a few tool calls to complete a task, and it is clear what you need to do, it is better to just do the task directly and NOT call this tool at all.`;\n\nexport const TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT = `## \\`write_todos\\`\n\nYou have access to the \\`write_todos\\` tool to help you manage and plan complex objectives. \nUse this tool for complex objectives to ensure that you are tracking each necessary step and giving the user visibility into your progress.\nThis tool is very helpful for planning complex objectives, and for breaking down these larger complex objectives into smaller steps.\n\nIt is critical that you mark todos as completed as soon as you are done with a step. Do not batch up multiple steps before marking them as completed.\nFor simple objectives that only require a few steps, it is better to just complete the objective directly and NOT use this tool.\nWriting todos takes time and tokens, use it when it is helpful for managing complex many-step problems! But not for simple few-step requests.\n\n## Important To-Do List Usage Notes to Remember\n- The \\`write_todos\\` tool should never be called multiple times in parallel.\n- Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;\n\nconst TodoStatus = z\n  .enum([\"pending\", \"in_progress\", \"completed\"])\n  .describe(\"Status of the todo\");\nconst TodoSchema = z.object({\n  content: z.string().describe(\"Content of the todo item\"),\n  status: TodoStatus,\n});\nexport type Todo = z.infer<typeof TodoSchema>;\n\nconst stateSchema = z.object({\n  todos: z.array(TodoSchema).default([]),\n});\nexport type TodoMiddlewareState = z.infer<typeof stateSchema>;\n\nexport interface TodoListMiddlewareOptions {\n  /**\n   * Custom system prompt to guide the agent on using the todo tool.\n   * If not provided, uses the default {@link PLANNING_MIDDLEWARE_SYSTEM_PROMPT}.\n   */\n  systemPrompt?: string;\n  /**\n   * Custom description for the {@link writeTodos} tool.\n   * If not provided, uses the default {@link WRITE_TODOS_DESCRIPTION}.\n   */\n  toolDescription?: string;\n}\n\n/**\n * Creates a middleware that provides todo list management capabilities to agents.\n *\n * This middleware adds a `write_todos` tool that allows agents to create and manage\n * structured task lists for complex multi-step operations. It's designed to help\n * agents track progress, organize complex tasks, and provide users with visibility\n * into task completion status.\n *\n * The middleware automatically injects system prompts that guide the agent on when\n * and how to use the todo functionality effectively. It also enforces that the\n * `write_todos` tool is called at most once per model turn, since the tool replaces\n * the entire todo list and parallel calls would create ambiguity about precedence.\n *\n * @example\n * ```typescript\n * import { todoListMiddleware, createAgent } from 'langchain';\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [todoListMiddleware()],\n * });\n *\n * // Agent now has access to write_todos tool and todo state tracking\n * const result = await agent.invoke({\n *   messages: [new HumanMessage(\"Help me refactor my codebase\")]\n * });\n *\n * console.log(result.todos); // Array of todo items with status tracking\n * ```\n *\n * @returns A configured middleware instance that provides todo management capabilities\n *\n * @see {@link TodoMiddlewareState} for the state schema\n * @see {@link writeTodos} for the tool implementation\n */\nexport function todoListMiddleware(options?: TodoListMiddlewareOptions) {\n  /**\n   * Write todos tool - manages todo list with Command return\n   */\n  const writeTodos = tool(\n    ({ todos }, config) => {\n      return new Command({\n        update: {\n          todos,\n          messages: [\n            new ToolMessage({\n              content: `Updated todo list to ${JSON.stringify(todos)}`,\n              tool_call_id: config.toolCall?.id as string,\n              name: \"write_todos\",\n            }),\n          ],\n        },\n      });\n    },\n    {\n      name: \"write_todos\",\n      description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,\n      schema: z.object({\n        todos: z.array(TodoSchema).describe(\"List of todo items to update\"),\n      }),\n    }\n  );\n\n  return createMiddleware({\n    name: \"todoListMiddleware\",\n    stateSchema,\n    tools: [writeTodos],\n    wrapModelCall: (request, handler) =>\n      handler({\n        ...request,\n        systemMessage: request.systemMessage.concat(\n          `\\n\\n${options?.systemPrompt ?? TODO_LIST_MIDDLEWARE_SYSTEM_PROMPT}`\n        ),\n      }),\n    afterModel: (state) => {\n      /**\n       * Check for parallel write_todos tool calls and return errors if detected.\n       *\n       * The todo list is designed to be updated at most once per model turn. Since\n       * the `write_todos` tool replaces the entire todo list with each call, making\n       * multiple parallel calls would create ambiguity about which update should take\n       * precedence. This method prevents such conflicts by rejecting any response that\n       * contains multiple write_todos tool calls.\n       */\n      const messages = state.messages;\n      if (!messages || messages.length === 0) {\n        return undefined;\n      }\n\n      /**\n       * Find the last AI message\n       */\n      const lastAiMsg = [...messages]\n        .reverse()\n        .find((msg) => AIMessage.isInstance(msg));\n      if (\n        !lastAiMsg ||\n        !lastAiMsg.tool_calls ||\n        lastAiMsg.tool_calls.length === 0\n      ) {\n        return undefined;\n      }\n\n      /**\n       * Count write_todos tool calls\n       */\n      const writeTodosCalls = lastAiMsg.tool_calls.filter(\n        (tc) => tc.name === writeTodos.name\n      );\n\n      if (writeTodosCalls.length > 1) {\n        /**\n         * Create error tool messages for all write_todos calls\n         */\n        const errorMessages = writeTodosCalls.map(\n          (tc) =>\n            new ToolMessage({\n              content:\n                \"Error: The `write_todos` tool should never be called multiple times \" +\n                \"in parallel. Please call it only once per model invocation to update \" +\n                \"the todo list.\",\n              tool_call_id: tc.id as string,\n              name: \"write_todos\",\n              status: \"error\",\n            })\n        );\n\n        /**\n         * Keep the tool calls in the AI message but return error messages\n         * This follows the same pattern as HumanInTheLoopMiddleware\n         */\n        return { messages: errorMessages };\n      }\n\n      return undefined;\n    },\n  });\n}\n"],"mappings":";;;;;;;;;;;AAWA,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsNhC,MAAa,qCAAqC;;;;;;;;;;;;;AAclD,MAAM,aAAa,EAChB,KAAK;CAAC;CAAW;CAAe;AAAW,CAAC,CAAC,CAC7C,SAAS,oBAAoB;AAChC,MAAM,aAAa,EAAE,OAAO;CAC1B,SAAS,EAAE,OAAO,CAAC,CAAC,SAAS,0BAA0B;CACvD,QAAQ;AACV,CAAC;AAGD,MAAM,cAAc,EAAE,OAAO,EAC3B,OAAO,EAAE,MAAM,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,EACvC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDD,SAAgB,mBAAmB,SAAqC;;;;CAItE,MAAM,aAAa,MAChB,EAAE,SAAS,WAAW;EACrB,OAAO,IAAI,QAAQ,EACjB,QAAQ;GACN;GACA,UAAU,CACR,IAAI,YAAY;IACd,SAAS,wBAAwB,KAAK,UAAU,KAAK;IACrD,cAAc,OAAO,UAAU;IAC/B,MAAM;GACR,CAAC,CACH;EACF,EACF,CAAC;CACH,GACA;EACE,MAAM;EACN,aAAa,SAAS,mBAAmB;EACzC,QAAQ,EAAE,OAAO,EACf,OAAO,EAAE,MAAM,UAAU,CAAC,CAAC,SAAS,8BAA8B,EACpE,CAAC;CACH,CACF;CAEA,OAAO,iBAAiB;EACtB,MAAM;EACN;EACA,OAAO,CAAC,UAAU;EAClB,gBAAgB,SAAS,YACvB,QAAQ;GACN,GAAG;GACH,eAAe,QAAQ,cAAc,OACnC,OAAO,SAAS,gBAAgB,oCAClC;EACF,CAAC;EACH,aAAa,UAAU;;;;;;;;;;GAUrB,MAAM,WAAW,MAAM;GACvB,IAAI,CAAC,YAAY,SAAS,WAAW,GACnC;;;;GAMF,MAAM,YAAY,CAAC,GAAG,QAAQ,CAAC,CAC5B,QAAQ,CAAC,CACT,MAAM,QAAQ,UAAU,WAAW,GAAG,CAAC;GAC1C,IACE,CAAC,aACD,CAAC,UAAU,cACX,UAAU,WAAW,WAAW,GAEhC;;;;GAMF,MAAM,kBAAkB,UAAU,WAAW,QAC1C,OAAO,GAAG,SAAS,WAAW,IACjC;GAEA,IAAI,gBAAgB,SAAS;;;;;GAqB3B,OAAO,EAAE,UAjBa,gBAAgB,KACnC,OACC,IAAI,YAAY;IACd,SACE;IAGF,cAAc,GAAG;IACjB,MAAM;IACN,QAAQ;GACV,CAAC,CAO0B,EAAE;EAIrC;CACF,CAAC;AACH"}

package/dist/agents/middleware/toolCallLimit.cjs

@@ -1,4 +1,3 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 const require_middleware = require("../middleware.cjs");
 let _langchain_core_messages = require("@langchain/core/messages");
 let zod_v4 = require("zod/v4");

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

@@ -1 +1 @@
-{"version":3,"file":"toolCallLimit.cjs","names":["z","z4","createMiddleware","AIMessage","ToolMessage"],"sources":["../../../src/agents/middleware/toolCallLimit.ts"],"sourcesContent":["import { AIMessage, ToolMessage } from \"@langchain/core/messages\";\nimport { z as z4 } from \"zod/v4\";\nimport { z } from \"zod/v3\";\nimport type { InferInteropZodInput } from \"@langchain/core/utils/types\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\n\nimport { createMiddleware } from \"../middleware.js\";\n\n/**\n * Build the error message content for ToolMessage when limit is exceeded.\n *\n * This message is sent to the model, so it should not reference thread/run concepts\n * that the model has no notion of.\n *\n * @param toolName - Tool name being limited (if specific tool), or undefined for all tools.\n * @returns A concise message instructing the model not to call the tool again.\n */\nfunction buildToolMessageContent(toolName: string | undefined): string {\n  // Always instruct the model not to call again, regardless of which limit was hit\n  if (toolName) {\n    return `Tool call limit exceeded. Do not call '${toolName}' again.`;\n  }\n  return \"Tool call limit exceeded. Do not make additional tool calls.\";\n}\n\nconst VALID_EXIT_BEHAVIORS = [\"continue\", \"error\", \"end\"] as const;\nconst DEFAULT_EXIT_BEHAVIOR = \"continue\";\n\n/**\n * Build the final AI message content for 'end' behavior.\n *\n * This message is displayed to the user, so it should include detailed information\n * about which limits were exceeded.\n *\n * @param threadCount - Current thread tool call count.\n * @param runCount - Current run tool call count.\n * @param threadLimit - Thread tool call limit (if set).\n * @param runLimit - Run tool call limit (if set).\n * @param toolName - Tool name being limited (if specific tool), or undefined for all tools.\n * @returns A formatted message describing which limits were exceeded.\n */\nfunction buildFinalAIMessageContent(\n  threadCount: number,\n  runCount: number,\n  threadLimit: number | undefined,\n  runLimit: number | undefined,\n  toolName: string | undefined\n): string {\n  const toolDesc = toolName ? `'${toolName}' tool` : \"Tool\";\n  const exceededLimits: string[] = [];\n\n  if (threadLimit !== undefined && threadCount > threadLimit) {\n    exceededLimits.push(\n      `thread limit exceeded (${threadCount}/${threadLimit} calls)`\n    );\n  }\n  if (runLimit !== undefined && runCount > runLimit) {\n    exceededLimits.push(`run limit exceeded (${runCount}/${runLimit} calls)`);\n  }\n\n  const limitsText = exceededLimits.join(\" and \");\n  return `${toolDesc} call limit reached: ${limitsText}.`;\n}\n\n/**\n * Schema for the exit behavior.\n */\nconst exitBehaviorSchema = z\n  .enum(VALID_EXIT_BEHAVIORS)\n  .default(DEFAULT_EXIT_BEHAVIOR);\n\n/**\n * Exception raised when tool call limits are exceeded.\n *\n * This exception is raised when the configured exit behavior is 'error'\n * and either the thread or run tool call limit has been exceeded.\n */\nexport class ToolCallLimitExceededError extends Error {\n  /**\n   * Current thread tool call count.\n   */\n  threadCount: number;\n  /**\n   * Current run tool call count.\n   */\n  runCount: number;\n  /**\n   * Thread tool call limit (if set).\n   */\n  threadLimit: number | undefined;\n  /**\n   * Run tool call limit (if set).\n   */\n  runLimit: number | undefined;\n  /**\n   * Tool name being limited (if specific tool), or undefined for all tools.\n   */\n  toolName: string | undefined;\n\n  constructor(\n    threadCount: number,\n    runCount: number,\n    threadLimit: number | undefined,\n    runLimit: number | undefined,\n    toolName: string | undefined = undefined\n  ) {\n    const message = buildFinalAIMessageContent(\n      threadCount,\n      runCount,\n      threadLimit,\n      runLimit,\n      toolName\n    );\n    super(message);\n\n    this.name = \"ToolCallLimitExceededError\";\n    this.threadCount = threadCount;\n    this.runCount = runCount;\n    this.threadLimit = threadLimit;\n    this.runLimit = runLimit;\n    this.toolName = toolName;\n  }\n}\n\n/**\n * Options for configuring the Tool Call Limit middleware.\n */\nexport const ToolCallLimitOptionsSchema = z.object({\n  /**\n   * Name of the specific tool to limit. If undefined, limits apply to all tools.\n   */\n  toolName: z.string().optional(),\n  /**\n   * Maximum number of tool calls allowed per thread.\n   * undefined means no limit.\n   */\n  threadLimit: z.number().optional(),\n  /**\n   * Maximum number of tool calls allowed per run.\n   * undefined means no limit.\n   */\n  runLimit: z.number().optional(),\n  /**\n   * What to do when limits are exceeded.\n   * - \"continue\": Block exceeded tools with error messages, let other tools continue (default)\n   * - \"error\": Raise a ToolCallLimitExceededError exception\n   * - \"end\": Stop execution immediately, injecting a ToolMessage and an AI message\n   *   for the single tool call that exceeded the limit. Raises NotImplementedError\n   *   if there are multiple tool calls.\n   *\n   * @default \"continue\"\n   */\n  exitBehavior: exitBehaviorSchema,\n});\n\nexport type ToolCallLimitConfig = InferInteropZodInput<\n  typeof ToolCallLimitOptionsSchema\n>;\n\n/**\n * Middleware state schema to track the number of model calls made at the thread and run level.\n */\nconst stateSchema = z.object({\n  threadToolCallCount: z.record(z.string(), z.number()).default({}),\n  runToolCallCount: z.record(z.string(), z.number()).default({}),\n});\n\nconst DEFAULT_TOOL_COUNT_KEY = \"__all__\";\n\n/**\n * Middleware that tracks tool call counts and enforces limits.\n *\n * This middleware monitors the number of tool calls made during agent execution\n * and can terminate the agent when specified limits are reached. It supports\n * both thread-level and run-level call counting with configurable exit behaviors.\n *\n * Thread-level: The middleware counts all tool calls in the entire message history\n * and persists this count across multiple runs (invocations) of the agent.\n *\n * Run-level: The middleware counts tool calls made after the last HumanMessage,\n * representing the current run (invocation) of the agent.\n *\n * @param options - Configuration options for the middleware\n * @param options.toolName - Name of the specific tool to limit. If undefined, limits apply to all tools.\n * @param options.threadLimit - Maximum number of tool calls allowed per thread. undefined means no limit.\n * @param options.runLimit - Maximum number of tool calls allowed per run. undefined means no limit.\n * @param options.exitBehavior - What to do when limits are exceeded.\n *   - \"continue\": Block exceeded tools with error messages, let other tools continue. Model decides when to end. (default)\n *   - \"error\": Raise a ToolCallLimitExceededError exception\n *   - \"end\": Stop execution immediately with a ToolMessage + AI message for the single tool call that exceeded the limit. Raises NotImplementedError if there are multiple tool calls.\n *\n * @throws {Error} If both limits are undefined, if exitBehavior is invalid, or if runLimit exceeds threadLimit.\n * @throws {NotImplementedError} If exitBehavior is \"end\" and there are multiple tool calls.\n *\n * @example Continue execution with blocked tools (default)\n * ```ts\n * import { toolCallLimitMiddleware } from \"@langchain/langchain/agents/middleware\";\n * import { createAgent } from \"@langchain/langchain/agents\";\n *\n * // Block exceeded tools but let other tools and model continue\n * const limiter = toolCallLimitMiddleware({\n *   threadLimit: 20,\n *   runLimit: 10,\n *   exitBehavior: \"continue\", // default\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n * ```\n *\n * @example Stop immediately when limit exceeded\n * ```ts\n * // End execution immediately with an AI message\n * const limiter = toolCallLimitMiddleware({\n *   runLimit: 5,\n *   exitBehavior: \"end\"\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n * ```\n *\n * @example Raise exception on limit\n * ```ts\n * // Strict limit with exception handling\n * const limiter = toolCallLimitMiddleware({\n *   toolName: \"search\",\n *   threadLimit: 5,\n *   exitBehavior: \"error\"\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n *\n * try {\n *   const result = await agent.invoke({ messages: [new HumanMessage(\"Task\")] });\n * } catch (error) {\n *   if (error instanceof ToolCallLimitExceededError) {\n *     console.log(`Search limit exceeded: ${error}`);\n *   }\n * }\n * ```\n */\nexport function toolCallLimitMiddleware(options: ToolCallLimitConfig) {\n  /**\n   * Validate that at least one limit is specified\n   */\n  if (options.threadLimit === undefined && options.runLimit === undefined) {\n    throw new Error(\n      \"At least one limit must be specified (threadLimit or runLimit)\"\n    );\n  }\n\n  /**\n   * Validate exitBehavior (Zod schema already validates, but provide helpful error)\n   */\n  const exitBehavior = options.exitBehavior ?? DEFAULT_EXIT_BEHAVIOR;\n  const parseResult = exitBehaviorSchema.safeParse(exitBehavior);\n  if (!parseResult.success) {\n    throw new Error(z4.prettifyError(parseResult.error).slice(2));\n  }\n\n  /**\n   * Validate that runLimit does not exceed threadLimit\n   */\n  if (\n    options.threadLimit !== undefined &&\n    options.runLimit !== undefined &&\n    options.runLimit > options.threadLimit\n  ) {\n    throw new Error(\n      `runLimit (${options.runLimit}) cannot exceed threadLimit (${options.threadLimit}). ` +\n        \"The run limit should be less than or equal to the thread limit.\"\n    );\n  }\n\n  /**\n   * Generate the middleware name based on the tool name\n   */\n  const middlewareName = options.toolName\n    ? `ToolCallLimitMiddleware[${options.toolName}]`\n    : \"ToolCallLimitMiddleware\";\n\n  return createMiddleware({\n    name: middlewareName,\n    stateSchema,\n    afterModel: {\n      canJumpTo: [\"end\"],\n      hook: (state) => {\n        /**\n         * Get the last AI message to check for tool calls\n         */\n        const lastAIMessage = [...state.messages]\n          .reverse()\n          .find(AIMessage.isInstance);\n\n        if (!lastAIMessage || !lastAIMessage.tool_calls) {\n          return undefined;\n        }\n\n        /**\n         * Helper to check if limit would be exceeded by one more call\n         */\n        const wouldExceedLimit = (\n          threadCount: number,\n          runCount: number\n        ): boolean => {\n          return (\n            (options.threadLimit !== undefined &&\n              threadCount + 1 > options.threadLimit) ||\n            (options.runLimit !== undefined && runCount + 1 > options.runLimit)\n          );\n        };\n\n        /**\n         * Helper to check if a tool call matches our filter\n         */\n        const matchesToolFilter = (toolCall: { name?: string }): boolean => {\n          return (\n            options.toolName === undefined || toolCall.name === options.toolName\n          );\n        };\n\n        /**\n         * Separate tool calls into allowed and blocked based on limits\n         */\n        const separateToolCalls = (\n          toolCalls: ToolCall[],\n          threadCount: number,\n          runCount: number\n        ): {\n          allowed: ToolCall[];\n          blocked: ToolCall[];\n          finalThreadCount: number;\n          finalRunCount: number;\n        } => {\n          const allowed: ToolCall[] = [];\n          const blocked: ToolCall[] = [];\n          let tempThreadCount = threadCount;\n          let tempRunCount = runCount;\n\n          for (const toolCall of toolCalls) {\n            if (!matchesToolFilter(toolCall)) {\n              // Tool call doesn't match our filter, skip it\n              continue;\n            }\n\n            if (wouldExceedLimit(tempThreadCount, tempRunCount)) {\n              blocked.push(toolCall);\n            } else {\n              allowed.push(toolCall);\n              tempThreadCount += 1;\n              tempRunCount += 1;\n            }\n          }\n\n          return {\n            allowed,\n            blocked,\n            finalThreadCount: tempThreadCount,\n            finalRunCount: tempRunCount + blocked.length,\n          };\n        };\n\n        /**\n         * Get the count key for this middleware instance\n         */\n        const countKey = options.toolName ?? DEFAULT_TOOL_COUNT_KEY;\n\n        /**\n         * Get current counts\n         */\n        const threadCounts = { ...(state.threadToolCallCount ?? {}) };\n        const runCounts = { ...(state.runToolCallCount ?? {}) };\n        const currentThreadCount = threadCounts[countKey] ?? 0;\n        const currentRunCount = runCounts[countKey] ?? 0;\n\n        /**\n         * Separate tool calls into allowed and blocked\n         */\n        const { allowed, blocked, finalThreadCount, finalRunCount } =\n          separateToolCalls(\n            lastAIMessage.tool_calls,\n            currentThreadCount,\n            currentRunCount\n          );\n\n        /**\n         * Update counts:\n         * - Thread count includes only allowed calls (blocked calls don't count towards thread-level tracking)\n         * - Run count includes blocked calls since they were attempted in this run\n         */\n        threadCounts[countKey] = finalThreadCount;\n        runCounts[countKey] = finalRunCount;\n\n        /**\n         * If no tool calls are blocked, just update counts\n         */\n        if (blocked.length === 0) {\n          if (allowed.length > 0) {\n            return {\n              threadToolCallCount: threadCounts,\n              runToolCallCount: runCounts,\n            };\n          }\n          return undefined;\n        }\n\n        /**\n         * Handle different exit behaviors\n         */\n        if (exitBehavior === \"error\") {\n          // Use hypothetical thread count to show which limit was exceeded\n          const hypotheticalThreadCount = finalThreadCount + blocked.length;\n          throw new ToolCallLimitExceededError(\n            hypotheticalThreadCount,\n            finalRunCount,\n            options.threadLimit,\n            options.runLimit,\n            options.toolName\n          );\n        }\n\n        /**\n         * Build tool message content (sent to model - no thread/run details)\n         */\n        const toolMsgContent = buildToolMessageContent(options.toolName);\n\n        /**\n         * Inject artificial error ToolMessages for blocked tool calls\n         */\n        const artificialMessages: Array<ToolMessage | AIMessage> = blocked.map(\n          (toolCall) =>\n            new ToolMessage({\n              content: toolMsgContent,\n              tool_call_id: toolCall.id!,\n              name: toolCall.name,\n              status: \"error\",\n            })\n        );\n\n        if (exitBehavior === \"end\") {\n          /**\n           * Check if there are tool calls to other tools that would continue executing\n           * For tool-specific limiters: check for calls to other tools\n           * For global limiters: check if there are multiple different tool types\n           */\n          let otherTools: ToolCall[] = [];\n          if (options.toolName !== undefined) {\n            /**\n             * Tool-specific limiter: check for calls to other tools\n             */\n            otherTools = lastAIMessage.tool_calls.filter(\n              (tc) => tc.name !== options.toolName\n            );\n          } else {\n            /**\n             * Global limiter: check if there are multiple different tool types\n             * If there are allowed calls, those would execute\n             * But even if all are blocked, we can't handle multiple tool types with \"end\"\n             */\n            const uniqueToolNames = new Set(\n              lastAIMessage.tool_calls.map((tc) => tc.name).filter(Boolean)\n            );\n            if (uniqueToolNames.size > 1) {\n              /**\n               * Multiple different tool types - use allowed calls to show which ones\n               */\n              otherTools =\n                allowed.length > 0 ? allowed : lastAIMessage.tool_calls;\n            }\n          }\n\n          if (otherTools.length > 0) {\n            const toolNames = Array.from(\n              new Set(otherTools.map((tc) => tc.name).filter(Boolean))\n            ).join(\", \");\n            throw new Error(\n              `Cannot end execution with other tool calls pending. Found calls to: ${toolNames}. Use 'continue' or 'error' behavior instead.`\n            );\n          }\n\n          /**\n           * Build final AI message content (displayed to user - includes thread/run details)\n           * Use hypothetical thread count (what it would have been if call wasn't blocked)\n           * to show which limit was actually exceeded\n           */\n          const hypotheticalThreadCount = finalThreadCount + blocked.length;\n          const finalMsgContent = buildFinalAIMessageContent(\n            hypotheticalThreadCount,\n            finalRunCount,\n            options.threadLimit,\n            options.runLimit,\n            options.toolName\n          );\n          artificialMessages.push(new AIMessage(finalMsgContent));\n\n          return {\n            threadToolCallCount: threadCounts,\n            runToolCallCount: runCounts,\n            jumpTo: \"end\" as const,\n            messages: artificialMessages,\n          };\n        }\n\n        /**\n         * For exit_behavior=\"continue\", return error messages to block exceeded tools\n         */\n        return {\n          threadToolCallCount: threadCounts,\n          runToolCallCount: runCounts,\n          messages: artificialMessages,\n        };\n      },\n    },\n    /**\n     * reset the run tool call count after the agent execution completes\n     */\n    afterAgent: () => ({\n      runToolCallCount: {},\n    }),\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;AAiBA,SAAS,wBAAwB,UAAsC;AAErE,KAAI,SACF,QAAO,0CAA0C,SAAS;AAE5D,QAAO;;AAGT,MAAM,uBAAuB;CAAC;CAAY;CAAS;CAAM;AACzD,MAAM,wBAAwB;;;;;;;;;;;;;;AAe9B,SAAS,2BACP,aACA,UACA,aACA,UACA,UACQ;CACR,MAAM,WAAW,WAAW,IAAI,SAAS,UAAU;CACnD,MAAM,iBAA2B,EAAE;AAEnC,KAAI,gBAAgB,KAAA,KAAa,cAAc,YAC7C,gBAAe,KACb,0BAA0B,YAAY,GAAG,YAAY,SACtD;AAEH,KAAI,aAAa,KAAA,KAAa,WAAW,SACvC,gBAAe,KAAK,uBAAuB,SAAS,GAAG,SAAS,SAAS;AAI3E,QAAO,GAAG,SAAS,uBADA,eAAe,KAAK,QACa,CAAC;;;;;AAMvD,MAAM,qBAAqBA,OAAAA,EACxB,KAAK,qBAAqB,CAC1B,QAAQ,sBAAsB;;;;;;;AAQjC,IAAa,6BAAb,cAAgD,MAAM;;;;CAIpD;;;;CAIA;;;;CAIA;;;;CAIA;;;;CAIA;CAEA,YACE,aACA,UACA,aACA,UACA,WAA+B,KAAA,GAC/B;EACA,MAAM,UAAU,2BACd,aACA,UACA,aACA,UACA,SACD;AACD,QAAM,QAAQ;AAEd,OAAK,OAAO;AACZ,OAAK,cAAc;AACnB,OAAK,WAAW;AAChB,OAAK,cAAc;AACnB,OAAK,WAAW;AAChB,OAAK,WAAW;;;AAOsBA,OAAAA,EAAE,OAAO;;;;CAIjD,UAAUA,OAAAA,EAAE,QAAQ,CAAC,UAAU;;;;;CAK/B,aAAaA,OAAAA,EAAE,QAAQ,CAAC,UAAU;;;;;CAKlC,UAAUA,OAAAA,EAAE,QAAQ,CAAC,UAAU;;;;;;;;;;;CAW/B,cAAc;CACf,CAAC;;;;AASF,MAAM,cAAcA,OAAAA,EAAE,OAAO;CAC3B,qBAAqBA,OAAAA,EAAE,OAAOA,OAAAA,EAAE,QAAQ,EAAEA,OAAAA,EAAE,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;CACjE,kBAAkBA,OAAAA,EAAE,OAAOA,OAAAA,EAAE,QAAQ,EAAEA,OAAAA,EAAE,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;CAC/D,CAAC;AAEF,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkF/B,SAAgB,wBAAwB,SAA8B;;;;AAIpE,KAAI,QAAQ,gBAAgB,KAAA,KAAa,QAAQ,aAAa,KAAA,EAC5D,OAAM,IAAI,MACR,iEACD;;;;CAMH,MAAM,eAAe,QAAQ,gBAAgB;CAC7C,MAAM,cAAc,mBAAmB,UAAU,aAAa;AAC9D,KAAI,CAAC,YAAY,QACf,OAAM,IAAI,MAAMC,OAAAA,EAAG,cAAc,YAAY,MAAM,CAAC,MAAM,EAAE,CAAC;;;;AAM/D,KACE,QAAQ,gBAAgB,KAAA,KACxB,QAAQ,aAAa,KAAA,KACrB,QAAQ,WAAW,QAAQ,YAE3B,OAAM,IAAI,MACR,aAAa,QAAQ,SAAS,+BAA+B,QAAQ,YAAY,oEAElF;AAUH,QAAOC,mBAAAA,iBAAiB;EACtB,MALqB,QAAQ,WAC3B,2BAA2B,QAAQ,SAAS,KAC5C;EAIF;EACA,YAAY;GACV,WAAW,CAAC,MAAM;GAClB,OAAO,UAAU;;;;IAIf,MAAM,gBAAgB,CAAC,GAAG,MAAM,SAAS,CACtC,SAAS,CACT,KAAKC,yBAAAA,UAAU,WAAW;AAE7B,QAAI,CAAC,iBAAiB,CAAC,cAAc,WACnC;;;;IAMF,MAAM,oBACJ,aACA,aACY;AACZ,YACG,QAAQ,gBAAgB,KAAA,KACvB,cAAc,IAAI,QAAQ,eAC3B,QAAQ,aAAa,KAAA,KAAa,WAAW,IAAI,QAAQ;;;;;IAO9D,MAAM,qBAAqB,aAAyC;AAClE,YACE,QAAQ,aAAa,KAAA,KAAa,SAAS,SAAS,QAAQ;;;;;IAOhE,MAAM,qBACJ,WACA,aACA,aAMG;KACH,MAAM,UAAsB,EAAE;KAC9B,MAAM,UAAsB,EAAE;KAC9B,IAAI,kBAAkB;KACtB,IAAI,eAAe;AAEnB,UAAK,MAAM,YAAY,WAAW;AAChC,UAAI,CAAC,kBAAkB,SAAS,CAE9B;AAGF,UAAI,iBAAiB,iBAAiB,aAAa,CACjD,SAAQ,KAAK,SAAS;WACjB;AACL,eAAQ,KAAK,SAAS;AACtB,0BAAmB;AACnB,uBAAgB;;;AAIpB,YAAO;MACL;MACA;MACA,kBAAkB;MAClB,eAAe,eAAe,QAAQ;MACvC;;;;;IAMH,MAAM,WAAW,QAAQ,YAAY;;;;IAKrC,MAAM,eAAe,EAAE,GAAI,MAAM,uBAAuB,EAAE,EAAG;IAC7D,MAAM,YAAY,EAAE,GAAI,MAAM,oBAAoB,EAAE,EAAG;IACvD,MAAM,qBAAqB,aAAa,aAAa;IACrD,MAAM,kBAAkB,UAAU,aAAa;;;;IAK/C,MAAM,EAAE,SAAS,SAAS,kBAAkB,kBAC1C,kBACE,cAAc,YACd,oBACA,gBACD;;;;;;AAOH,iBAAa,YAAY;AACzB,cAAU,YAAY;;;;AAKtB,QAAI,QAAQ,WAAW,GAAG;AACxB,SAAI,QAAQ,SAAS,EACnB,QAAO;MACL,qBAAqB;MACrB,kBAAkB;MACnB;AAEH;;;;;AAMF,QAAI,iBAAiB,QAGnB,OAAM,IAAI,2BADsB,mBAAmB,QAAQ,QAGzD,eACA,QAAQ,aACR,QAAQ,UACR,QAAQ,SACT;;;;IAMH,MAAM,iBAAiB,wBAAwB,QAAQ,SAAS;;;;IAKhE,MAAM,qBAAqD,QAAQ,KAChE,aACC,IAAIC,yBAAAA,YAAY;KACd,SAAS;KACT,cAAc,SAAS;KACvB,MAAM,SAAS;KACf,QAAQ;KACT,CAAC,CACL;AAED,QAAI,iBAAiB,OAAO;;;;;;KAM1B,IAAI,aAAyB,EAAE;AAC/B,SAAI,QAAQ,aAAa,KAAA;;;;AAIvB,kBAAa,cAAc,WAAW,QACnC,OAAO,GAAG,SAAS,QAAQ,SAC7B;cAUG,IAHwB,IAC1B,cAAc,WAAW,KAAK,OAAO,GAAG,KAAK,CAAC,OAAO,QAAQ,CAE5C,CAAC,OAAO;;;;AAIzB,kBACE,QAAQ,SAAS,IAAI,UAAU,cAAc;AAInD,SAAI,WAAW,SAAS,GAAG;MACzB,MAAM,YAAY,MAAM,KACtB,IAAI,IAAI,WAAW,KAAK,OAAO,GAAG,KAAK,CAAC,OAAO,QAAQ,CAAC,CACzD,CAAC,KAAK,KAAK;AACZ,YAAM,IAAI,MACR,uEAAuE,UAAU,+CAClF;;KASH,MAAM,kBAAkB,2BADQ,mBAAmB,QAAQ,QAGzD,eACA,QAAQ,aACR,QAAQ,UACR,QAAQ,SACT;AACD,wBAAmB,KAAK,IAAID,yBAAAA,UAAU,gBAAgB,CAAC;AAEvD,YAAO;MACL,qBAAqB;MACrB,kBAAkB;MAClB,QAAQ;MACR,UAAU;MACX;;;;;AAMH,WAAO;KACL,qBAAqB;KACrB,kBAAkB;KAClB,UAAU;KACX;;GAEJ;;;;EAID,mBAAmB,EACjB,kBAAkB,EAAE,EACrB;EACF,CAAC"}
+{"version":3,"file":"toolCallLimit.cjs","names":["z","z4","createMiddleware","AIMessage","ToolMessage"],"sources":["../../../src/agents/middleware/toolCallLimit.ts"],"sourcesContent":["import { AIMessage, ToolMessage } from \"@langchain/core/messages\";\nimport { z as z4 } from \"zod/v4\";\nimport { z } from \"zod/v3\";\nimport type { InferInteropZodInput } from \"@langchain/core/utils/types\";\nimport type { ToolCall } from \"@langchain/core/messages/tool\";\n\nimport { createMiddleware } from \"../middleware.js\";\n\n/**\n * Build the error message content for ToolMessage when limit is exceeded.\n *\n * This message is sent to the model, so it should not reference thread/run concepts\n * that the model has no notion of.\n *\n * @param toolName - Tool name being limited (if specific tool), or undefined for all tools.\n * @returns A concise message instructing the model not to call the tool again.\n */\nfunction buildToolMessageContent(toolName: string | undefined): string {\n  // Always instruct the model not to call again, regardless of which limit was hit\n  if (toolName) {\n    return `Tool call limit exceeded. Do not call '${toolName}' again.`;\n  }\n  return \"Tool call limit exceeded. Do not make additional tool calls.\";\n}\n\nconst VALID_EXIT_BEHAVIORS = [\"continue\", \"error\", \"end\"] as const;\nconst DEFAULT_EXIT_BEHAVIOR = \"continue\";\n\n/**\n * Build the final AI message content for 'end' behavior.\n *\n * This message is displayed to the user, so it should include detailed information\n * about which limits were exceeded.\n *\n * @param threadCount - Current thread tool call count.\n * @param runCount - Current run tool call count.\n * @param threadLimit - Thread tool call limit (if set).\n * @param runLimit - Run tool call limit (if set).\n * @param toolName - Tool name being limited (if specific tool), or undefined for all tools.\n * @returns A formatted message describing which limits were exceeded.\n */\nfunction buildFinalAIMessageContent(\n  threadCount: number,\n  runCount: number,\n  threadLimit: number | undefined,\n  runLimit: number | undefined,\n  toolName: string | undefined\n): string {\n  const toolDesc = toolName ? `'${toolName}' tool` : \"Tool\";\n  const exceededLimits: string[] = [];\n\n  if (threadLimit !== undefined && threadCount > threadLimit) {\n    exceededLimits.push(\n      `thread limit exceeded (${threadCount}/${threadLimit} calls)`\n    );\n  }\n  if (runLimit !== undefined && runCount > runLimit) {\n    exceededLimits.push(`run limit exceeded (${runCount}/${runLimit} calls)`);\n  }\n\n  const limitsText = exceededLimits.join(\" and \");\n  return `${toolDesc} call limit reached: ${limitsText}.`;\n}\n\n/**\n * Schema for the exit behavior.\n */\nconst exitBehaviorSchema = z\n  .enum(VALID_EXIT_BEHAVIORS)\n  .default(DEFAULT_EXIT_BEHAVIOR);\n\n/**\n * Exception raised when tool call limits are exceeded.\n *\n * This exception is raised when the configured exit behavior is 'error'\n * and either the thread or run tool call limit has been exceeded.\n */\nexport class ToolCallLimitExceededError extends Error {\n  /**\n   * Current thread tool call count.\n   */\n  threadCount: number;\n  /**\n   * Current run tool call count.\n   */\n  runCount: number;\n  /**\n   * Thread tool call limit (if set).\n   */\n  threadLimit: number | undefined;\n  /**\n   * Run tool call limit (if set).\n   */\n  runLimit: number | undefined;\n  /**\n   * Tool name being limited (if specific tool), or undefined for all tools.\n   */\n  toolName: string | undefined;\n\n  constructor(\n    threadCount: number,\n    runCount: number,\n    threadLimit: number | undefined,\n    runLimit: number | undefined,\n    toolName: string | undefined = undefined\n  ) {\n    const message = buildFinalAIMessageContent(\n      threadCount,\n      runCount,\n      threadLimit,\n      runLimit,\n      toolName\n    );\n    super(message);\n\n    this.name = \"ToolCallLimitExceededError\";\n    this.threadCount = threadCount;\n    this.runCount = runCount;\n    this.threadLimit = threadLimit;\n    this.runLimit = runLimit;\n    this.toolName = toolName;\n  }\n}\n\n/**\n * Options for configuring the Tool Call Limit middleware.\n */\nexport const ToolCallLimitOptionsSchema = z.object({\n  /**\n   * Name of the specific tool to limit. If undefined, limits apply to all tools.\n   */\n  toolName: z.string().optional(),\n  /**\n   * Maximum number of tool calls allowed per thread.\n   * undefined means no limit.\n   */\n  threadLimit: z.number().optional(),\n  /**\n   * Maximum number of tool calls allowed per run.\n   * undefined means no limit.\n   */\n  runLimit: z.number().optional(),\n  /**\n   * What to do when limits are exceeded.\n   * - \"continue\": Block exceeded tools with error messages, let other tools continue (default)\n   * - \"error\": Raise a ToolCallLimitExceededError exception\n   * - \"end\": Stop execution immediately, injecting a ToolMessage and an AI message\n   *   for the single tool call that exceeded the limit. Raises NotImplementedError\n   *   if there are multiple tool calls.\n   *\n   * @default \"continue\"\n   */\n  exitBehavior: exitBehaviorSchema,\n});\n\nexport type ToolCallLimitConfig = InferInteropZodInput<\n  typeof ToolCallLimitOptionsSchema\n>;\n\n/**\n * Middleware state schema to track the number of model calls made at the thread and run level.\n */\nconst stateSchema = z.object({\n  threadToolCallCount: z.record(z.string(), z.number()).default({}),\n  runToolCallCount: z.record(z.string(), z.number()).default({}),\n});\n\nconst DEFAULT_TOOL_COUNT_KEY = \"__all__\";\n\n/**\n * Middleware that tracks tool call counts and enforces limits.\n *\n * This middleware monitors the number of tool calls made during agent execution\n * and can terminate the agent when specified limits are reached. It supports\n * both thread-level and run-level call counting with configurable exit behaviors.\n *\n * Thread-level: The middleware counts all tool calls in the entire message history\n * and persists this count across multiple runs (invocations) of the agent.\n *\n * Run-level: The middleware counts tool calls made after the last HumanMessage,\n * representing the current run (invocation) of the agent.\n *\n * @param options - Configuration options for the middleware\n * @param options.toolName - Name of the specific tool to limit. If undefined, limits apply to all tools.\n * @param options.threadLimit - Maximum number of tool calls allowed per thread. undefined means no limit.\n * @param options.runLimit - Maximum number of tool calls allowed per run. undefined means no limit.\n * @param options.exitBehavior - What to do when limits are exceeded.\n *   - \"continue\": Block exceeded tools with error messages, let other tools continue. Model decides when to end. (default)\n *   - \"error\": Raise a ToolCallLimitExceededError exception\n *   - \"end\": Stop execution immediately with a ToolMessage + AI message for the single tool call that exceeded the limit. Raises NotImplementedError if there are multiple tool calls.\n *\n * @throws {Error} If both limits are undefined, if exitBehavior is invalid, or if runLimit exceeds threadLimit.\n * @throws {NotImplementedError} If exitBehavior is \"end\" and there are multiple tool calls.\n *\n * @example Continue execution with blocked tools (default)\n * ```ts\n * import { toolCallLimitMiddleware } from \"@langchain/langchain/agents/middleware\";\n * import { createAgent } from \"@langchain/langchain/agents\";\n *\n * // Block exceeded tools but let other tools and model continue\n * const limiter = toolCallLimitMiddleware({\n *   threadLimit: 20,\n *   runLimit: 10,\n *   exitBehavior: \"continue\", // default\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n * ```\n *\n * @example Stop immediately when limit exceeded\n * ```ts\n * // End execution immediately with an AI message\n * const limiter = toolCallLimitMiddleware({\n *   runLimit: 5,\n *   exitBehavior: \"end\"\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n * ```\n *\n * @example Raise exception on limit\n * ```ts\n * // Strict limit with exception handling\n * const limiter = toolCallLimitMiddleware({\n *   toolName: \"search\",\n *   threadLimit: 5,\n *   exitBehavior: \"error\"\n * });\n *\n * const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   middleware: [limiter]\n * });\n *\n * try {\n *   const result = await agent.invoke({ messages: [new HumanMessage(\"Task\")] });\n * } catch (error) {\n *   if (error instanceof ToolCallLimitExceededError) {\n *     console.log(`Search limit exceeded: ${error}`);\n *   }\n * }\n * ```\n */\nexport function toolCallLimitMiddleware(options: ToolCallLimitConfig) {\n  /**\n   * Validate that at least one limit is specified\n   */\n  if (options.threadLimit === undefined && options.runLimit === undefined) {\n    throw new Error(\n      \"At least one limit must be specified (threadLimit or runLimit)\"\n    );\n  }\n\n  /**\n   * Validate exitBehavior (Zod schema already validates, but provide helpful error)\n   */\n  const exitBehavior = options.exitBehavior ?? DEFAULT_EXIT_BEHAVIOR;\n  const parseResult = exitBehaviorSchema.safeParse(exitBehavior);\n  if (!parseResult.success) {\n    throw new Error(z4.prettifyError(parseResult.error).slice(2));\n  }\n\n  /**\n   * Validate that runLimit does not exceed threadLimit\n   */\n  if (\n    options.threadLimit !== undefined &&\n    options.runLimit !== undefined &&\n    options.runLimit > options.threadLimit\n  ) {\n    throw new Error(\n      `runLimit (${options.runLimit}) cannot exceed threadLimit (${options.threadLimit}). ` +\n        \"The run limit should be less than or equal to the thread limit.\"\n    );\n  }\n\n  /**\n   * Generate the middleware name based on the tool name\n   */\n  const middlewareName = options.toolName\n    ? `ToolCallLimitMiddleware[${options.toolName}]`\n    : \"ToolCallLimitMiddleware\";\n\n  return createMiddleware({\n    name: middlewareName,\n    stateSchema,\n    afterModel: {\n      canJumpTo: [\"end\"],\n      hook: (state) => {\n        /**\n         * Get the last AI message to check for tool calls\n         */\n        const lastAIMessage = [...state.messages]\n          .reverse()\n          .find(AIMessage.isInstance);\n\n        if (!lastAIMessage || !lastAIMessage.tool_calls) {\n          return undefined;\n        }\n\n        /**\n         * Helper to check if limit would be exceeded by one more call\n         */\n        const wouldExceedLimit = (\n          threadCount: number,\n          runCount: number\n        ): boolean => {\n          return (\n            (options.threadLimit !== undefined &&\n              threadCount + 1 > options.threadLimit) ||\n            (options.runLimit !== undefined && runCount + 1 > options.runLimit)\n          );\n        };\n\n        /**\n         * Helper to check if a tool call matches our filter\n         */\n        const matchesToolFilter = (toolCall: { name?: string }): boolean => {\n          return (\n            options.toolName === undefined || toolCall.name === options.toolName\n          );\n        };\n\n        /**\n         * Separate tool calls into allowed and blocked based on limits\n         */\n        const separateToolCalls = (\n          toolCalls: ToolCall[],\n          threadCount: number,\n          runCount: number\n        ): {\n          allowed: ToolCall[];\n          blocked: ToolCall[];\n          finalThreadCount: number;\n          finalRunCount: number;\n        } => {\n          const allowed: ToolCall[] = [];\n          const blocked: ToolCall[] = [];\n          let tempThreadCount = threadCount;\n          let tempRunCount = runCount;\n\n          for (const toolCall of toolCalls) {\n            if (!matchesToolFilter(toolCall)) {\n              // Tool call doesn't match our filter, skip it\n              continue;\n            }\n\n            if (wouldExceedLimit(tempThreadCount, tempRunCount)) {\n              blocked.push(toolCall);\n            } else {\n              allowed.push(toolCall);\n              tempThreadCount += 1;\n              tempRunCount += 1;\n            }\n          }\n\n          return {\n            allowed,\n            blocked,\n            finalThreadCount: tempThreadCount,\n            finalRunCount: tempRunCount + blocked.length,\n          };\n        };\n\n        /**\n         * Get the count key for this middleware instance\n         */\n        const countKey = options.toolName ?? DEFAULT_TOOL_COUNT_KEY;\n\n        /**\n         * Get current counts\n         */\n        const threadCounts = { ...(state.threadToolCallCount ?? {}) };\n        const runCounts = { ...(state.runToolCallCount ?? {}) };\n        const currentThreadCount = threadCounts[countKey] ?? 0;\n        const currentRunCount = runCounts[countKey] ?? 0;\n\n        /**\n         * Separate tool calls into allowed and blocked\n         */\n        const { allowed, blocked, finalThreadCount, finalRunCount } =\n          separateToolCalls(\n            lastAIMessage.tool_calls,\n            currentThreadCount,\n            currentRunCount\n          );\n\n        /**\n         * Update counts:\n         * - Thread count includes only allowed calls (blocked calls don't count towards thread-level tracking)\n         * - Run count includes blocked calls since they were attempted in this run\n         */\n        threadCounts[countKey] = finalThreadCount;\n        runCounts[countKey] = finalRunCount;\n\n        /**\n         * If no tool calls are blocked, just update counts\n         */\n        if (blocked.length === 0) {\n          if (allowed.length > 0) {\n            return {\n              threadToolCallCount: threadCounts,\n              runToolCallCount: runCounts,\n            };\n          }\n          return undefined;\n        }\n\n        /**\n         * Handle different exit behaviors\n         */\n        if (exitBehavior === \"error\") {\n          // Use hypothetical thread count to show which limit was exceeded\n          const hypotheticalThreadCount = finalThreadCount + blocked.length;\n          throw new ToolCallLimitExceededError(\n            hypotheticalThreadCount,\n            finalRunCount,\n            options.threadLimit,\n            options.runLimit,\n            options.toolName\n          );\n        }\n\n        /**\n         * Build tool message content (sent to model - no thread/run details)\n         */\n        const toolMsgContent = buildToolMessageContent(options.toolName);\n\n        /**\n         * Inject artificial error ToolMessages for blocked tool calls\n         */\n        const artificialMessages: Array<ToolMessage | AIMessage> = blocked.map(\n          (toolCall) =>\n            new ToolMessage({\n              content: toolMsgContent,\n              tool_call_id: toolCall.id!,\n              name: toolCall.name,\n              status: \"error\",\n            })\n        );\n\n        if (exitBehavior === \"end\") {\n          /**\n           * Check if there are tool calls to other tools that would continue executing\n           * For tool-specific limiters: check for calls to other tools\n           * For global limiters: check if there are multiple different tool types\n           */\n          let otherTools: ToolCall[] = [];\n          if (options.toolName !== undefined) {\n            /**\n             * Tool-specific limiter: check for calls to other tools\n             */\n            otherTools = lastAIMessage.tool_calls.filter(\n              (tc) => tc.name !== options.toolName\n            );\n          } else {\n            /**\n             * Global limiter: check if there are multiple different tool types\n             * If there are allowed calls, those would execute\n             * But even if all are blocked, we can't handle multiple tool types with \"end\"\n             */\n            const uniqueToolNames = new Set(\n              lastAIMessage.tool_calls.map((tc) => tc.name).filter(Boolean)\n            );\n            if (uniqueToolNames.size > 1) {\n              /**\n               * Multiple different tool types - use allowed calls to show which ones\n               */\n              otherTools =\n                allowed.length > 0 ? allowed : lastAIMessage.tool_calls;\n            }\n          }\n\n          if (otherTools.length > 0) {\n            const toolNames = Array.from(\n              new Set(otherTools.map((tc) => tc.name).filter(Boolean))\n            ).join(\", \");\n            throw new Error(\n              `Cannot end execution with other tool calls pending. Found calls to: ${toolNames}. Use 'continue' or 'error' behavior instead.`\n            );\n          }\n\n          /**\n           * Build final AI message content (displayed to user - includes thread/run details)\n           * Use hypothetical thread count (what it would have been if call wasn't blocked)\n           * to show which limit was actually exceeded\n           */\n          const hypotheticalThreadCount = finalThreadCount + blocked.length;\n          const finalMsgContent = buildFinalAIMessageContent(\n            hypotheticalThreadCount,\n            finalRunCount,\n            options.threadLimit,\n            options.runLimit,\n            options.toolName\n          );\n          artificialMessages.push(new AIMessage(finalMsgContent));\n\n          return {\n            threadToolCallCount: threadCounts,\n            runToolCallCount: runCounts,\n            jumpTo: \"end\" as const,\n            messages: artificialMessages,\n          };\n        }\n\n        /**\n         * For exit_behavior=\"continue\", return error messages to block exceeded tools\n         */\n        return {\n          threadToolCallCount: threadCounts,\n          runToolCallCount: runCounts,\n          messages: artificialMessages,\n        };\n      },\n    },\n    /**\n     * reset the run tool call count after the agent execution completes\n     */\n    afterAgent: () => ({\n      runToolCallCount: {},\n    }),\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;AAiBA,SAAS,wBAAwB,UAAsC;CAErE,IAAI,UACF,OAAO,0CAA0C,SAAS;CAE5D,OAAO;AACT;AAEA,MAAM,uBAAuB;CAAC;CAAY;CAAS;AAAK;AACxD,MAAM,wBAAwB;;;;;;;;;;;;;;AAe9B,SAAS,2BACP,aACA,UACA,aACA,UACA,UACQ;CACR,MAAM,WAAW,WAAW,IAAI,SAAS,UAAU;CACnD,MAAM,iBAA2B,CAAC;CAElC,IAAI,gBAAgB,KAAA,KAAa,cAAc,aAC7C,eAAe,KACb,0BAA0B,YAAY,GAAG,YAAY,QACvD;CAEF,IAAI,aAAa,KAAA,KAAa,WAAW,UACvC,eAAe,KAAK,uBAAuB,SAAS,GAAG,SAAS,QAAQ;CAI1E,OAAO,GAAG,SAAS,uBADA,eAAe,KAAK,OACY,EAAE;AACvD;;;;AAKA,MAAM,qBAAqBA,OAAAA,EACxB,KAAK,oBAAoB,CAAC,CAC1B,QAAQ,qBAAqB;;;;;;;AAQhC,IAAa,6BAAb,cAAgD,MAAM;;;;CAIpD;;;;CAIA;;;;CAIA;;;;CAIA;;;;CAIA;CAEA,YACE,aACA,UACA,aACA,UACA,WAA+B,KAAA,GAC/B;EACA,MAAM,UAAU,2BACd,aACA,UACA,aACA,UACA,QACF;EACA,MAAM,OAAO;EAEb,KAAK,OAAO;EACZ,KAAK,cAAc;EACnB,KAAK,WAAW;EAChB,KAAK,cAAc;EACnB,KAAK,WAAW;EAChB,KAAK,WAAW;CAClB;AACF;AAK0CA,OAAAA,EAAE,OAAO;;;;CAIjD,UAAUA,OAAAA,EAAE,OAAO,CAAC,CAAC,SAAS;;;;;CAK9B,aAAaA,OAAAA,EAAE,OAAO,CAAC,CAAC,SAAS;;;;;CAKjC,UAAUA,OAAAA,EAAE,OAAO,CAAC,CAAC,SAAS;;;;;;;;;;;CAW9B,cAAc;AAChB,CAAC;;;;AASD,MAAM,cAAcA,OAAAA,EAAE,OAAO;CAC3B,qBAAqBA,OAAAA,EAAE,OAAOA,OAAAA,EAAE,OAAO,GAAGA,OAAAA,EAAE,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;CAChE,kBAAkBA,OAAAA,EAAE,OAAOA,OAAAA,EAAE,OAAO,GAAGA,OAAAA,EAAE,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkF/B,SAAgB,wBAAwB,SAA8B;;;;CAIpE,IAAI,QAAQ,gBAAgB,KAAA,KAAa,QAAQ,aAAa,KAAA,GAC5D,MAAM,IAAI,MACR,gEACF;;;;CAMF,MAAM,eAAe,QAAQ,gBAAgB;CAC7C,MAAM,cAAc,mBAAmB,UAAU,YAAY;CAC7D,IAAI,CAAC,YAAY,SACf,MAAM,IAAI,MAAMC,OAAAA,EAAG,cAAc,YAAY,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC;;;;CAM9D,IACE,QAAQ,gBAAgB,KAAA,KACxB,QAAQ,aAAa,KAAA,KACrB,QAAQ,WAAW,QAAQ,aAE3B,MAAM,IAAI,MACR,aAAa,QAAQ,SAAS,+BAA+B,QAAQ,YAAY,mEAEnF;CAUF,OAAOC,mBAAAA,iBAAiB;EACtB,MALqB,QAAQ,WAC3B,2BAA2B,QAAQ,SAAS,KAC5C;EAIF;EACA,YAAY;GACV,WAAW,CAAC,KAAK;GACjB,OAAO,UAAU;;;;IAIf,MAAM,gBAAgB,CAAC,GAAG,MAAM,QAAQ,CAAC,CACtC,QAAQ,CAAC,CACT,KAAKC,yBAAAA,UAAU,UAAU;IAE5B,IAAI,CAAC,iBAAiB,CAAC,cAAc,YACnC;;;;IAMF,MAAM,oBACJ,aACA,aACY;KACZ,OACG,QAAQ,gBAAgB,KAAA,KACvB,cAAc,IAAI,QAAQ,eAC3B,QAAQ,aAAa,KAAA,KAAa,WAAW,IAAI,QAAQ;IAE9D;;;;IAKA,MAAM,qBAAqB,aAAyC;KAClE,OACE,QAAQ,aAAa,KAAA,KAAa,SAAS,SAAS,QAAQ;IAEhE;;;;IAKA,MAAM,qBACJ,WACA,aACA,aAMG;KACH,MAAM,UAAsB,CAAC;KAC7B,MAAM,UAAsB,CAAC;KAC7B,IAAI,kBAAkB;KACtB,IAAI,eAAe;KAEnB,KAAK,MAAM,YAAY,WAAW;MAChC,IAAI,CAAC,kBAAkB,QAAQ,GAE7B;MAGF,IAAI,iBAAiB,iBAAiB,YAAY,GAChD,QAAQ,KAAK,QAAQ;WAChB;OACL,QAAQ,KAAK,QAAQ;OACrB,mBAAmB;OACnB,gBAAgB;MAClB;KACF;KAEA,OAAO;MACL;MACA;MACA,kBAAkB;MAClB,eAAe,eAAe,QAAQ;KACxC;IACF;;;;IAKA,MAAM,WAAW,QAAQ,YAAY;;;;IAKrC,MAAM,eAAe,EAAE,GAAI,MAAM,uBAAuB,CAAC,EAAG;IAC5D,MAAM,YAAY,EAAE,GAAI,MAAM,oBAAoB,CAAC,EAAG;IACtD,MAAM,qBAAqB,aAAa,aAAa;IACrD,MAAM,kBAAkB,UAAU,aAAa;;;;IAK/C,MAAM,EAAE,SAAS,SAAS,kBAAkB,kBAC1C,kBACE,cAAc,YACd,oBACA,eACF;;;;;;IAOF,aAAa,YAAY;IACzB,UAAU,YAAY;;;;IAKtB,IAAI,QAAQ,WAAW,GAAG;KACxB,IAAI,QAAQ,SAAS,GACnB,OAAO;MACL,qBAAqB;MACrB,kBAAkB;KACpB;KAEF;IACF;;;;IAKA,IAAI,iBAAiB,SAGnB,MAAM,IAAI,2BADsB,mBAAmB,QAAQ,QAGzD,eACA,QAAQ,aACR,QAAQ,UACR,QAAQ,QACV;;;;IAMF,MAAM,iBAAiB,wBAAwB,QAAQ,QAAQ;;;;IAK/D,MAAM,qBAAqD,QAAQ,KAChE,aACC,IAAIC,yBAAAA,YAAY;KACd,SAAS;KACT,cAAc,SAAS;KACvB,MAAM,SAAS;KACf,QAAQ;IACV,CAAC,CACL;IAEA,IAAI,iBAAiB,OAAO;;;;;;KAM1B,IAAI,aAAyB,CAAC;KAC9B,IAAI,QAAQ,aAAa,KAAA;;;;KAIvB,aAAa,cAAc,WAAW,QACnC,OAAO,GAAG,SAAS,QAAQ,QAC9B;UAUA,IAAI,IAHwB,IAC1B,cAAc,WAAW,KAAK,OAAO,GAAG,IAAI,CAAC,CAAC,OAAO,OAAO,CAE5C,CAAC,CAAC,OAAO;;;;KAIzB,aACE,QAAQ,SAAS,IAAI,UAAU,cAAc;KAInD,IAAI,WAAW,SAAS,GAAG;MACzB,MAAM,YAAY,MAAM,KACtB,IAAI,IAAI,WAAW,KAAK,OAAO,GAAG,IAAI,CAAC,CAAC,OAAO,OAAO,CAAC,CACzD,CAAC,CAAC,KAAK,IAAI;MACX,MAAM,IAAI,MACR,uEAAuE,UAAU,8CACnF;KACF;KAQA,MAAM,kBAAkB,2BADQ,mBAAmB,QAAQ,QAGzD,eACA,QAAQ,aACR,QAAQ,UACR,QAAQ,QACV;KACA,mBAAmB,KAAK,IAAID,yBAAAA,UAAU,eAAe,CAAC;KAEtD,OAAO;MACL,qBAAqB;MACrB,kBAAkB;MAClB,QAAQ;MACR,UAAU;KACZ;IACF;;;;IAKA,OAAO;KACL,qBAAqB;KACrB,kBAAkB;KAClB,UAAU;IACZ;GACF;EACF;;;;EAIA,mBAAmB,EACjB,kBAAkB,CAAC,EACrB;CACF,CAAC;AACH"}