\jddlZddlmZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZdd lmZ dd lmZdd lmZer dd lmZmZmZddZ ddZy#e$r ed wxYw)Nwraps)SPANDATA) DidNotEnable)capture_internal_exceptionsreraise)set_span_errored)agent_workflow_spanend_invoke_agent_span)_capture_exception)AgentsExceptionzOpenAI Agents not installed) TYPE_CHECKING)Any AsyncIteratorCallablec0tdfd }|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. c Ktj5|dj}t|5}|j d}|r'||_|j tj||g|dd}  |i|d{}t%j*||cdddcdddS7.#t$r}tj}t5t|t|jdd}|2t|dd} | #| j t#| t%||dddn #1swYnxYwt'|Yd}~d}~wt($rR}tj}t5t|dddn #1swYnxYwt'|Yd}~%d}~wwxYw#1swYnxYw dddy#1swYyxYww)Nrconversation_idcontext_wrapper_sentry_agent_span) sentry_sdkisolation_scopecloner get_sentry_conversation_idset_datarGEN_AI_CONVERSATION_IDrsysexc_inforr getattrrun_data timestampr r r Exceptionr) argskwargsagent workflow_spanr run_resultexcr!rinvoke_agent_span original_funcs ^/root/env/lib/python3.12/site-packages/sentry_sdk/integrations/openai_agents/patches/runner.pywrapperz$_create_run_wrapper..wrapper"s ' ' ), "GMMOE$U+( "}"(**->"?"4CE1!** 77)QR)''4d'Ef'E!EJ8&j&@&@%H!Q( "( " , ", ""F&'"||~H46N*3/*1#,,@QSW*X*607 /1Et1- !2 =$5$?$?$G 01B C 5ou MNNNX& '"||~H460+3/ 000 X&'=( "( "( " , ", ", "sGGAF9< B8B6B8 F9# G, G6B88 F6EAD? 6 E?E EF9 F6$F1 F  F1F F1+F91F66F99G >G GGGr&rr'rreturnrrr-r/s` r._create_run_wrapperr3s$ =/"/"b Nc0tdfd }|S)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). c~ |dj|jd}|r|_t j |r j t j| _g|dd}  |i|}d  fd t|dr.|j t!  d  fd }||_t|dr(|j" t! d  fd }||_|S#t$r2} jtjt|d}~wwxYw) Nrrrcztdr.jtjt dyy)N_sentry_workflow_span)hasattr__exit__r r!delattr)r(r)sr._close_workflow_spanzK_create_run_streamed_wrapper..wrapper.._close_workflow_spans4u56& && 7677r4 stream_eventsclK |i|23d{}|7 6 y#wxYwwN) stream_args stream_kwargseventr<original_stream_eventss r.wrapped_stream_eventszL_create_run_streamed_wrapper..wrapper..wrapped_stream_eventssL+'=$((5($$e$ $( )*(*s,4(((4 14cancelc< |i|S#wxYwr?r@) cancel_args cancel_kwargsr<original_cancels r.wrapped_cancelzE_create_run_streamed_wrapper..wrapper..wrapped_cancels#+*KI=I(*(*s )r1N)rArrBrr1zAsyncIterator[Any])rHrrIrr1r)rrrr __enter__rrrr8r%r:r r!r r9r=rrF) r&r'rr*r+rErKr<r(rJrDr)r-s @@@@@r.r/z-_create_run_streamed_wrapper..wrapperhs[Q !**%67 ,;E ),E2 !   " "8#B#BO T'4#!QR! &77J 8 : /%/%=%= " ) * +# +6; +% ++ +(=J $ :x ((//O ? # +$ + !/J Q  "M " "CLLN 3 s #   sD D< -D77D<r0rr2s` r._create_run_streamed_wrapperrMWs&" =AAF Nr4)r-Callable[..., Any]r1rN)r functoolsrrsentry_sdk.constsrsentry_sdk.integrationsrsentry_sdk.utilsrrsentry_sdk.tracing_utilsr spansr r utilsr agents.exceptionsr ImportErrortypingrrrrr3rMr@r4r.rYsr &0A5>&61!33<~U'UUQ6 4 556s AA+