OKi%(SrSSKJr SSKJrJrJrJr SSKJ r SSK J r SSK J r SSKJrJrJrJr SSKJr SS KJr SS KJrJrJrJrJrJr S r\\\/\4r "S S \5r!\ "SS9"SS\!55r""SS\\\\\45r#SS/r$g)aContext editing middleware. Mirrors Anthropic's context editing capabilities by clearing older tool results once the conversation grows beyond a configurable token threshold. The implementation is intentionally model-agnostic so it can be used with any LangChain chat model. ) annotations) AwaitableCallableIterableSequence)deepcopy) dataclass)Literal) AIMessage AnyMessage BaseMessage ToolMessagecount_tokens_approximately)Protocol)AgentMiddleware AgentStateContextT ModelRequest ModelResponse ResponseTz [cleared]c.\rSrSrSrSSjrSrg) ContextEdit,z/Protocol describing a context editing strategy.cg)z+Apply an edit to the message list in place.N)selfmessages count_tokenss m/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/langchain/agents/middleware/context_editing.pyapplyContextEdit.apply/s rNrzlist[AnyMessage]r TokenCounterreturnNone)__name__ __module__ __qualname____firstlineno____doc__r!__static_attributes__rr#r rr,s&9 " #    r#rT)slotsc\rSrSr%SrSrS\S'SrS\S'SrS\S 'S r S \S 'S r S\S'\ r S\S'SSjr \SSj5rS rg)ClearToolUsesEdit9zGConfiguration for clearing tool outputs when token limits are exceeded.iinttriggerrclear_at_leastkeepFboolclear_tool_inputsrz Sequence[str] exclude_toolsstr placeholderc ^ U"U5nX0R::ag[U5VVs/sHupE[U[5(dMXE4PM! nnnUR[ U5:a/nO!UR(aUSUR*nSn[ UR5nUGHJunm T RRS05RS5(aM9[S[USU55S5n U cM][U 4SjU R5S5n U cMT R=(d U SU;aMT RSUR0T RESSS S .0ES .S 9X'UR (a-UR#U T R$5XR'U 5'UR(S:dGM#U"U5n [+SX;- 5nXpR(:dGMK g gs snnf) z#Apply the clear-tool-uses strategy.Nrcontext_editingclearedc3T# UHn[U[5(dMUv M g7fN) isinstancer ).0ms r *ClearToolUsesEdit.apply..lsQ4q 1i8P4s( (c3j># UH(nURS5TR:XdM$Uv M* g7f)idN)get tool_call_id)rBcall tool_messages r rDrEss0 5xx~)B)BBD 5s#3 3nameTclear_tool_uses)r>strategy)artifactcontentresponse_metadataupdate)r3 enumeraterArr6lensetr9rQrHnextreversed tool_callsrL model_copyr;r8!_build_cleared_tool_input_messagerIindexr4max) rrrtokensidxmsg candidatescleared_tokensexcluded_tools ai_message tool_callnew_token_countrKs @r r!ClearToolUsesEdit.applyOsh' \\ ! (1': ':83jk>ZJSJ':   99J 'J YY#Ltyyj1JT//0!+ C--112CRHLLYWWQHXds^4QSWJ! * 5 5  I !!6Yv%6>I(33 $#//*&88*)'+(9,* 4 HM%%7;7]7] --8 34 ""Q&".x"8!$Q(@!A!%8%88i",h A s G7G7c/nSnURH;n[U5nURS5U:Xa0US'SnURU5 M= [[ US055n[URS055nU(a>[ URS/55nUR U5 [U5US'XvS'URUUS.S 9$) NFrGargsTrQr=cleared_tool_inputs)rYrQrR) rYdictrHappendgetattrrVaddsortedrZ) messagerIupdated_tool_calls cleared_anyre updated_callmetadata context_entry cleared_idss r r[3ClearToolUsesEdit._build_cleared_tool_input_messages    ++I ?L%5') V$"  % %l 3 ,)[TU]5 [U=(d [545UlX lg)a}Initialize an instance of context editing middleware. Args: edits: Sequence of edit strategies to apply. Defaults to a single `ClearToolUsesEdit` mirroring Anthropic defaults. token_count_method: Whether to use approximate token counting (faster, less accurate) or exact counting implemented by the chat model (potentially slower, more accurate). N)super__init__listr0r~r)rr~r __class__s r r!ContextEditingMiddleware.__init__s. %9$5$7#9: "4r#c`^^TR(dU"T5$URS:XaSSjnO)TR(a TR/O/mSUU4Sjjn[[ TR55nUR HnUR XCS9 M U"TRUS95$):Apply context edits before invoking the model via handler. Args: request: Model request to execute (includes state and runtime). handler: Async callback that executes the model request and returns `ModelResponse`. Returns: The result of invoking the handler with potentially edited messages. rc[U5$r@rrs r r>ContextEditingMiddleware.wrap_model_call..count_tokens 1(;;r#ch>TRRT[U5-TR5$r@modelget_num_tokens_from_messagesrtoolsrrequest system_msgs r rr,}}AAh/r#rrrzSequence[BaseMessage]r&r2rrsystem_messagerrr~r!overriderrhandlerredited_messageseditrs ` @r wrap_model_call(ContextEditingMiddleware.wrap_model_calls7# #  " "m 3 <6=5K5K'001QSJ   #4(8(8#9:JJD JJJ Bw'''ABBr#c^^# TR(dU"T5IShvN $URS:XaSSjnO)TR(a TR/O/mSUU4Sjjn[[ TR55nUR HnUR XCS9 M U"TRUS95IShvN $NN7f)rNrc[U5$r@rrs r r?ContextEditingMiddleware.awrap_model_call..count_tokensrr#ch>TRRT[U5-TR5$r@rrs r rrrr#rrrrrs ` @r awrap_model_call)ContextEditingMiddleware.awrap_model_calls )) )  " "m 3 <6=5K5K'001QSJ   #4(8(8#9:JJD JJJ BW---GHHH'*&Is"CCBC=C>CC)r~zIterable[ContextEdit] | Nonerrr&r')rModelRequest[ContextT]rzK 5,5< 5  55(#C'#CN#C . #CJ#I'#IY#I . #I#Ir#r|N)%r, __future__rcollections.abcrrrrcopyr dataclassesr typingr langchain_core.messagesr r r rlangchain_core.messages.utilsrtyping_extensionsr!langchain.agents.middleware.typesrrrrrrryr2r%rr0r|__all__rr#r rs#CC! E&' k   (   ~ ~ ~ BiIz)/DhPY/YZiIZ r#