iEBSrSSKJr SSKJrJrJrJrJr SSK J r J r J r SSK Jr SSKJr SSKJr SSKJrJrJrJrJr \(aSS KJr \S r"S S \\\\5rSS jrSSjr"SS\5r "SS\\\\4\\\45r!g)z&Tool call limit middleware for agents.) annotations) TYPE_CHECKING AnnotatedAnyGenericLiteral) AIMessageToolCall ToolMessage)UntrackedValue)ContextT) NotRequired)AgentMiddleware AgentStatePrivateStateAttr ResponseT hook_config)Runtimecontinueerrorendc.\rSrSr%SrS\S'S\S'Srg) ToolCallLimitState"aCState schema for ToolCallLimitMiddleware. Extends AgentState with tool call tracking fields. The count fields are dictionaries mapping tool names to execution counts. This allows multiple middleware instances to track different tools independently. The special key "__all__" is used for tracking all tool calls globally. z8NotRequired[Annotated[dict[str, int], PrivateStateAttr]]thread_tool_call_countzHNotRequired[Annotated[dict[str, int], UntrackedValue, PrivateStateAttr]]run_tool_call_countN)__name__ __module__ __qualname____firstlineno____doc____annotations____static_attributes__re/home/james-whalen/.local/lib/python3.13/site-packages/langchain/agents/middleware/tool_call_limit.pyrr"sUTaar&rcU(aSUS3$g)agBuild the error message content for ToolMessage when limit is exceeded. This message is sent to the model, so it should not reference thread/run concepts that the model has no notion of. Args: tool_name: Tool name being limited (if specific tool), or None for all tools. Returns: A concise message instructing the model not to call the tool again. z'Tool call limit exceeded. Do not call 'z' again.zXlX lX0lX@lXPl[ XX4U5n[ TU]U5 g)aMInitialize the exception with call count information. Args: thread_count: Current thread tool call count. run_count: Current run tool call count. thread_limit: Thread tool call limit (if set). run_limit: Run tool call limit (if set). tool_name: Tool name being limited (if specific tool), or None for all tools. N)r3r4r5r6r*r:super__init__)selfr3r4r5r6r*msg __class__s r'r@#ToolCallLimitExceededError.__init__ks@")"(""- \i  r&)r4r6r3r5r*)N) r3intr4rEr5 int | Noner6rFr* str | NonereturnNone)rr r!r"r#r@r% __classcell__rCs@r'r<r<dsQ!% !      r&r<c^\rSrSrSr\rSSSSS.SU4Sjjjr\SSj5r SSjr SS jr SS jr \ "S /S 9SS j5rSrU=r$)ToolCallLimitMiddlewarea\Track tool call counts and enforces limits during agent execution. This middleware monitors the number of tool calls made and can terminate or restrict execution when limits are exceeded. It supports both thread-level (persistent across runs) and run-level (per invocation) call counting. Configuration: - `exit_behavior`: How to handle when limits are exceeded - `"continue"`: Block exceeded tools, let execution continue (default) - `"error"`: Raise an exception - `"end"`: Stop immediately with a ToolMessage + AI message for the single tool call that exceeded the limit (raises `NotImplementedError` if there are other pending tool calls (due to parallel tool calling). Examples: Continue execution with blocked tools (default): ```python from langchain.agents.middleware.tool_call_limit import ToolCallLimitMiddleware from langchain.agents import create_agent # Block exceeded tools but let other tools and model continue limiter = ToolCallLimitMiddleware( thread_limit=20, run_limit=10, exit_behavior="continue", # default ) agent = create_agent("openai:gpt-4o", middleware=[limiter]) ``` Stop immediately when limit exceeded: ```python # End execution immediately with an AI message limiter = ToolCallLimitMiddleware(run_limit=5, exit_behavior="end") agent = create_agent("openai:gpt-4o", middleware=[limiter]) ``` Raise exception on limit: ```python # Strict limit with exception handling limiter = ToolCallLimitMiddleware(tool_name="search", thread_limit=5, exit_behavior="error") agent = create_agent("openai:gpt-4o", middleware=[limiter]) try: result = await agent.invoke({"messages": [HumanMessage("Task")]}) except ToolCallLimitExceededError as e: print(f"Search limit exceeded: {e}") ``` Nr)r*r5r6 exit_behaviorc>[TU]5 UcUc Sn[U5eSnXF;aSU<SU3n[U5eUbUbX2:aSUSUS3n[U5eXlX lX0lX@lg) a*Initialize the tool call limit middleware. Args: tool_name: Name of the specific tool to limit. If `None`, limits apply to all tools. Defaults to `None`. thread_limit: Maximum number of tool calls allowed per thread. `None` means no limit. Defaults to `None`. run_limit: Maximum number of tool calls allowed per run. `None` means no limit. Defaults to `None`. exit_behavior: How to handle when limits are exceeded. - `"continue"`: Block exceeded tools with error messages, let other tools continue. Model decides when to end. (default) - `"error"`: Raise a `ToolCallLimitExceededError` exception - `"end"`: Stop execution immediately with a ToolMessage + AI message for the single tool call that exceeded the limit. Raises `NotImplementedError` if there are multiple parallel tool calls to other tools or multiple pending tool calls. Raises: ValueError: If both limits are `None`, if exit_behavior is invalid, or if run_limit exceeds thread_limit. Nz@At least one limit must be specified (thread_limit or run_limit)rzInvalid exit_behavior: z. Must be one of z run_limit (z) cannot exceed thread_limit (zB). The run limit should be less than or equal to the thread limit.)r?r@ ValueErrorr*r5r6rO)rAr*r5r6rOrBvalid_behaviorsrCs r'r@ ToolCallLimitMiddleware.__init__s<   I$5TCS/ !6  /+M+<>[$..!13 3r&cURSL=(a US-UR:=(d' URSL=(a US-UR:$)zCheck if incrementing the counts would exceed any configured limit. Args: thread_count: Current thread call count. run_count: Current run call count. Returns: True if either limit would be exceeded by one more call. N)r5r6)rAr3r4s r'_would_exceed_limit+ToolCallLimitMiddleware._would_exceed_limitsO!!-V,2BTEVEV2V NN$ & I9q=4>>+I r&cPURSL=(d USUR:H$)zCheck if a tool call matches this middleware's tool filter. Args: tool_call: The tool call to check. Returns: True if this middleware should track this tool call. NrXr))rA tool_calls r'_matches_tool_filter,ToolCallLimitMiddleware._matches_tool_filters&~~%L6):dnn)LLr&c/n/nUnUnUH_nURU5(dMURXg5(aURU5 MDURU5 US- nUS- nMa XEXg4$)a%Separate tool calls into allowed and blocked based on limits. Args: tool_calls: List of tool calls to evaluate. thread_count: Current thread call count. run_count: Current run call count. Returns: Tuple of (allowed_calls, blocked_calls, final_thread_count, final_run_count). r[)r`r\r1) rA tool_callsr3r4 allowed_calls blocked_callstemp_thread_counttemp_run_countr_s r'_separate_tool_calls,ToolCallLimitMiddleware._separate_tool_callss)+ (* ("#I,,Y77''(9JJ$$Y/$$Y/!Q&!!#$->NNr&r) can_jump_toc vURS/5nU(dgSn[U5Hn[U[5(dMUn O U(aUR(dgUR (a UR OSnURS05R 5nURS05R 5nURUS5n URUS5n URURX5uppXU'U[U 5-X'U (d U (aUUS.$gXvnXnURS:Xa9U[U 5-n[UUURURUR S 9e[UR 5nU Vs/sH!n[UUS URS 5SS 9PM# nnURS :XaURVs/sH)nUR cMUS UR :wdM'UPM+ nnU(a6SRUVs1sHnUS iM sn5nSUS3n[!U5eU[U 5-n[#UUURURUR 5nUR%[US95 UUS US.$UUUS.$s snfs snfs snf)aDIncrement tool call counts after a model call and check limits. Args: state: The current agent state. runtime: The langgraph runtime. Returns: State updates with incremented tool call counts. If limits are exceeded and exit_behavior is "end", also includes a jump to end with a ToolMessage and AI message for the single exceeded tool call. Raises: ToolCallLimitExceededError: If limits are exceeded and exit_behavior is "error". NotImplementedError: If limits are exceeded, exit_behavior is "end", and there are multiple tool calls. messagesN__all__rrr)rrr)r3r4r5r6r*idrX)content tool_call_idrXstatusrz, zDCannot end execution with other tool calls pending. Found calls to: z-. Use 'continue' or 'error' behavior instead.)ro)rrjump_torl)rrrl)getreversed isinstancer rcr*copyrhlenrOr<r5r6r+r r2NotImplementedErrorr:r1)rAstateruntimerllast_ai_messagemessage count_key thread_counts run_countscurrent_thread_countcurrent_run_countrdrenew_thread_count new_run_countfinal_thread_countfinal_run_counthypothetical_thread_counttool_msg_contentr_artificial_messagestc other_tools tool_namesrBfinal_msg_contents r' after_model#ToolCallLimitMiddleware.after_model<s099Z,)G'9--")* o&@&@'+nnDNN)  ":B?DDF YY4b9>>@ ,00A>&NN9a8IMHaHa  & &(<I E &6$4i -M0B B .;+5+5$/    ((:S=O(O %,6)!......  7t~~F+> + (&t_]]6*   + >     &*444B>>24V*2N4  !YY['I[r6 ['IJ ''1l2_a*#.. );S=O(O % ?)!! !  & &y9J'K L+8'1 /  '4#-+  _> (Js(J,J12J1J1%J6)rOr6r5r*) r*rGr5rFr6rFrO ExitBehaviorrHrI)rHstr)r3rEr4rErHbool)r_r rHr)rczlist[ToolCall]r3rEr4rErHz/tuple[list[ToolCall], list[ToolCall], int, int])ryzToolCallLimitState[ResponseT]rzzRuntime[ContextT]rHzdict[str, Any] | None)rr r!r"r#r state_schemar@propertyrXr\r`rhrrr%rJrKs@r'rMrMs3j&L !%#' $&0 3+3+! 3+  3+ $ 3+ 3+3+j     MO(O8;OHKO 8O>eW%I ,I #I   I &I r&rMN)r*rGrHr) r3rEr4rEr5rFr6rFr*rGrHr)"r# __future__rtypingrrrrrlangchain_core.messagesr r r "langgraph.channels.untracked_valuer langgraph.typingr typing_extensionsr!langchain.agents.middleware.typesrrrrrlanggraph.runtimerrrr+r: Exceptionr<rMrr&r'rs,"BBDD=%))12  bI. 0B bJ$==== =  =  =D!!H~ &y18;< Ix  ~ r&