IigddlmZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZmZddlZddlmZmZmZmZdd lm Z m!Z!dd l"m#Z#m$Z$d d l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,d dl-m.Z.d dl/m0Z0m1Z1m2Z2d dl3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9ddl:m;Z;ddlm?Z?m@Z@e de e'e e)e e+e,fZAe dZBejeDZEGddedZFGdde eAe7fZGGddeGe(e7fe eBe7feZHGd d!eHe8e7e7fZIGd"d#eHe4e7e7fZJGd$d%eGe*e7fe eBe7feZKGd&d'eKe8e7e7fZLGd(d)eKe5e7e7fZMy)*) annotationsN)ABCabstractmethod) TYPE_CHECKINGAnyListUnionGenericTypeVarCallableIterableIterator Coroutine AsyncIterator)contextmanagerasynccontextmanager) TypedDictoverride)BodyQueryHeadersNotGiven)consume_sync_iteratorconsume_async_iterator) BetaMessageBetaMessageParam) ToolErrorBetaFunctionToolBetaRunnableToolBetaAsyncFunctionToolBetaAsyncRunnableToolBetaBuiltinFunctionToolBetaAsyncBuiltinFunctionTool)stainless_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/volume1/homes/robertsu/coba/app/.venv/lib/python3.12/site-packages/anthropic/lib/tools/_beta_runner.pyr8r8:s!! 44rBr8F)totalcReZdZddd ddZ ddZd dZd dZy) BaseToolRunnerNmax_iterationscompaction_controlc|Dcic]}|j|c}|_i|d|dDcgc]}|c}i|_t|jj |j d}|r!i||j dxsi} i|d| i}||_d|_d|_||_ d|_ ||_ ycc}wcc}w)Nmessages)toolsrKr9Fr) name_tools_by_name_paramsr'valuesget_options_messages_modified_cached_tool_call_response_max_iterations_iteration_count_compaction_control) selfparamsoptionsrLrHrItoolmessage helper_headermerged_headerss rC__init__zBaseToolRunner.__init__BsBG "'CG'- !#5 #BDs C CcLt|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)callablerO)rXrYs rCset_messages_paramsz"BaseToolRunner.set_messages_params^s! F DLL)F rBc|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)NrKrA)rYmessage_paramss rCz0BaseToolRunner.append_messages..xs$0n60n:GmPZI[Gm^lGm0nrBN) isinstancerrerfrSrbrT)rXrKr\rhs @rCappend_messageszBaseToolRunner.append_messagesmsg$2 #CMWVaBbW\\goo >ho o#2 #'   !no*.' 2 s2AcP|j|j|jk\ryy)NTF)rUrVrXs rC _should_stopzBaseToolRunner._should_stop{s'    +0E0EI]I]0]rB) rY-ParseMessageCreateParamsBase[ResponseFormatT]rZr8rLzIterable[AnyFunctionToolT]rH int | NonerICompactionControl | NonereturnNone)rYzParseMessageCreateParamsBase[ResponseFormatT] | Callable[[ParseMessageCreateParamsBase[ResponseFormatT]], ParseMessageCreateParamsBase[ResponseFormatT]]rrrs)rKz5BetaMessageParam | ParsedBetaMessage[ResponseFormatT]rrrsrrbool)r=r>r?r_rbrkrnrArBrCrFrFAso&*7;6>6 6 * 6 # 656 68 s    /rBrFceZdZddd dfdZddZddZeeddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZxZS)BaseSyncToolRunnerNrGct||||||||_|-|jdrt j dt d|j|_d|_ yN)rYrZrLrHrIenableda!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 superr__clientrQwarningswarnDeprecationWarning__run__ _iterator _last_messagerXrYrZrLclientrHrI __class__s rCr_zBaseSyncToolRunner.__init__sy )1    ).@.D.DY.O MM}#    rBc6|jjSN)r__next__rms rCrzBaseSyncToolRunner.__next__s~~&&((rBc#6K|jD]}|ywrrrXitems rC__iter__zBaseSyncToolRunner.__iter__sNNDJ#sc#KtwrNotImplementedErrorrms rC_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+|||jdddi} tjd| jj t| j,d j.dk7r t1d|j3 fdycc}w) Check token usage and compact messages if threshold exceeded. Returns True if compaction was performed, False otherwise. rzFrcontext_token_threshold Token usage  has exceeded the threshold of . Performing compaction.modelrKre assistantrftypetool_useusersummary_promptrd max_tokensX-Stainless-Helper compactionrrKrr9&Compaction complete. New token usage: text1Compaction response content is not of type 'text'c8i|dddjdgdgiSNrKrr)rrrdrrY first_contents rCriz7BaseSyncToolRunner._check_and_compact..=   &)/(5(:(:$  rBTrW_get_last_messageusage input_tokenscache_creation_input_tokenscache_read_input_tokens output_tokensrQr(loginforOlistrjdictpoprr)rbetarKcreaterfr ValueErrorrb rXr\ tokens_usedtotal_input_tokens thresholdrrKblocknon_tool_blocksresponsers @rC_check_and_compactz%BaseSyncToolRunner._check_and_compactsa  # # +43K3KI3V((*   **==<<AC==88=A?  -w}}/J/JJK,,001JL]^  " < },KI;Vnop((,,Wdll76KL Z01 B< ; . &b\)44EeT*uyy/@J/N4  *9 Y'   00445EG]^  <<$$--44||L1/> 5  9(..:V:V9WXYX--.q1    'PQ Q     cs/-Ic#>K|js|j5}||j}|J|j}|/|j#|jj |j d<ddd|xjdz c_|jsF|j}|tjdy|js|j|d|_ d|_|jsyy#1swYxYwwN containerrz;Tool call was not requested, exiting from tool runner loop.Frnrr_get_last_assistant_messageridrOrVrgenerate_tool_call_responserdebugrSrkrTrXrr\last_assistant_messagers rCrzBaseSyncToolRunner.__run__s##%%%'4 002****.)I)I)K&)5:P:Z:Z:f0F0P0P0S0SDLL-(  ! !Q & !**,;;=#II[\..(((;&+D #.2D +1##%''s$!DAD=BDDDDcDt||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. )rrrX last_messages rC until_donezBaseSyncToolRunner.until_dones, d#--/ '''rBc|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.rTrr_generate_tool_call_responserXrs rCrz.BaseSyncToolRunner.generate_tool_call_response)sD  * * 6 II< =22 2446*2'rBc |j}|sy|Dcgc]}|jdk(s|}}|syg}|D]}|jj|j}|}t j d|jdt|jjdtd|jd|jd|jd d d  |j|j}|jd|j|d d|dScc}w#t$r5}|jd|j|jd d Yd}~'d}~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_idrfis_errorrrrf#Error occurred while calling tool: exc_inforrd#_get_last_assistant_message_contentrrNrQrMrrrkeys UserWarningappendrcallinputrrf Exceptionr exceptionrepr rXrfrtool_use_blocksresultsrr[resultexcs rCrz/BaseSyncToolRunner._generate_tool_call_response7s::<.5RgUz9Q5gR24'H&&**8==9D| X]]O,((,T-@-@-E-E-G(H'IJmn   -'/{{%28==/#M$(  8>>2 hkk^def/(X733eS<  -'/{{#&;;$(   CDII;OZ] ^ -'/{{#'9$(  s/D#D#":D(( G1*E!! G-A GGcdt|jr|jS|jSrrarrms rCrz$BaseSyncToolRunner._get_last_messageps+ D&& '%%' '!!!rBcb|j}||jdk7s |jsy|SNrrrerfrs rCrz.BaseSyncToolRunner._get_last_assistant_messageus4--/  <#4#4 #Cr?r_rrrrrrrrrrrrr __classcell__rs@rCrwrws&*7;>  *   #5 B)Rh36 74r" .rBrwc(eZdZeeddZy)BetaToolRunnerc#K|jjjjdi|j|j }||_|ywNrArrrKparserOrRrrXr\s rCrzBetaToolRunner._handle_requestsG3$,,##,,22ST\\ST]]S$ sAAN)rrz,Iterator[ParsedBetaMessage[ResponseFormatT]]r=r>r?rrrrArBrCr r s rBr c(eZdZeeddZy)BetaStreamingToolRunnerc#K|jjjjdi|j|j 5}|j |_|dddy#1swYyxYwwr rrrKstreamrOrRget_final_messagerrXrs rCrz'BetaStreamingToolRunner._handle_requestsZ/T\\   ' ' . . O O OSY!'!9!9D LP O OsAA4 A( A4(A1-A4N)rrz,Iterator[BetaMessageStream[ResponseFormatT]]rrArBrCrrs rBrceZdZddd dfdZddZddZeeddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZxZS)BaseAsyncToolRunnerNrGct||||||||_|-|jdrt j dt d|j|_d|_ yryr}rs rCr_zBaseAsyncToolRunner.__init__sy )1    ).@.D.DY.O MM}#    rBcRK|jjd{S7wr)r __anext__rms rCrzBaseAsyncToolRunner.__anext__s^^--////s '%'cHK|j23d{}|7 6ywrrrs rC __aiter__zBaseAsyncToolRunner.__aiter__s!.. $J .s "  " "cKtwrrrms rCrz#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+|||jdddid{} tjd| jj t| j,d j.dk7r t1d|j3 fdy7. rrBTrrs @rCrz&BaseAsyncToolRunner._check_and_compactsw  # # +43K3KI3V..00   **==<<AC==88=A?  -w}}/J/JJK,,001JL]^  " < },KI;Vnop((,,Wdll76KL Z01 B< ; . &b\)44EeT*uyy/@J/N4  *9 Y'   00445EG]^  **33::||L1/> ;   9(..:V:V9WXYX--.q1    'PQ Q     U12& s00I9I/DI99-I2&BI97I78A8I92I9cK|js/|j4d{}||jd{}|J|jd{}|/|j#|jj |j d<dddd{|xjdz c_|jd{sN|jd{}|tjdy|js|j|d|_ d|_|js.yy7777#1d{7swYxYw77}wrrrs rCrzBaseAsyncToolRunner.__run__s=##%++-- $ 6 6 88***04/O/O/Q)Q&)5:P:Z:Z:f0F0P0P0S0SDLL-.-  ! !Q & !00222!%!A!A!CC#II[\..(((;&+D #.2D +1##%%-8*R .---3Cs&E'EE'EEE!E "5E E'"E #,E'E#E''E%(AE'E'E E E'E E E E'%E'ctKt|d{|jd{}|J|S7"7 w)rN)rrrs rCrzBaseAsyncToolRunner.until_done8sB %T***!3355 ''' +5s8486 88cK|j!tjd|jS|jd{}||_|S7 w)rNrrrs rCrz/BaseAsyncToolRunner.generate_tool_call_responseBsP  * * 6 II< =22 2::<<*2'=sAAAAcKt|jr|jd{S|jS7wrrrms rCrz%BaseAsyncToolRunner._get_last_messageQs7 D&& '++-- -!!!.s )><>c~K|jd{}||jdk7s |jsy|S7$wrrrs rCrz/BaseAsyncToolRunner._get_last_assistant_messageVs@!3355  <#4#4 #Cr?r_rrrrrrrrrrrrrrrs@rCrrs&*7;!>! ! / !  !#!5! !F0Rh36 " .74rBrc(eZdZeeddZy)BetaAsyncToolRunnercK|jjjjdi|j|j d{}||_|y7wr r rs rCrz#BetaAsyncToolRunner._handle_requestsP9 ))2288Y4<<Y4==YY$ ZsA A  A A N)rrz1AsyncIterator[ParsedBetaMessage[ResponseFormatT]]r=r>r?rrrrArBrCr)r)s rBr)c(eZdZeeddZy)BetaAsyncStreamingToolRunnerc(K|jjjjdi|j|j 4d{}|j |_|dddd{y7+7#1d{7swYyxYwwr rrs rCrz,BetaAsyncStreamingToolRunner._handle_requestsk54<<$$--44Ut||Ut}}UUY_!'!9!9D LVUUUUUUsHA B A9BA=( B3A;4B;B=BB B BN)rrz6AsyncIterator[BetaAsyncMessageStream[ResponseFormatT]]r+rArBrCr-r-s rBr-)N __future__rloggingrabcrrtypingrrrr r r r r rrr contextlibrrtyping_extensionsrrhttpx_typesrrrr_utilsrr types.betarr_beta_functionsrr r!r"r#r$r%_stainless_helpersr'_beta_compaction_controlr(r)r*streaming._beta_messagesr+r,types.beta.parsed_beta_messager-r.r/ types.beta.message_create_paramsr0'types.beta.beta_tool_result_block_paramr1rr2r3r4r6 getLoggerr=rr8rFrwr rrr)r-rArBrCrAs"#    ;1 44C79bbPhhLO4 4S9;RTpp  m$ g!5Ye5=W->?=@@.(8/(IJGT_apTpLqsv@.F'(9/(JO([\01B?1SUd1deE4(/9:GKQ`D`