`iuZrddlmZddlZddlZddlmZmZddlmZm 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+d d l,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:erddl;me dZ?ej@eAZBGddedZCGdde e>e4fZDGddeDe'e4fe e?e4feZEGdd eEe5e4e4fZFGd!d"eEe1e4e4fZGGd#d$eDe)e4fe e?e4feZHGd%d&eHe5e4e4fZIGd'd(eHe2e4e4fZJdS))) annotationsN)ABCabstractmethod) TYPE_CHECKINGAnyListUnionGenericTypeVarCallableIterableIterator Coroutine AsyncIterator)contextmanagerasynccontextmanager) TypedDictoverride)BodyQueryHeadersNotGiven)consume_sync_iteratorconsume_async_iterator) BetaMessageBetaMessageParam)BetaFunctionToolBetaRunnableToolBetaAsyncFunctionToolBetaAsyncRunnableToolBetaBuiltinFunctionToolBetaAsyncBuiltinFunctionTool)DEFAULT_THRESHOLDDEFAULT_SUMMARY_PROMPTCompactionControl)BetaMessageStreamBetaAsyncMessageStream)ResponseFormatTParsedBetaMessageParsedBetaContentBlock)ParseMessageCreateParamsBase)BetaToolResultBlockParam) AnthropicAsyncAnthropicAnyFunctionToolT)bound RunnerItemTc8eZdZUded<ded<ded<ded<d S) RequestOptionszHeaders | None extra_headersz Query | None extra_queryz Body | None extra_bodyz'float | httpx.Timeout | None | NotGiventimeoutN)__name__ __module__ __qualname____annotations__t/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/anthropic/lib/tools/_beta_runner.pyr6r68sB!!!!444444r@r6F)totalc6eZdZdddddZddZddZddZdS)BaseToolRunnerNmax_iterationscompaction_controlparams-ParseMessageCreateParamsBase[ResponseFormatT]optionsr6toolsIterable[AnyFunctionToolT]rF int | NonerGCompactionControl | NonereturnNonecd|D|_i|dd|dDi|_||_d|_d|_||_d|_||_dS)Nci|] }|j| Sr?)name).0tools rA z+BaseToolRunner.__init__..IsAAA4ty$AAAr@messagescg|]}|Sr?r?rTmessages rA z+BaseToolRunner.__init__..LsCCCWCCCr@Fr)_tools_by_name_params_options_messages_modified_cached_tool_call_response_max_iterations_iteration_count_compaction_control)selfrHrJrKrFrGs rA__init__zBaseToolRunner.__init__@sBA5AAAG G CCz0BCCCG G    "'CG'- !#5   r@ParseMessageCreateParamsBase[ResponseFormatT] | Callable[[ParseMessageCreateParamsBase[ResponseFormatT]], ParseMessageCreateParamsBase[ResponseFormatT]]cRt|r||j}||_dS)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)callabler])rdrHs rAset_messages_paramsz"BaseToolRunner.set_messages_paramsUs0 F   *VDL))F r@rW5BetaMessageParam | ParsedBetaMessage[ResponseFormatT]cpd|Dd_fdd_dS)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. cZg|](}t|tr|j|jdn|)S)rolecontent) isinstancerrnrorYs rAr[z2BaseToolRunner.append_messages..jsL2 2 2 CMWVaBbBb oW\go > > >ho2 2 2 r@Tc2i|dgjdiS)NrW)r])rHmessage_paramsrds rAz0BaseToolRunner.append_messages..os+0t60t:GsV`IaGsdrGs0t0tr@N)r_rir`)rdrWrrs` @rAappend_messageszBaseToolRunner.append_messagesds[ 2 2 #2 2 2 #'   !t!t!t!t!tuuu*.'''r@boolc8|j|j|jkrdSdS)NTF)rarbrds rA _should_stopzBaseToolRunner._should_stoprs%   +0EI]0]0]4ur@) rHrIrJr6rKrLrFrMrGrNrOrP)rHrfrOrP)rWrjrOrPrOru)r;r<r=rerirtrxr?r@rArDrD?sv&*7;666666*     / / / /r@rDceZdZdddd#fdZd$dZd%dZeed%dZd&dZ d%dZ d'dZ d(dZ d(dZ d)d Zd*d"ZxZS)+BaseSyncToolRunnerNrErHrIrJr6rKIterable[BetaRunnableTool]clientr0rFrMrGrNrOrPct|||||||_||_d|_dSN)rHrJrKrFrGsuperre_client__run__ _iterator _last_messagerdrHrJrKr}rFrG __class__s rArezBaseSyncToolRunner.__init__ys` )1       r@r4c4|jSN)r__next__rws rArzBaseSyncToolRunner.__next__s~&&(((r@Iterator[RunnerItemT]c#&K|jD]}|VdSrrrditems rA__iter__zBaseSyncToolRunner.__iter__s,N  DJJJJ  r@c#"KtrNotImplementedErrorrws rA_handle_requestz"BaseSyncToolRunner._handle_request"###r@ruc6 |j |jdsdS|}d}|9|jj|jjpdz|jjpdz}||jjz}|jdt}||krdSt d|d|d|jd |j d }t|j d }|d d d kr:d|d dD}|r ||d d<n| g|td|jdt}|jjj|||j dddi}t d|jjt|jd jdkrt-d| fddS) Check token usage and compact messages if threshold exceeded. Returns True if compaction was performed, False otherwise. NenabledFrcontext_token_threshold Token usage  has exceeded the threshold of . Performing compaction.modelrWrn assistantcng|]2}t|tr|ddk0|3Stypetool_userpdictgetrTblocks rAr[z9BaseSyncToolRunner._check_and_compact..MeT**05yy/@/@J/N/N/N/N/Nr@rousersummary_promptrm max_tokensX-Stainless-Helper compactionrrWrr7&Compaction complete. New token usage: text1Compaction response content is not of type 'text'c.i|dddjdgdgiSNrWrr)rrrmrrH first_contents rArsz7BaseSyncToolRunner._check_and_compact..K   &)/(5(:$   r@Trc_get_last_messageusage input_tokenscache_creation_input_tokenscache_read_input_tokens output_tokensrr%loginfor]listpoprr&rbetarWcreateror ValueErrorri rdrZ tokens_usedtotal_input_tokens thresholdrrWnon_tool_blocksresponsers @rA_check_and_compactz%BaseSyncToolRunner._check_and_compactsk  # +43KI3V +5((**   *=<AC=8=A?  -w}/JJK,001JL]^^  " "5 o ooIoooppp(,,Wdl76KLL Z011 B< ; . .%b\)4O  *9 Y''   0445EG]^^    <$-44|L1/> 5   X(.:VXXYYYX-..q1   ' 'PQQ Q          tr@c#K|s|5}|V|}|J dddn #1swxYwY|xjdz c_|sO|}|t ddS|js| ||d|_d|_ |dSdSNrz;Tool call was not requested, exiting from tool runner loop.F rxrrrbrgenerate_tool_call_responserdebugr_rtr`rdrrZrs rArzBaseSyncToolRunner.__run__s]##%% 3%%'' +4 0022**** + + + + + + + + + + + + + + +  ! !Q & ! !**,, <;;==#II[\\\F.<(((;;;&+D #.2D +'##%% 3 3 3 3 3sAAA"ParsedBetaMessage[ResponseFormatT]cTt||}|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. )rrrd last_messages rA until_donezBaseSyncToolRunner.until_dones5 d###--// '''r@BetaMessageParam | Nonec|j!td|jS|}||_|SaGenerate 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. Nz$Returning cached tool call response.r`rr_generate_tool_call_responserdrs rArz.BaseSyncToolRunner.generate_tool_call_responsesG  * 6 II< = = =2 24466*2'r@c |}|sdSd|D}|sdSg}|D]9}|j|j}|ut jd|jdt |jdtd| d|j d|jd d d  | |j }| d|j |d #t$r[}td |j|| d|j t!|d d Yd}~3d}~wwxYwd|dS)Nc(g|]}|jdk |Srrrs rAr[zCBaseSyncToolRunner._generate_tool_call_response..%$RRRUz9Q9Q59Q9Q9Qr@Tool '-' 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.r stacklevel tool_result Error: Tool ' ' not foundTr tool_use_idrois_errorrrro#Error occurred while calling tool: exc_inforrm#_get_last_assistant_message_contentr\rrSwarningswarnrkeys UserWarningappendidcallinput Exceptionr exceptionreprrdrotool_use_blocksresultsrrUresultexcs rArz/BaseSyncToolRunner._generate_tool_call_response s::<< 4RRgRRR 424'! ! H&**8=99D| nX]nn(,T-@-E-E-G-G(H(Hnnn   -'/{#M8=#M#M#M$(  8>22 hk^deeffff    ODIOOZ] ^^^ -'/{#'99$(  7333s8D E% AE  E%)ParsedBetaMessage[ResponseFormatT] | Nonec`t|jr|S|jSrrhrrws rArz$BaseSyncToolRunner._get_last_messagePs0 D& ' ' (%%'' '!!r@4list[ParsedBetaContentBlock[ResponseFormatT]] | Nonecd|}||jdks|jsdS|jSNrrrnrors rArz6BaseSyncToolRunner._get_last_assistant_message_contentUs<--//  <#4 #C#C.rr@rorrrmrrrrrrrc.i|dddjdgdgiSrrrs rArsz8BaseAsyncToolRunner._check_and_compact..rr@Trrs @rArz&BaseAsyncToolRunner._check_and_compacts  # +43KI3V +5..00000000   *=<AC=8=A?  -w}/JJK,001JL]^^  " "5 o ooIoooppp(,,Wdl76KLL Z011 B< ; . .%b\)4O  *9 Y''  \* % 0445EG]^^    *3::|L1/> ;         X(.:VXXYYYX-..q1   ' 'PQQ Q          tr@c@K|s|4d{V}|WV|d{V}|J dddd{Vn#1d{VswxYwY|xjdz c_|d{VsU|d{V}|t ddS|js| ||d|_d|_ |dSdSrrrs rArzBaseAsyncToolRunner.__run__s##%% 3++-- + + + + + + + $ 6 6 8 8888888**** + + + + + + + + + + + + + + + + + + + + + + + + + + +  ! !Q & ! !0022222222 .-rr@rrz. 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_async_tool(func)` or a `@beta_async_tool` decorated function.rrrrrTrrrrrrmrrs rArz0BaseAsyncToolRunner._generate_tool_call_response(s@@BBBBBBBB 4RRgRRR 424'! ! H&**8=99D| zX]zz(,T-@-E-E-G-G(H(Hzzz   -'/{#M8=#M#M#M$(  #yy88888888 hk^deeffff    ODIOOZ] ^^^ -'/{#'99$(  7333s>D E3AE..E3)rHrIrJr6rKr&r}r1rFrMrGrNrOrPr )rOr*ryr rrr)r;r<r=rer)r,rrrrrrrrrrrrs@rAr%r%os3&*7;40000^RRRRh3333,    """" $$$$.4.4.4.4.4.4.4.4r@r%c6eZdZeeddZdS)BetaAsyncToolRunnerrO1AsyncIterator[ParsedBetaMessage[ResponseFormatT]]c~K|jjjjdi|j|jd{V}||_|WVdSrrrs rArz#BetaAsyncToolRunner._handle_requestZsZ9 )28YY4<Y4=YYYYYYYY$ r@N)rOr:r;r<r=rrrr?r@rAr9r9Ys= Xr@r9c6eZdZeeddZdS)BetaAsyncStreamingToolRunnerrO6AsyncIterator[BetaAsyncMessageStream[ResponseFormatT]]cK|jjjjdi|j|j4d{V}|j|_|WVdddd{VdS#1d{VswxYwYdSrr r#s rArz,BetaAsyncStreamingToolRunner._handle_requestcs54<$-4UUt|Ut}UU       Y_!'!9D LLLL                              sA A #A N)rOr?r<r?r@rAr>r>bs= Xr@r>)K __future__rloggingrabcrrtypingrrrr r r r r rrr contextlibrrtyping_extensionsrrhttpx_typesrrrr_utilsrr types.betarr_beta_functionsrr r!r"r#r$_beta_compaction_controlr%r&r'streaming._beta_messagesr)r*types.beta.parsed_beta_messager+r,r- types.beta.message_create_paramsr.'types.beta.beta_tool_result_block_paramr/rr0r1r2r4 getLoggerr;rr6rDr{rrr%r9r>r?r@rArRs""""""########                          ;:::::::11111111 444444444444CCCCCCCC77777777cbbbbbbbbbPPPPPPPPhhhhhhhhhhLLLLLLOOOOOO5444444447 4S9;RTpp  gm$$ g!!55555Ye555566666W->?666rb$b$b$b$b$(8/(IJGT_apTpLqsvb$b$b$J'(9/(JO([\01B?1SUd1deg4g4g4g4g4(/9:GKQ`D`