o ٰi@sddlZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZmZzdd lmZWn eyEed wdd lmZerXdd lmZmZmZdddZ   dddZdS)Nwraps)SPANDATA) DidNotEnable)capture_internal_exceptionsreraise)agent_workflow_spanend_invoke_agent_span)_capture_exception_record_exception_on_span)AgentsExceptionzOpenAI Agents not installed) TYPE_CHECKING)Any AsyncIteratorCallable original_funcCallable[..., Any]returnctdfdd }|S) aA Wraps the agents.Runner.run methods to - create and manage a root span for the agent workflow runs. - end the agent invocation span if an `AgentsException` is raised in `run()`. Note agents.Runner.run_sync() is a wrapper around agents.Runner.run(), so it does not need to be wrapped separately. argsrkwargsrc st|d}t|}|d}|r"||_|tj||g|ddR}z |i|IdH}Wnt y}zIt }t 0t |t|jdd}|durpt|dd} | durp| jdurpt| |t||Wdn1szwYt|WYd}~n5d}~wty}z%t }t t |Wdn1swYt|WYd}~nd}~wwt|j||WdWdS1swYWddS1swYdS)Nrconversation_idcontext_wrapper_sentry_agent_span) sentry_sdkisolation_scopecloner get_sentry_conversation_idset_datarGEN_AI_CONVERSATION_IDr sysexc_inforr getattrrun_data timestampr r r Exceptionr) rragent workflow_spanr run_resultexcr$rinvoke_agent_spanrh/home/ubuntu/.local/lib/python3.10/site-packages/sentry_sdk/integrations/openai_agents/patches/runner.pywrapper!sX         "z$_create_run_wrapper..wrapperNrrrrrrrrr1r/r.r0_create_run_wrappers 1r4cr) a Wraps the agents.Runner.run_streamed method to - create a root span for streaming agent workflow runs. - end the workflow span if and only if the response stream is consumed or cancelled. Unlike run(), run_streamed() returns immediately with a RunResultStreaming object while execution continues in a background task. The workflow span must stay open throughout the streaming operation and close when streaming completes or is abandoned. Note: We don't use isolation_scope() here because it uses context variables that cannot span async boundaries (the __enter__ and __exit__ would be called from different async contexts, causing ValueError). rrrrc s|d|d}|r|_t|r!tj|_g|ddR}z |i|}Wnt yO}z j t t |d}~wwdfdd t|dru|jt   dfd d }||_t|dr|jtdfdd }||_|S)Nrrrrcs*tdrjttddSdS)N_sentry_workflow_span)hasattr__exit__r#r$delattrr/)r)r*r/r0_close_workflow_spans zK_create_run_streamed_wrapper..wrapper.._close_workflow_span stream_events stream_argsr stream_kwargsAsyncIterator[Any]cs<z|i|2z 3dHW}|Vq 6WdSwNr/)r;r<event)r9original_stream_eventsr/r0wrapped_stream_eventsszL_create_run_streamed_wrapper..wrapper..wrapped_stream_eventscancel cancel_args cancel_kwargscs z |i|WSwr>r/)rCrD)r9original_cancelr/r0wrapped_cancelszE_create_run_streamed_wrapper..wrapper..wrapped_cancel)rN)r;rr<rrr=)rCrrDrrr)rrr r __enter__r!rr"r5r(r7r#r$r r6r:rrB)rrrr+r,rArFr.)r9r)rEr@r*r0r1gsH    z-_create_run_streamed_wrapper..wrapperNr2rr3r/r.r0_create_run_streamed_wrapperVsCrH)rrrr)r# functoolsrrsentry_sdk.constsrsentry_sdk.integrationsrsentry_sdk.utilsrrspansr r utilsr r agents.exceptionsr ImportErrortypingrrrrr4rHr/r/r/r0s,      ?