o h0@srddlmZddlZddlZddlmZddlmZmZddl 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/eGdddej0Z1eGddde1Z2eGddde1Z3dS)) 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)ToolInputGuardrailResultToolOutputGuardrailResultTc@seZdZUded< ded< ded< ded< d ed < d ed < d ed< ded< ded< eejd&ddZd'd(ddZd)d d!Z ed*d#d$Z d%S)+ 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_wrapperreturn Agent[Any]cCsdS)The last agent that was run.Nselfr.r.N/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/result.py last_agentHszRunResultBase.last_agentFclstype[T]raise_if_incorrect_typeboolr cCs,|rt|j|std|jtt|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__rr )r0r3r5r.r.r1final_output_asMs  zRunResultBase.final_output_aslist[TResponseInputItem]cCs$t|j}dd|jD}||S)zVCreates a new input list, merging the original input with all the new items generated.cSsg|]}|qSr.) to_input_item).0itemr.r.r1 bsz/RunResultBase.to_input_list..)rinput_to_new_input_listr"r#)r0original_itemsr#r.r.r1 to_input_list_s zRunResultBase.to_input_list str | NonecCs|jsdS|jdjS)zEConvenience method to get the response ID of the last model response.N)r$ response_idr/r.r.r1last_response_idfs zRunResultBase.last_response_idNr+r,)F)r3r4r5r6r+r )r+r;)r+rC) r9 __module__ __qualname____annotations__propertyabcabstractmethodr2r:rBrFr.r.r.r1r!'s4   r!c@s.eZdZUded<ed ddZd ddZd S) RunResultr, _last_agentr+cC|jS)r-)rOr/r.r.r1r2sszRunResult.last_agentstrcCt|SN)rr/r.r.r1__str__xzRunResult.__str__NrGr+rQ)r9rHrIrJrKr2rTr.r.r.r1rNos  rNc@s8eZdZUdZded< ded< ded< ded< ed d Zd ed <ed d Zd ed<d Zded< ee j d dZ ded<ee j d dZ ded<edd dZ ded<edd dZded<edd dZded<edd dZded<ed4dd Zd5d"d#Zd6d%d&Zd7d(d)Zd*d+Zd,d-Zd8d/d0Zd9d2d3ZdS):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. r, current_agentint current_turn max_turnsrr%F)reprzAgentOutputSchemaBase | None_current_agent_output_schemaz Trace | Nonetracer6 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_exceptionr+cCrP)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. )rXr/r.r.r1r2szRunResultStreaming.last_agentNonecCsR|d|_|js|j|jr |js'|j|jrdSdS)zaCancels the streaming run, stopping all background tasks and marking the run as complete.TN)_cleanup_tasksr_raempty get_nowaitrbr/r.r.r1cancels    zRunResultStreaming.cancelAsyncIterator[StreamEvent]c Csz_ ||jrtdd|_n?|jr|jrn6z |jIdH}Wn tj y0Yn"wt |t rI| |j IdH|j|n |V|jqW| |jIdH|n| |jIdH|w|jru|jdS)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_errorsrhrdebugr_rarkgetasyncioCancelledErrorr7r _await_task_safelyrf task_donererj)r0r>r.r.r1 stream_eventss>       z RunResultStreaming.stream_eventsrc Cs$t|j|j|j|j|j|j|jdS)zRReturn a `RunErrorDetails` object considering the current attributes of the class.)r"r#r$r2r*r&r')rr"r#r$rXr*r&r'r/r.r.r1_create_error_detailssz(RunResultStreaming._create_error_detailscCsd|j|jkrtd|jd}||_||_|js6|j}|j j r1t |}||_||_|jr|j r\|j r\|j }|r\t|tr\t|trY|jdurY||_||_|jr|j r|j}|rt|trt|tr|jdur||_||_|jr|j r|j}|rt|trt|tr|jdur||_||_dSdSdSdSdS)Nz Max turns (z ) exceeded)rZr[rrwrun_datarhrbrkrloutputtripwire_triggeredrredone exceptionr7 Exceptionrrfrg)r0 max_turns_excguardrail_result tripwire_exc run_impl_exc in_guard_exc out_guard_excr.r.r1ros>             z RunResultStreaming._check_errorscCsZ|jr |js |j|jr|js|j|jr)|js+|jdSdSdSrS)rer{rmrfrgr/r.r.r1rjs  z!RunResultStreaming._cleanup_tasksrQcCrRrS)rr/r.r.r1rT%rUzRunResultStreaming.__str__taskcsP|r$|s&z|IdHWdStjyYdSty#YdSwdSdS)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)r{rrrsr})r0rr.r.r1rt(s  z%RunResultStreaming._await_task_safelyrG)r+ri)r+rn)r+rrV)rrdr+ri)r9rHrI__doc__rJrr]r^r_rrQueuerarbrerfrgrhrKr2rmrvrwrorjrTrtr.r.r.r1rW|sB     1 $ rW)4 __future__rrLrrcollections.abcr dataclassesrrtypingrrrtyping_extensionsr _run_implr agentr agent_outputr exceptionsrrrr guardrailrritemsrrrrr run_contextrrvrtracingrutil._pretty_printrrtool_guardrailsrrr ABCr!rNrWr.r.r.r1s:            G