܎jpddlmZddlZddlZddlmZmZddlmZm 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 dZIejJeKZLGddedZMGdde 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>fZTdS),) 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 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__NC:\Projects\DSCons\.venv\Lib\site-packages\anthropic\lib\tools\_beta_runner.pyr=r=<sB!!!!444444rGr=F)totalc>eZdZdddddZddZddZddZddZdS)BaseToolRunnerNmax_iterationscompaction_controlparams-ParseMessageCreateParamsBase[ResponseFormatT]optionsr=toolsIterable[AnyFunctionToolT]rM int | NonerNCompactionControl | NonereturnNonect||_i|dd|dDi|_t|j|d}|r-t ||dpi}i|d|i}||_d|_d|_ ||_ d|_ ||_ dS)Nmessagescg|]}|SrFrF.0messages rH z+BaseToolRunner.__init__..PsCCCWCCCrG)rRrYr>Fr) r _tools_by_name_paramsr,valuesgetr_options_messages_modified_cached_tool_call_response_max_iterations_iteration_count_compaction_control)selfrOrQrRrMrNr+merged_headerss rH__init__zBaseToolRunner.__init__Ds,E22G G CCz0BCCCG G  0%,,..ZZ ++     C*='++o:V:V:\Z\]]NBB/>BBG "'CG'- !#5   rGParseMessageCreateParamsBase[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`)rirOs rHset_messages_paramsz"BaseToolRunner.set_messages_params`s0 F   *VDL))F rGrY5BetaMessageParam | ParsedBetaMessage[ResponseFormatT]cld|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) isinstancerrtrur[s rHr^z2BaseToolRunner.append_messages..usL2 2 2 CMWVaBbBb oW\go > > >ho2 2 2 rGTc(i|dg|diS)NrYrF)rOmessage_paramss rHz0BaseToolRunner.append_messages..zs)0n60n:GmPZI[Gm^lGm0n0nrGN)rdrore)rirYrxs @rHappend_messageszBaseToolRunner.append_messagesosW 2 2 #2 2 2 #'   !n!n!n!nooo*.'''rGboolc8|j|j|jkrdSdS)NTF)rfrgris rH _should_stopzBaseToolRunner._should_stop}s%   +0EI]0]0]4urGset[str]cBt|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. rY)r"r`r_r}s rH_available_tool_namesz$BaseToolRunner._available_tool_namess$DL$QRRRrG) rOrPrQr=rRrSrMrTrNrUrVrW)rOrlrVrW)rYrprVrWrVr{)rVr)rBrCrDrkrorzr~rrFrGrHrKrKCs&*7;6666668     / / / / SSSSSSrGrKceZdZdddd$fdZd%dZd&dZeed&dZd'dZ d&dZ d(dZ d)dZ d)dZ d*d Zd*d!Zd+d#ZxZS),BaseSyncToolRunnerNrLrOrPrQr=rRIterable[BetaRunnableTool]clientr7rMrTrNrUrVrWct|||||||_|1|drt jdt d||_d|_ dSN)rOrQrRrMrNenableda!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 superrk_clientrbwarningswarnDeprecationWarning__run__ _iterator _last_messagerirOrQrRrrMrN __class__s rHrkzBaseSyncToolRunner.__init__s )1      ).@.D.DY.O.O ) M}#       rGr;c4|jSN)r__next__r}s rHrzBaseSyncToolRunner.__next__s~&&(((rGIterator[RunnerItemT]c#&K|jD]}|VdSrrriitems rH__iter__zBaseSyncToolRunner.__iter__s,N  DJJJJ  rGc#"KtrNotImplementedErrorr}s rH_handle_requestz"BaseSyncToolRunner._handle_request"###rGr{cL |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 dt)d}t d|jjt|jd jdkrt/d| fddS) Check token usage and compact messages if threshold exceeded. Returns True if compaction was performed, False otherwise. NrFrcontext_token_threshold Token usage  has exceeded the threshold of . Performing compaction.modelrYrt assistantcng|]2}t|tr|ddk0|3Stypetool_uservdictrbr\blocks rHr^z9BaseSyncToolRunner._check_and_compact..MeT**05yy/@/@J/N/N/N/N/NrGruusersummary_promptrs max_tokens compactionrrYrr>&Compaction complete. New token usage: text1Compaction response content is not of type 'text'c.i|dddjdgdgiSNrYrr)rrrsrrO first_contents rHryz7BaseSyncToolRunner._check_and_compact..K   &)/(5(:$   rGTrh_get_last_messageusage input_tokenscache_creation_input_tokenscache_read_input_tokens output_tokensrbr-loginfor`listpoprr.rbetarYcreater+rur ValueErrorro rir] tokens_usedtotal_input_tokens thresholdrrYnon_tool_blocksresponsers @rH_check_and_compactz%BaseSyncToolRunner._check_and_compactsn  # +43KI3V +5((**   *=<AC=8=A?  -w}/JJK,001JL]^^  " "5 o ooIoooppp(,,Wdl76KLL Z011 B< ; . .%b\)4O  *9 Y''   0445EG]^^    <$-44|L1' 55 5   X(.:VXXYYYX-..q1   ' 'PQQ Q          trGc#K|s8|5}|V|}|J|}||j|jj|jd<dddn #1swxYwY|xjdz c_|jdkrt ddS| sO| }|t ddS|j s|||d|_ d|_|6dSdSN containerrrefusalz9Turn ended with a refusal, exiting from tool runner loop.z;Tool call was not requested, exiting from tool runner loop.Fr~rr_get_last_assistant_messageridr`rg stop_reasonrdebugrgenerate_tool_call_responserdrzrerirr]last_assistant_messagers rHrzBaseSyncToolRunner.__run__s##%% 3%%'' T4 0022****.)I)I)K)K&)5:P:Z:f0F0P0SDL- T T T T T T T T T T T T T T T  ! !Q & ! ! "i// UVVV**,, <;;==#II[\\\F.<(((;;;&+D #.2D +?##%% 3 3 3 3 3sABB  B "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. )rrri last_messages rH until_donezBaseSyncToolRunner.until_done2s5 d###--// '''rGBetaMessageParam | 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.rerr_generate_tool_call_responserirs rHrz.BaseSyncToolRunner.generate_tool_call_response<sG  * 6 II< = = =2 24466*2'rGc |}|sdSd|D}|sdSg}|}|D]}|j|vr|j|jnd}|ut jd|jdt|jdtd| d|j d|jd d d  | |j }| d|j |d #t$r7}| d|j t|d d Yd}~d}~wt $r[}t"d |j|| d|j t|d d Yd}~}d}~wwxYwd|dS)Nc(g|]}|jdk |Srrrs rHr^zCBaseSyncToolRunner._generate_tool_call_response..O$RRRUz9Q9Q59Q9Q9QrGTool '-' 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_idruis_errorrrru#Error occurred while calling tool: exc_inforrs#_get_last_assistant_message_contentrnamer_rbrrrkeys UserWarningappendrcallinputr#r! Exceptionr exception rirutool_use_blocksresults availablertoolresultexcs rHrz/BaseSyncToolRunner._generate_tool_call_responseJsz::<< 4RRgRRR 424..00 '* * H=E]i=W=W4&**8=999]aD| nX]nn(,T-@-E-E-G-G(H(Hnnn   -'/{#M8=#M#M#M$(  8>22 hk^deeffff    -'/{#5c#:#:$(     ODIOOZ] ^^^ -'/{#5c#:#:$(  7333s%&8D G),E G(AF>>G)ParsedBetaMessage[ResponseFormatT] | Nonec`t|jr|S|jSrrnrr}s rHrz$BaseSyncToolRunner._get_last_messages0 D& ' ' (%%'' '!!rGcZ|}||jdks|jsdS|SNrrrtrurs rHrz.BaseSyncToolRunner._get_last_assistant_messages9--//  <#4 #C#C.rrGrurrrsrrrrrrc.i|dddjdgdgiSrrrs rHryz8BaseAsyncToolRunner._check_and_compact.. rrGTrrs @rHrz&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' 55 ;         X(.:VXXYYYX-..q1   ' 'PQQ Q          trGcK|sd|4d{V}|WV|d{V}|J|d{V}||j|jj|jd< dddd{Vn#1d{VswxYwY|xjdz c_|jdkrt ddS| d{VsU| d{V}|t ddS|j s|||d|_ d|_|bdSdSrrrs rHrzBaseAsyncToolRunner.__run__1s##%% 3++-- T T T T T T T $ 6 6 8 8888888***04/O/O/Q/Q)Q)Q)Q)Q)Q)Q&)5:P:Z:f0F0P0SDL-S T T T T T T T T T T T T T T T T T T T T T T T T T T T  ! !Q & ! ! "i// UVVV0022222222 '+'G'G'I'I!I!I!I!I!I!I ! )4%--rGc K|d{V}|sdSd|D}|sdSg}|}|D]}|j|vr|j|jnd}|ut jd|jdt|jdtd| d|j d|jd d d  | |j d{V}| d|j |d #t$r7}| d|j t|d d Yd}~ d}~wt $r[}t"d |j|| d|j t|d d Yd}~d}~wwxYwd|dS)Nc(g|]}|jdk |Srrrs rHr^zDBaseAsyncToolRunner._generate_tool_call_response..rrGrrz. 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.rrrrrTrrrrrrsrr s rHrz0BaseAsyncToolRunner._generate_tool_call_responses@@BBBBBBBB 4RRgRRR 424..00 '* * H=E]i=W=W4&**8=999]aD| zX]zz(,T-@-E-E-G-G(H(Hzzz   -'/{#M8=#M#M#M$(  #yy88888888 hk^deeffff    -'/{#5c#:#:$(     ODIOOZ] ^^^ -'/{#5c#:#:$(  7333s%.>D-- G7,E)) G6AG  G)rOrPrQr=rRr8rr8rMrTrNrUrVrWr)rVr<rrr r!r")rBrCrDrkr;r>rrrrrrrrrrrr#r$s@rHr7r7sI&*7;!!!!!!!!F0000^RRRRh 3 3 3 3D    """" ....8484848484848484rGr7c6eZdZeeddZdS)BetaAsyncToolRunnerrV1AsyncIterator[ParsedBetaMessage[ResponseFormatT]]c~K|jjjjdi|j|jd{V}||_|WVdSr)r*r,s rHrz#BetaAsyncToolRunner._handle_requestsZ9 )28YY4<Y4=YYYYYYYY$ rGN)rVrMrBrCrDrrrrFrGrHrLrLs= XrGrLc6eZdZeeddZdS)BetaAsyncStreamingToolRunnerrV6AsyncIterator[BetaAsyncMessageStream[ResponseFormatT]]cK|jjjjdi|j|j4d{V}|j|_|WVdddd{VdS#1d{VswxYwYdSr)r2r5s rHrz,BetaAsyncStreamingToolRunner._handle_requests54<$-4UUt|Ut}UU       Y_!'!9D LLLL                              sA A #A N)rVrRrOrFrGrHrQrQs= XrGrQ)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=rKrr&r/r7rLrQrFrGrHrhs""""""########                          ;:::::::11111111 444444444444CCCCCCCC77777777))))))SSSSSSSSSSHGGGGGGGbbbbbbbbbbPPPPPPPPhhhhhhhhhhLLLLLLOOOOOO5444444447 4S9;RTpp  gm$$ g!!55555Ye5555GSGSGSGSGSW->?GSGSGSTH.H.H.H.H.(8/(IJGT_apTpLqsvH.H.H.V'(9/(JO([\01B?1SUd1deM4M4M4M4M4(/9:GKQ`D`