蘺h;(ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZmZddlmZddlmZddlmZddlmZdd lmZmZmZmZm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&erdd l'm(Z(m)Z)ej*e+Z,GddZ-dS)) annotationsN) TYPE_CHECKINGAny AsyncIteratorDictIteratorListOptionalTupleUnion) AgentAction AgentFinish AgentStep)dumpd)RunInfo) AddableDict)get_color_mapping)AsyncCallbackManagerAsyncCallbackManagerForChainRunCallbackManagerCallbackManagerForChainRun Callbacks)RUN_KEY)BaseTool)asyncio_timeout) AgentExecutorNextStepOutputcpeZdZUdZ d:ddddddd;dZded<d ed <d ed <ded <ded<ded<ded<eddZ e jd?dZ ed@dZ ed> r2c|jSr7)r-r8s r0r%z$AgentExecutorIterator.agent_executorZs ##r2c,||_|j|_dSr7)r-r&)r/r%s r0r%z$AgentExecutorIterator.agent_executor^s-k r2Dict[str, BaseTool]c.d|jjDS)Nci|] }|j| Sname.0tools r0 z:AgentExecutorIterator.name_to_tool_map..fsFFFD 4FFFr2)r%toolsr8s r0name_to_tool_mapz&AgentExecutorIterator.name_to_tool_mapdsFFD,?,EFFFFr2cPtd|jjDddgS)Ncg|] }|j SrArBrDs r0 z7AgentExecutorIterator.color_mapping..ks = = =4TY = = =r2greenred)excluded_colors)rr%rHr8s r0 color_mappingz#AgentExecutorIterator.color_mappinghs8 = =4#6#< = = =$e,    r2ctdg|_d|_d|_t j|_dS)z} Reset the iterator to its initial state, clearing intermediate steps, iterations, and time elapsed. z0(Re)setting AgentExecutorIterator to fresh statergN)loggerdebugintermediate_steps iterations time_elapsedtime start_timer8s r0r.zAgentExecutorIterator.resetos?  GHHHAC)++r2c|xjdz c_tj|jz |_td|jd|jdddS)zQ Increment the number of iterations and update the time elapsed. zAgent Iterations: z (z.2fz s elapsed)N)rUrWrXrVrRrSr8s r0update_iterationsz'AgentExecutorIterator.update_iterations{sg 1 IKK$/9 U U UD4E U U U U     r2outputsDict[str, Any] run_managerBUnion[CallbackManagerForChainRun, AsyncCallbackManagerForChainRun]rct|j|j|d}|jrt |j|t<|S)NT)return_only_outputs)run_id)rr% prep_outputsr&r#rrbr)r/r\r^prepared_outputss r0make_final_outputsz(AgentExecutorIterator.make_final_outputsse'   , , W$ -       K(/{7I(J(J(J W %r2r/'AgentExecutorIterator'Iterator[AddableDict]c #Ktd|tj|j|jj|jj|j|jj|j |jj }| t|j|j |j } |j|j|jr/g}|j|j|j|j |j|D]{}|||jr]t/|t0rt3|g|jVMt/|t6rt3|g|jV||j|}||||}d|v}|jr|r|V|rdS|j|j|j/n(#t>$r}| |d}~wwxYw|!|VdS)Nz"Initialising AgentExecutorIteratorrBactionsmessagesstepsrkintermediate_step)"rRrSr.r configurer'r%verboser r!on_chain_startrr&r"_should_continuerUrV_iter_next_steprIrPrTappendr$ isinstancer rrkr_consume_next_stepr[_process_next_step_output BaseExceptionon_chain_error_stop r/callback_managerr^ next_step_seqchunk next_stepoutputis_finales r0__iter__zAgentExecutorIterator.__iter__s 9::: *4 N   )   ' I   $ M   (  '55 $% & & K6  ' %66!2#  13 !0@@)&K+ VVE"((///)V%e[99V"-ug"W"W"WWWWW'y99V"-UGen"U"U"UUUU!/BB=QQ &&(((77 ;OO.f<)!X! LLLFG%66!2# H     & &q ) ) )   jj%%%%%%%s7D-H &&H H2H--H2AsyncIterator[AddableDict]c Ktd|tj|j|jj|jj|j|jj|j |jj }| t|j|j |j d{V} t|jj4d{V|j|j|jrPg}|j|j|j|j |j|23d{V}|||jr_t3|t4rt7|g|jWVSt3|t:rt7|g|jWV6|j|}|| ||d{V}d|v}|jr|r|WV|r dddd{VdS|j|j|jP dddd{Vn#1d{VswxYwYnd#tBtDj!f$r"|#|d{VWVYdStH$r!}|%|d{Vd}~wwxYw|#|d{VWVdS)z N.B. __aiter__ must be a normal method, so need to initialize async run manager on first __anext__ call where we can await it z*Initialising AgentExecutorIterator (async)rBNrirlrn)&rRrSr.rror'r%rpr r!rqrr&r"rmax_execution_timerrrUrV_aiter_next_steprIrPrTrtr$rur rrkrrvr[_aprocess_next_step_output TimeoutErrorasyncio_astoprxryr{s r0 __aiter__zAgentExecutorIterator.__aiter__s  ABBB /9 N   )   ' I   $ M   (  -;; $% & & K<        1 &t':'MNN* * * * * * * * )::OT%6) 57M'+':'K'K-* /# (("""""""e&,,U333 -")%=="&1-2Gen'"'"'"!"!"!"!"!"",E9!=!="&1+0'EN'"'"'"!"!"!"!"!(*!% 3 F F} U UI**,,,#'#B#B!;$$F 3&@H -%%$ U* * * * * * * * * * * * * * )::OT%6))* * * * * * * * * * * * * * * * * * * * * * * * * * * Vg23    K00000000 0 0 0 0 FF   ,,Q// / / / / / / /   KK ,,,,,,,,,,,,,,s\=I8AI&7F:=CI&I8-&I&I8& I00I83I04I884K/ K8KKnext_step_output1Union[AgentFinish, List[Tuple[AgentAction, str]]]rctdt|tr1td|||S|j|tdt|dkr;|d}|j |}||||St|S) zj Process the output of the next step, handling AgentFinish and tool return cases. z$Processing output of Agent loop stepzBHit AgentFinish: _return -> on_chain_end -> run final output logicr^+Updated intermediate_steps with step outputrZrNrn) rRrSrur_returnrTextendlenr%_get_tool_returnrr/rr^next_step_action tool_returns r0rwz/AgentExecutorIterator._process_next_step_outputs  ;<<< & 4 4 K LLT   << 0k<JJ J &&'7888 BCCC  A % %/2 ->>?OPPK&||K[|III-=>>>>r2rcKtdt|tr7td|||d{VS|j|tdt|dkrA|d}|j |}||||d{VSt|S) zp Process the output of the next async step, handling AgentFinish and tool return cases. z*Processing output of async Agent loop stepzCHit AgentFinish: _areturn -> on_chain_end -> run final output logicrNrrZrr) rRrSrur_areturnrTrrr%rrrs r0rz0AgentExecutorIterator._aprocess_next_step_output:s  ABBB & 4 4 R LLU   '7[QQQQQQQQ Q &&'7888 BCCC  A % %/2 ->>?OPPK&!]];K]PPPPPPPPP-=>>>>r2ctd|jjj|jj|jfi|j}|||S)zb Stop the iterator and raise a StopIteration exception with the stopped response. ;Stopping agent prematurely due to triggering stop conditionr) rRwarningr%agentreturn_stopped_responseearly_stopping_methodrTr&rr/r^rs r0rzzAgentExecutorIterator._stopVsi TUUUB$*B   5  #  k   ||F |<<)r5r9)r\r]r^r_r5r)r/rfr5rg)r5r)rrr^rr5r)rrr^rr5r)r^rr5r)r^rr5r)rrr^rr5r)rrr^rr5r)__name__ __module__ __qualname____doc__r1__annotations__propertyr&setterr%rIrPr.r[rerrrwrrzrrrrAr2r0rr-sc%% $  %)-1"&!&#4&&&& X ]???]?$$$X$"""" GGGXG   X & & & &        ";&;&;&;&zI-I-I-I-V????8????8 = = = = D D D D E E E E E E E E E Er2r). __future__rrloggingrWtypingrrrrrr r r r langchain_core.agentsr rrlangchain_core.load.dumprlangchain_core.outputsrlangchain_core.runnables.utilsrlangchain_core.utils.inputrlangchain.callbacks.managerrrrrrlangchain.schemarlangchain.toolsrlangchain.utilities.asynciorlangchain.agents.agentrr getLoggerrrRrrAr2r0rs""""""                        +***********666666888888%$$$$$$$$$$$777777EDDDDDDDD  8 $ $[E[E[E[E[E[E[E[E[E[Er2