Áj5n ddlmZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZmZddlZddlmZmZmZmZdd lm Z m!Z!dd l"m#Z#m$Z$dd l%m&Z&d d l'm(Z(m)Z)m*Z*d dl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2ddl3m4Z4m5Z5d dl6m7Z7m8Z8m9Z9ddl:m;Z;mZ>m?Z?m@Z@ddlAmBZBddlCmDZDerddlEmFZFmGZGe de e-e e/e e1e2fZHe dZIejeKZLGddedZMGdde eHe>fZNGd d!eNe.e>fe eIe>feZOGd"d#eOe?e>e>fZPGd$d%eOe;e>e>fZQGd&d'eNe0e>fe eIe>feZRGd(d)eRe?e>e>fZSGd*d+eRee>fZTy),) annotationsN)ABCabstractmethod) TYPE_CHECKINGAnyListUnionGenericTypeVarCallableIterableIterator Coroutine AsyncIterator)contextmanagerasynccontextmanager) TypedDictoverride)BodyQueryHeadersNotGiven)consume_sync_iteratorconsume_async_iterator) BetaMessageBetaMessageParam) merge_headers) tool_registrytool_error_contentavailable_tool_names) ToolErrorBetaFunctionToolBetaRunnableToolBetaAsyncFunctionToolBetaAsyncRunnableToolBetaBuiltinFunctionToolBetaAsyncBuiltinFunctionTool) helper_headerstainless_helper_header)DEFAULT_THRESHOLDDEFAULT_SUMMARY_PROMPTCompactionControl)BetaMessageStreamBetaAsyncMessageStream)ResponseFormatTParsedBetaMessageParsedBetaContentBlock)ParseMessageCreateParamsBase)BetaToolResultBlockParam) AnthropicAsyncAnthropicAnyFunctionToolT)bound RunnerItemTc6eZdZUded<ded<ded<ded<y ) RequestOptionszHeaders | None extra_headersz Query | None extra_queryz Body | None extra_bodyz'float | httpx.Timeout | None | NotGiventimeoutN)__name__ __module__ __qualname____annotations__G/tmp/pip-target-k_scjv0h/lib/python/anthropic/lib/tools/_beta_runner.pyr=r=<s!! 44rGr=F)totalcZeZdZddd ddZ d dZd dZd dZd dZy) BaseToolRunnerNmax_iterationscompaction_controlcxt||_i|d|dDcgc]}|c}i|_t|jj |j d}|r't ||j dxsi}i|d|i}||_d|_d|_ ||_ d|_ ||_ ycc}w)Nmessages)toolsrPr>Fr) r _tools_by_name_paramsr,valuesgetr_options_messages_modified_cached_tool_call_response_max_iterations_iteration_count_compaction_control) selfparamsoptionsrQrMrNmessager+merged_headerss rH__init__zBaseToolRunner.__init__Ds,E2G G z0BCWCG  0%%,,.ZZ +  *='++o:V:\Z\]NBB/>BG "'CG'- !#5 Ds B7cLt|r||j}||_y)a  Update the parameters for the next API call. This invalidates any cached tool responses. Args: params (ParsedMessageCreateParamsBase[ResponseFormatT] | Callable): Either new parameters or a function to mutate existing parameters N)callablerS)r\r]s rHset_messages_paramsz"BaseToolRunner.set_messages_params`s! F DLL)F rGc|Dcgc]-}t|tr|j|jdn|/c}d|_|j fdd|_ycc}w)zAdd one or more messages to the conversation history. This invalidates the cached tool response, i.e. if tools were already called, then they will be called again on the next loop iteration. rolecontentTc"i|dg|diS)NrPrF)r]message_paramss rHz0BaseToolRunner.append_messages..zs&0n60n:GmPZI[Gm^lGm0nrGN) isinstancerrgrhrWrdrX)r\rPr_rjs @rHappend_messageszBaseToolRunner.append_messagesosb$2 CMWVaBbW\\goo >ho o2 #'   !no*.' 2 s2AcP|j|j|jk\ryy)NTF)rYrZr\s rH _should_stopzBaseToolRunner._should_stop}s'    +0E0EI]I]0]rGcHt|jd|jS)amThe tool names currently available, after applying any mid-conversation ``tool_removal`` / ``tool_addition`` blocks. Removal is only a hint to the model, which can still emit a ``tool_use`` for a withdrawn tool; a name absent from this set routes that call down the same unknown-tool path as a tool that was never declared. rP)r"rSrRros rH_available_tool_namesz$BaseToolRunner._available_tool_namess $DLL$Q>QRRrG) r]-ParseMessageCreateParamsBase[ResponseFormatT]r^r=rQzIterable[AnyFunctionToolT]rM int | NonerNCompactionControl | NonereturnNone)r]zParseMessageCreateParamsBase[ResponseFormatT] | Callable[[ParseMessageCreateParamsBase[ResponseFormatT]], ParseMessageCreateParamsBase[ResponseFormatT]]rvrw)rPz5BetaMessageParam | ParsedBetaMessage[ResponseFormatT]rvrwrvbool)rvzset[str])rBrCrDrardrmrprrrFrGrHrKrKCsu&*7;6>6 6 * 6 # 656 68 s    / SrGrKceZdZddd dfdZddZddZeeddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZxZS)BaseSyncToolRunnerNrLct||||||||_|-|jdrt j dt d|j|_d|_ yN)r]r^rQrMrNenableda!The 'compaction_control' parameter is deprecated and will be removed in a future version. Use server-side compaction instead by passing `edits=[{'type': 'compact_20260112'}]` in your the params passed to `tool_runner()`. See https://platform.claude.com/docs/en/build-with-claude/compactionr stacklevel superra_clientrUwarningswarnDeprecationWarning__run__ _iterator _last_messager\r]r^rQclientrMrN __class__s rHrazBaseSyncToolRunner.__init__sy )1    ).@.D.DY.O MM}#    rGc6|jjSN)r__next__ros rHrzBaseSyncToolRunner.__next__s~~&&((rGc#6K|jD]}|ywrrr\items rH__iter__zBaseSyncToolRunner.__iter__sNN DJ sc#KtwrNotImplementedErrorros rH_handle_requestz"BaseSyncToolRunner._handle_request"## c |j|jdsy|j}d}|e|jj|jjxsdz|jj xsdz}||jj z}|jjdt}||krytjd|d|d|jjd|jd}t|jd }|d d d k(rT|d d Dcgc](}t|tr|jddk7r|*}}|r ||d d <n|jg|t!d|jjdt"}|j$j&j(j+|||jdt-d} tjd| jj t| j.d j0dk7r t3d|j5 fdycc}w) Check token usage and compact messages if threshold exceeded. Returns True if compaction was performed, False otherwise. r~Frcontext_token_threshold Token usage  has exceeded the threshold of . Performing compaction.modelrPrg assistantrhtypetool_useusersummary_promptrf max_tokens compactionrrPrr>&Compaction complete. New token usage: text1Compaction response content is not of type 'text'c8i|dddjdgdgiSNrPrr)rrrfrr] first_contents rHrkz7BaseSyncToolRunner._check_and_compact..?   &)/(5(:(:$  rGTr[_get_last_messageusage input_tokenscache_creation_input_tokenscache_read_input_tokens output_tokensrUr-loginforSlistrldictpoprr.rbetarPcreater+rhr ValueErrorrd r\r_ tokens_usedtotal_input_tokens thresholdrrPblocknon_tool_blocksresponsers @rH_check_and_compactz%BaseSyncToolRunner._check_and_compacts^  # # +43K3KI3V((*   **==<<AC==88=A?  -w}}/J/JJK,,001JL]^  " < },KI;Vnop((,,Wdll76KL Z01 B< ; . &b\)4eT*uyy/@J/NO *9 Y'   00445EG]^  <<$$--44||L1' 5 5  9(..:V:V9WXYX--.q1    'PQ Q     cs/-I$c#K|js#|j5}||j}|J|j}|/|j#|jj |j d<ddd|xjdz c_jdk(rtjdy|jsF|j}|tjdy|js|j||d|_ d|_|js"yy#1swYxYwwN containerrrefusalz9Turn ended with a refusal, exiting from tool runner loop.z;Tool call was not requested, exiting from tool runner loop.Frprr_get_last_assistant_messageridrSrZ stop_reasonrdebugrgenerate_tool_call_responserWrmrXr\rr_last_assistant_messagers rHrzBaseSyncToolRunner.__run__s3##%%%' T4 002****.)I)I)K&)5:P:Z:Z:f0F0P0P0S0SDLL- T  ! !Q & ! ""i/ UV**,;;=#II[\..(((;&+D #.2D +?##%% T Ts$"EAD8>B6E6E8E=EcDt||j}|J|S) Consumes the tool runner stream and returns the last message if it has not been consumed yet. If it has, it simply returns the last message. )rrr\ last_messages rH until_donezBaseSyncToolRunner.until_done2s, d#--/ '''rGc|j!tjd|jS|j}||_|S)Generate a MessageParam by calling tool functions with any tool use blocks from the last message. Note the tool call response is cached, repeated calls to this method will return the same response. None can be returned if no tool call was applicable. $Returning cached tool call response.rXrr_generate_tool_call_responser\rs rHrz.BaseSyncToolRunner.generate_tool_call_response<sD  * * 6 II< =22 2446*2'rGc |j}|sy|Dcgc]}|jdk(s|}}|syg}|j}|D]}|j|vr%|jj |jnd}|}t jd|jdt|jjdtd|jd|jd|jd d d  |j|j}|jd|j|d d|dScc}w#t$r4} |jd|jt!| d d Yd} ~ 6d} ~ wt"$rX} t$j'd |j| |jd|jt!| d d Yd} ~ d} ~ wwxYw)NrTool '-' not found in tool runner. Available tools: z. If using a raw tool definition, handle the tool call manually and use `append_messages()` to add the result. Otherwise, pass the tool using `beta_tool(func)` or a `@beta_tool` decorated function.rr tool_result Error: Tool ' ' not foundTr tool_use_idrhis_errorrrrh#Error occurred while calling tool: exc_inforrf#_get_last_assistant_message_contentrrrnamerRrUrrrkeys UserWarningappendrcallinputr#r! Exceptionr exception r\rhrtool_use_blocksresults availablertoolresultexcs rHrz/BaseSyncToolRunner._generate_tool_call_responseJs::<.5RUz9Q5RR24..0 '* H=E]]i=W4&&**8==9]aD| X]]O,((,T-@-@-E-E-G(H'IJmn   -'/{{%28==/#M$(  8>>2 hkk^def/* X733gS>  -'/{{#5c#:$(   CDII;OZ] ^ -'/{{#5c#:$(  s/EE:E G$)F G$ A GG$cdt|jr|jS|jSrrcrros rHrz$BaseSyncToolRunner._get_last_messages+ D&& '%%' '!!!rGcb|j}||jdk7s |jsy|SNrrrgrhrs rHrz.BaseSyncToolRunner._get_last_assistant_messages4--/  <#4#4 #C  *   #5 B)Rh 3D 84t" .rGr{c(eZdZeeddZy)BetaToolRunnerc#K|jjjjdi|j|j }||_|ywNrFrrrPparserSrVrr\r_s rHrzBetaToolRunner._handle_requestsG3$,,##,,22ST\\ST]]S$ sAAN)rvz,Iterator[ParsedBetaMessage[ResponseFormatT]]rBrCrDrrrrFrGrHrrs rGrc(eZdZeeddZy)BetaStreamingToolRunnerc#K|jjjjdi|j|j 5}|j |_|dddy#1swYyxYwwrrrrPstreamrSrVget_final_messagerr\rs rHrz'BetaStreamingToolRunner._handle_requestsb/T\\   ' ' . . O O O SY!'!9!9D L   sAA4 A( A4(A1-A4N)rvz,Iterator[BetaMessageStream[ResponseFormatT]]rrFrGrHrrs rGrceZdZddd dfdZddZddZeeddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZxZS)BaseAsyncToolRunnerNrLct||||||||_|-|jdrt j dt d|j|_d|_ yr}rrs rHrazBaseAsyncToolRunner.__init__sy )1    ).@.D.DY.O MM}#    rGcRK|jjd{S7wr)r __anext__ros rHr"zBaseAsyncToolRunner.__anext__s^^--////s '%'cHK|j23d{}|7 6ywrrrs rH __aiter__zBaseAsyncToolRunner.__aiter__s&..  $J .s "  " "cKtwrrros rHrz#BaseAsyncToolRunner._handle_requestrrc K|j|jdsy|jd{}d}|e|jj|jjxsdz|jj xsdz}||jj z}|jjdt}||krytjd|d|d|jjd |jd }t|jd }|d d d k(rT|d dDcgc](}t|tr|jddk7r|*}}|r ||d d<n|jg|t!d|jjdt"}|j$j&j(j+|||jdt-dd{} tjd| jj t| j.d j0dk7r t3d|j5 fdy7Ccc}w7w)rNr~FrrrrrrrPrrgrrhrrrrrfrrrrrrc8i|dddjdgdgiSrrrs rHrkz8BaseAsyncToolRunner._check_and_compact.. rrGTrrs @rHrz&BaseAsyncToolRunner._check_and_compactst  # # +43K3KI3V..00   **==<<AC==88=A?  -w}}/J/JJK,,001JL]^  " < },KI;Vnop((,,Wdll76KL Z01 B< ; . &b\)4eT*uyy/@J/NO *9 Y'   00445EG]^  **33::||L1' 5 ;   9(..:V:V9WXYX--.q1    'PQ Q     U12& s00JI6DJ9-I9&BJ>I>?A8J9JcK|jsT|j4d{}||jd{}|J|jd{}|/|j#|jj |j d<dddd{|xjdz c_jdk(rtjdy|jd{sN|jd{}|tjdy|js|j||d|_ d|_|jsSyy7A7'77#1d{7swYxYw77~wrrrs rHrzBaseAsyncToolRunner.__run__1sw##%++- T T $ 6 6 88***04/O/O/Q)Q&)5:P:Z:Z:f0F0P0P0S0SDLL- T T  ! !Q & ! ""i/ UV00222!%!A!A!CC#II[\..(((;&+D #.2D +?##%% T8*R  T T T T(3Cs&F E)F E4E,E4!E/"5E4 F "E2#AF 4F 5F  F  AF 'F ,E4/E42F 4F:E= ;FF  F ctKt|d{|jd{}|J|S7"7 w)rN)rrrs rHrzBaseAsyncToolRunner.until_doneSsB %T***!3355 ''' +5s8486 88cK|j!tjd|jS|jd{}||_|S7 w)rNrrrs rHrz/BaseAsyncToolRunner.generate_tool_call_response]sP  * * 6 II< =22 2::<<*2'=sAAAAcKt|jr|jd{S|jS7wrrros rHrz%BaseAsyncToolRunner._get_last_messagels7 D&& '++-- -!!!.s )><>c~K|jd{}||jdk7s |jsy|S7$wrrrs rHrz/BaseAsyncToolRunner._get_last_assistant_messageqs@!3355  <#4#4 #CE G>EECG> E*E+#EG>G>E G;()FG> G;#A G60G>6G;;G>)r]rsr^r=rQzIterable[BetaAsyncRunnableTool]rr8rMrtrNrurvrwr)rvzAsyncIterator[RunnerItemT]rxr r r r )rBrCrDrar"r$rrrrrrrrrrrr rs@rHrrs&*7;!>! ! / !  !#!5! !F0Rh 3D " .84rGrc(eZdZeeddZy)BetaAsyncToolRunnercK|jjjjdi|j|j d{}||_|y7wrrrs rHrz#BetaAsyncToolRunner._handle_requestsP9 ))2288Y4<<Y4==YY$ ZsA A  A A N)rvz1AsyncIterator[ParsedBetaMessage[ResponseFormatT]]rBrCrDrrrrFrGrHr0r0s rGr0c(eZdZeeddZy)BetaAsyncStreamingToolRunnerc(K|jjjjdi|j|j 4d{}|j |_|dddd{y7+7#1d{7swYyxYwwrrrs rHrz,BetaAsyncStreamingToolRunner._handle_requests}54<<$$--44Ut||Ut}}U  Y_!'!9!9D L       sHA B A9BA=( B3A;4B;B=BB B BN)rvz6AsyncIterator[BetaAsyncMessageStream[ResponseFormatT]]r2rFrGrHr4r4s rGr4)U __future__rloggingrabcrrtypingrrrr r r r r rrr contextlibrrtyping_extensionsrrhttpx_typesrrrr_utilsrr types.betarr _base_clientr_tool_dispatchr r!r"_beta_functionsr#r$r%r&r'r(r)_stainless_helpersr+r,_beta_compaction_controlr-r.r/streaming._beta_messagesr0r1types.beta.parsed_beta_messager2r3r4 types.beta.message_create_paramsr5'types.beta.beta_tool_result_block_paramr6rr7r8r9r; getLoggerrBrr=rKr{rrrr0r4rFrGrHrJs"#    ;1 44C7)SSHbbPhhLO4 4S9;RTpp  m$ g!5Ye5GSW->?GSTH.(8/(IJGT_apTpLqsvH.V'(9/(JO([\01B?1SUd1deM4(/9:GKQ`D`