.Ui0ddlmZddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZddlmZdd lmZdd lmZdd lmZmZmZmZdd lmZmZdd lmZmZm Z m!Z!ddl"m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+e rddlmZdd lmZddl,m-Z-m.Z.edZ/eGddej`Z1eGdde1Z2eGdde1Z3y)) annotationsN) AsyncIterator) dataclassfield) TYPE_CHECKINGAnycast)TypeVar)QueueCompleteSentinel)Agent)AgentOutputSchemaBase)AgentsExceptionInputGuardrailTripwireTriggeredMaxTurnsExceededRunErrorDetails)InputGuardrailResultOutputGuardrailResult) ItemHelpers ModelResponseRunItemTResponseInputItem)logger)RunContextWrapper) StreamEvent)Trace)pretty_print_result!pretty_print_run_result_streaming)ToolInputGuardrailResultToolOutputGuardrailResultTceZdZUded< ded< ded< ded< d ed < d ed < d ed< ded< ded< eej ddZdddZddZ eddZ y) RunResultBasezstr | list[TResponseInputItem]inputz list[RunItem] new_itemszlist[ModelResponse] raw_responsesr final_outputzlist[InputGuardrailResult]input_guardrail_resultszlist[OutputGuardrailResult]output_guardrail_resultszlist[ToolInputGuardrailResult]tool_input_guardrail_resultszlist[ToolOutputGuardrailResult]tool_output_guardrail_resultszRunContextWrapper[Any]context_wrappercy)The last agent that was run.Nselfs i/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/result.py last_agentzRunResultBase.last_agentHsc|r.t|j|std|jt t |jS)aA convenience method to cast the final output to a specific type. By default, the cast is only for the typechecker. If you set `raise_if_incorrect_type` to True, we'll raise a TypeError if the final output is not of the given type. Args: cls: The type to cast the final output to. raise_if_incorrect_type: If True, we'll raise a TypeError if the final output is not of the given type. Returns: The final output casted to the given type. zFinal output is not of type ) isinstancer' TypeError__name__r r!)r1clsraise_if_incorrect_types r2final_output_aszRunResultBase.final_output_asMs@ #:d6G6G+M:3<<.IJ JAt(())r4ctj|j}|jDcgc]}|j }}||zScc}w)zVCreates a new input list, merging the original input with all the new items generated.)rinput_to_new_input_listr$r% to_input_item)r1original_itemsitemr%s r2 to_input_listzRunResultBase.to_input_list_sJ3>3V3VW[WaWa3b6:nnEdT'')E E ))FsA cN|jsy|jdjS)zEConvenience method to get the response ID of the last model response.N)r& response_idr0s r2last_response_idzRunResultBase.last_response_idfs'!!!!"%111r4Nreturn Agent[Any])F)r9ztype[T]r:boolrGr!)rGzlist[TResponseInputItem])rGz str | None) r8 __module__ __qualname____annotations__propertyabcabstractmethodr3r;rArEr/r4r2r#r#'s ))'&L'77399>"@@N#BBO++0 ++*$*22r4r#c2eZdZUded<eddZddZy) RunResultrH _last_agentc|jS)r.)rRr0s r2r3zRunResult.last_agentssr4ct|SN)rr0s r2__str__zRunResult.__str__xs "4((r4NrFrGstr)r8rJrKrLrMr3rVr/r4r2rQrQos   )r4rQceZdZUdZded< ded< ded< ded< ed Zd ed <ed Zd ed<d Zded< ee jd Z ded<ee jd Z ded<edd Z ded<edd Zded<edd Zded<edd Zded<ed&dZd'dZd(d Zd)d!Zd"Zd#Zd*d$Zd+d%Zy),RunResultStreamingaJThe result of an agent run in streaming mode. You can use the `stream_events` method to receive semantic events as they are generated. The streaming method will raise: - A MaxTurnsExceeded exception if the agent exceeds the max_turns limit. - A GuardrailTripwireTriggered exception if a guardrail is tripped. rH current_agentint current_turn max_turnsrr'F)reprzAgentOutputSchemaBase | None_current_agent_output_schemaz Trace | NonetracerI is_complete)default_factoryr_z2asyncio.Queue[StreamEvent | QueueCompleteSentinel] _event_queuez#asyncio.Queue[InputGuardrailResult]_input_guardrail_queueN)defaultr_asyncio.Task[Any] | None_run_impl_task_input_guardrails_task_output_guardrails_taskzException | None_stored_exceptionc|jS)zThe last agent that was run. Updates as the agent run progresses, so the true last agent is only available after the agent run is complete. )r[r0s r2r3zRunResultStreaming.last_agents !!!r4cp|jd|_|jjs5|jj |jjs5|j js6|j j |j js5yy)zaCancels the streaming run, stopping all background tasks and marking the run as complete.TN)_cleanup_tasksrbrdempty get_nowaitrer0s r2cancelzRunResultStreaming.cancels ##))+    ( ( *##))+--335  ' ' 2 2 4--335r4cZK |j|jrtjdd|_n|jr|j j rn |j jd{}t|trN|j|jd{|j j|jn!||j j|j|jd{|j!|jr |jy7#tj$rYcwxYw77J#|j|jd{7|j!wxYww)aStream deltas for new items as they are generated. We're using the types from the OpenAI Responses API, so these are semantic events: each event has a `type` field that describes the type of the event, along with the data for that event. This will raise: - A MaxTurnsExceeded exception if the agent exceeds the max_turns limit. - A GuardrailTripwireTriggered exception if a guardrail is tripped. Tz Breaking due to stored exceptionN) _check_errorsrkrdebugrbrdrogetasyncioCancelledErrorr6r _await_task_safelyri task_donerhrn)r1r@s r2 stream_eventsz RunResultStreaming.stream_eventssn# """$))LL!CD'+D$##(9(9(?(?(A!%!2!2!6!6!88Dd$9:11$2M2MNNN%%//1&&( !!++-9@))$*=*=> > >    !  ! !(( ( "39-- O ?$))$*=*=> > >    !s~F+A!E2&EEE.E26E.7AE2F+%E0&-F+EE+(E2*E++E20F+2F(FF((F+c t|j|j|j|j|j |j |jS)zRReturn a `RunErrorDetails` object considering the current attributes of the class.)r$r%r&r3r,r(r))rr$r%r&r[r,r(r)r0s r2_create_error_detailsz(RunResultStreaming._create_error_detailssJ**nn,,)) 00$($@$@%)%B%B  r4c|j|jkDr5td|jd}|j|_||_|j jsr|j j}|jjr't|}|j|_||_|j jsr|jr~|jjrd|jj}|rHt|t r8t|t"r!|j|j|_||_|j$r~|j$jrd|j$j}|rHt|t r8t|t"r!|j|j|_||_|j&r|j&jrg|j&j}|rJt|t r9t|t"r!|j|j|_||_yyyyy)Nz Max turns (z ) exceeded)r]r^rr|run_datarkrerorpoutputtripwire_triggeredrrhdone exceptionr6 Exceptionrrirj)r1 max_turns_excguardrail_result tripwire_exc run_impl_exc in_guard_exc out_guard_excs r2rsz RunResultStreaming._check_errorss   t~~ -,{4>>:J*-UVM%)%?%?%AM "%2D "--335#::EEG &&99>?OP (,(B(B(D %)5& --335   4#6#6#;#;#=..88:L < ClO<AVAVA^,0,F,F,HL))5&  & &4+F+F+K+K+M66@@BL < ClO<AVAVA^,0,F,F,HL))5&  ' 'D,H,H,M,M,O 88BBDMM9!Em_=-BXBXB`-1-G-G-IM*)6&"F}-P 'r4c|jr4|jjs|jj|jr4|jjs|jj|jr6|jjs|jjyyyrU)rhrrqrirjr0s r2rnz!RunResultStreaming._cleanup_taskss   t':':'?'?'A    & & (  & &t/J/J/O/O/Q  ' ' . . 0  ' '0L0L0Q0Q0S  ( ( / / 11T 'r4ct|SrU)rr0s r2rVzRunResultStreaming.__str__%s 066r4cK|r|js |d{yyy7#tj$rYyt$rYywxYww)zAwait a task if present, ignoring cancellation and storing exceptions elsewhere. This ensures we do not lose late guardrail exceptions while not surfacing CancelledError to callers of stream_events. N)rrvrwr)r1tasks r2rxz%RunResultStreaming._await_task_safely(sK    $4))   s;A%#%A%AAAAAArF)rGNone)rGzAsyncIterator[StreamEvent])rGrrW)rrgrGr)r8rJrK__doc__rLrr`rarbrvQueuerdrerhrirjrkrMr3rqrzr|rsrnrVrxr/r4r2rZrZ|s ,"N<[AFEAR ">RU+E<+K1HM EHLDCH EC? 05T/NN,N7rs" )(++%,/ CJJ*& 0T CL D2CGGD2 D2N  ) )  ) yy yr4