yj"fdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z m Z m Z ddlmZddlmZmZmZddlmZdd lmZdd lmZmZmZmZmZd d lmZd d l m!Z!d dl"m#Z#m$Z$m%Z%d dl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-e dZ.e dZ/dXdZ0dYdZ1dZdZ2ej3e4Z5d Z6d Z7 d[d\d+Z8d]d.Z9dd/d^dDZ:e0dEdddFd_dVZ;gdWZ.wrapper?s    - -T,V,,,,,,,, - - - - - - - - - - - - - - - - - -s -11)r)r*r+r,r'r$) functoolswraps)r%r/s` r._with_iteration_scoper26s?_R------ NbaseLLMTelemetryContext | None iterationintc |dSt|j|j|j|j||j|j|j|j S)a Return a copy of `base` with `iteration` set, or None if no base. We always copy rather than mutate the caller-supplied context so callers that pass the same context into multiple `honcho_llm_call` invocations don't see drift across concurrent runs. N) workspace_name call_purposeparent_categoryrun_idr6observerobserved peer_name agent_type) r r9r:r;r<r=r>r?r@)r4r6s r._telemetry_for_iterationrAGsS |t *&,{.?    r3 telemetryresponseHonchoLLMCallResponse[Any]Nonec||jsdS|jr|jsdS|jsdS ddlm}m}|||j|j|j|j|j|j|j |d|j D|j |j |j pd|jpd dS#t$r t ddYdSwxYw) uemit AgentIterationEvent after each per-iteration LLM response. Fired immediately after `response = await call_func()` in the per-iteration loop AND after the max-iteration synthesis call. Emitted regardless of whether the model requested tool calls — the no-tool terminating iteration still counts as an iteration for cost calibration. Skipped when telemetry context is missing or lacks the required agent identifiers (no agent → no agent.iteration event). Nr)AgentIterationEventemitcg|] }|d Sname.0tcs r. z)_emit_agent_iteration..sJJJ2BvJJJJr3) r<r;r@r9r=r>r?r6 tool_calls input_tokens output_tokenscache_read_tokenscache_creation_tokensz"Failed to emit AgentIterationEventTexc_info)r<r;r@r9src.telemetry.eventsrGrHr=r>r?tool_calls_maderRrScache_read_input_tokenscache_creation_input_tokens Exceptionloggerdebug)rBr6rCrGrHs r._emit_agent_iterationr__s; 0  $I,@   #J CBBBBBBB    ' ) 9$/(7"+"+#-#JJ1IJJJ%2&4"*"B"Ga&.&J&Oa        " JJJ 9D IIIIIIJsA3B&CCdproviderrcontentrrQlist[dict[str, Any]]thinking_blockslist[dict[str, Any]] | Nonereasoning_detailsdict[str, Any]cddlm}ddlmt|}||fd|D|pg|pg}||S)zEFormat an assistant message with tool calls in provider-native shape.r)CompletionResult)ToolCallResultc xg|]6}|d|d|d|d7S)idrKinputthought_signature)rlrKrmrn)get)rN tool_callrjs r.rPz1format_assistant_tool_message..sc    NT?v&("+--0C"D"D       r3)rbrQrdrf)backendrirjrformat_assistant_tool_message) rarbrQrdrfBackendCompletionResultadapterresultrjs @r.rrrrsEDDDDD''''''*844G $ $    (   (-2+1r   F  0 0 8 88r3 tool_resultsconversation_messagesctt|}|||dS)zHAppend tool results to `conversation_messages` in provider-native shape.N)rextendformat_tool_results)rarvrwrts r.append_tool_resultsr{s9 +844G  !._setup_streams! '. ")    +  !       !+ . .   )   /(/*(8!'          r3r  multiplierminmaxstopwait before_sleepN)r'rr )r|r}rrwrrrrrrrrrBrwrappedrchunkrs````````` ` ` @r.stream_final_responsers6>N                  D' %#N33!QA2>>>.      wyy$}&&&&&&e vvs?B F) stream_finaliteration_callbackrBrrrstr | dict[str, Any] | None tool_executor$Callable[[str, dict[str, Any]], Any]max_tool_iterationsmax_input_tokens int | Noneget_attempt_planCallable[[], AttemptPlan]rrIterationCallback | None:HonchoLLMCallResponse[Any] | StreamingResponseWithMetadatac  45Kddlm}m}t|cxkr tks+nt ddtdtdzd|z|r|ndd g4d }g}d }d }d }d }d }d }|}||krdtjdt d |dzd |||4|krd}|4|4|4|dzfd7   f d }| r9tt| tddd||} n|} | d{V}!||!jz }||!jz }||!jz }||!jz }t%|dz|!|!jst dt)|!jt,rL|!js3|dkr-||dz kr$|dz }4ddd |dz }|rW}"t3|"4 | | |t5|dz }#t7|#||||||!j|dz| S||!_||!_||!_||!_||!_|dz|!_|!jp||!_|!Sj}$tA|$|!j|!j|!j!|!j"}%4|%tG|dzg}&tI|!jD]\}'}(|(d!})|(d"}*|(%d#d$}+t d%|)tM|'|+pdtOi ||)|*d{V},tQ}-|&|+|)|,d&||)|*|,|-d'#tR$rU}.t*d(|)d)|.|&|+|)d*t-|.dd+Yd}.~.d}.~.wwxYwtW|$|&4|x tY|dzd,|!jD|!j|!j|!jpd |!jpd -}/||/n,#tR$rt-d.d/YnwxYw|d kr |d0vrd1}t d2|dz }||kdt-d3|d4|dz5d5}04d|0d ||4|krd}|4|4|rO}"t3|"4 | | |t55 }#t7|#|||||d|dz| Stjdd84 5 f d6 }1| r9tt| tddd||1}2n|1}2|2d{V}3t%t555|3||3_|dz|3_||3jz|3_||3jz|3_||3jz|3_||3jz|3_|3jp||3_|3S)9aRun the iterative tool calling loop for agentic LLM interactions. Loop per iteration: 1. Make an LLM call with tools available 2. Execute any tool calls the LLM requests 3. Append tool results to the conversation 4. Repeat until the LLM stops calling tools or max iterations reached Returns: Final HonchoLLMCallResponse with accumulated token counts and tool call history, or a StreamingResponseWithMetadata if stream_final=True. r)count_message_tokenstruncate_messages_to_fitzmax_tool_iterations must be in [z, z]; zgot user)rolerbrFzTool execution iteration /NTeffective_tool_choicerrwrciteration_for_callr7r'rDc K}t|j|jt  |j |jd|j |||j|t |d{VSNFr rrarrrrrrrA)rrwrrrrrr}rrrBrrrs r._call_with_messagesz.execute_tool_loop.._call_with_messages_s $#%%D.  %k22%+ $ 1. $ 429>PQQ' r3rrrrz$No tool calls in response, finishingziYour last response was empty. Provide a concise answer to the original query using the available context.) r|r}rrwrrrrrrrrrB) rrYrRrSr[rZthinking_content iterationshit_input_token_caprKrmrlzExecuting tool: )tool_id tool_nameru)r tool_input tool_resulttool_result_metadatazTool execution failed for z: zError: )rrruis_errorcg|] }|d SrJrLrMs r.rPz%execute_tool_loop..sNNNr6 NNNr3)r6rQrRrSrTrUziteration_callback failedrV)requiredanyautozJSwitched tool_choice from 'required'/'any' to 'auto' after first iterationz,Tool execution loop reached max iterations ()zYou have reached the maximum number of tool calls. Based on all the information you have gathered, provide your final response now. Do not attempt to call any more tools.c K}t|j|jt |j |jd|jdd|j|t d{VSrr) rrwrrrr}rrsynthesis_iterationrBrrs r. _final_callz&execute_tool_loop.._final_callas!!* M J     !+ . .   !   ' K* 0.y:MNN'          r3)rrrwrcrr7r'rD)r'rD). conversationrrMIN_TOOL_ITERATIONSMAX_TOOL_ITERATIONSrcopyrsetr]r^r r rrRrSr[rZr_rY isinstancerbr~stripappendrrAr!rrrrarrrdrfr enumeraterorrrr\errorr{rwarning)6r}rrrrrrrrrrrrrrrrrrrBrrr6all_tool_callstotal_input_tokenstotal_output_tokenstotal_cache_creation_tokenstotal_cache_read_tokensempty_response_retriesrrr call_funcrCr|rcurrent_providerassistant_messagervseqrprrrrreiteration_datasynthesis_promptrfinal_call_funcfinal_responserwrs6`` ` ````` ` ` @@r.execute_tool_looprs HMLLLLLLL "5 L L L L9L L L L L! -A%AA)<AAA B*(** +   $N 6f*M*M)NI+-N"# ' ) ) )A VQVVATVVWWW  '##$9::=MMM&*#$<$<%'7%% ! BW:O&/!m                8  ,'77%rBBB2" ##II ,I"$$$$$$h33x55#x'KK#8#CC iQAAA'?  LL? @ @ @8+S11  (..00 +Q.. 3a 777&!+&%,, &QQ   0/11 .!-!)*?#1' +''!-#1*?6y)a-PP5!$2!3"50K,C%-%>(1}(;    (6H $$6H !%8H "3NH 0/FH ,"+a-H ,C0C  (O++--69     $  $  &    $$%6777 i!m,,,-/ '(@AA- - NC!&)I"7+JmmD"--G LL7I77 8 8 8 &c7?d ; ; ; "2 & & & $1M)Z$H$HHHHHHH (>'?'?$###*%."-%%%.&0'20D     J)JJqJJKKK###*%."4CFF"4"4$(   ,l2222 K K K:TJJJJJ K >>37JJJ$* ! LL\    Q g ) ) )j NNM7JMMM $a- 1   &=M!N!NOOO#   5 6 69I I I"&  8 8 !#3! !  (')) &%!"7)#%)"7.y:MNN   -*+-(C$;! 1} 3                    0& %#N33!QA2>>>.      &*?,,,,,,,,N  ,?@@ &4N" )A N"4~7R"RN#69U#UN #n&PP. ."HH* *A.A& s-?AO P2A P--P2 A R&R=<R=)rrr{rrrr)r%r&r'r&)r4r5r6r7r'r5)rBr5r6r7rCrDr'rE)NN) rarrbrrQrcrdrerfrer'rg)rarrvrcrwrcr'rE)r|rr}r~rr7rwrcrrrrrrrrrr"rrrr7rrrBr5r'r)*r}r~rr7rrerrcrrrrrr7rrrrrrrrrr"rrrr7rrrrrrrrrrrBr5r'r)=__doc__ __future__rrr0loggingcollections.abcrrrtypingrrr pydanticr tenacityr r r src.configrsrc.exceptionsrsrc.utils.typesrrrrrexecutorrregistryrruntimerrrtypesrrrrr r!r"r#r$r2rAr_ getLogger__name__r]rrrrr{rr__all__rLr3r.rs  #""""">>>>>>>>>>**********@@@@@@@@@@%%%%%%......,+++++222222 Yt__ WT]]"0/J/J/J/Jd  8 $ $4859 99999:LLLL0-1NNNNNNb(37,0+IIIIIIX    r3