\jdddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z ddl mZddlmZmZddlZ ddlmZddlZddlmZmZmZmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z,dd lm-Z-e-r6dd lm.Z.dd lm/Z/dd lm0Z0ddlm1Z1ddlm2Z2ddlm3Z3ddlm4Z4ddl5m6Z6ddl7m8Z8ejrdZ:dZ;GddeZddde=fdZ?ej dwddd dd!dd"d#d$e=d%e=d&eAdd'fd(ZBej dwddd dd!dd"d#d$e=d%e=d&eAdd)fd*ZC dxd.ZDdyd/ZEd0e=d1d#d2d3d4d3d5d#d6d#de=fd7ZF dzd8ZG d{d9ZH d{d:ZI d|d;ZJdddd?ZLGd@dAZMdBdCdDeAde=fdEZNd}dFZOdddejf d~dJZQ ddGd#dHddKe=ddfdLZR d ddMZSdddNZTdOdPdQd#dRdSdeUfdTZV ddUZWdVddWeAddXfdYZX ddIdZdGeAd[ddeAfd\ZYdIdZdeAfd]ZZdIdZd^e=d_d`d[daddaf dbZ[ddcZ\dIdZd^e=dddddafdeZ]d-dfdIdZd^e=dGeAdgdd_d`d[daddfdhZ^d-dfdIdZd^e=dddddf diZ_djdkde=fdlZ`dmdndoeAddfdpZadGeAdHdqd+d,ddrfdsZbdGeAdHdqde=fdtZcddul*mdZdmeZemfZfm+Z+ddvlgmhZhmiZimjZke-rdd l*m+Z+yy#e$r ddlmZYfwxYw)N)MappingMutableMapping)datetime timedeltatimezone)Random)quoteunquote)Pattern)OPSPANDATA SPANSTATUS SPANTEMPLATE) capture_internal_exceptionsfilename_for_moduleloggermatch_regex_listqualname_from_function safe_repr to_string try_convert is_sentry_urlis_valid_sample_rate_is_external_source_is_in_project_root_module_in_list)Span) TYPE_CHECKING)Any)Dict) Generator)Optional)Union)Iterator)Tuple) FrameType) Attributesz5^[ ]*([0-9a-f]{32})?-?([0-9a-f]{16})?-?([01])?[ ]*$z(([a-zA-Z0-9+/]{4})*([a-zA-Z0-9+/]{2,3})?cHeZdZ d dddeddfdZdeddfd Zdefd Zd d Zy)EnvironHeadersenvironzMapping[str, str]prefixreturnNc ||_||_yN)r*r+)selfr*r+s B/root/env/lib/python3.12/site-packages/sentry_sdk/tracing_utils.py__init__zEnvironHeaders.__init__Ms   key Optional[Any]cv|j|j|jddjzSN-_)r*r+replaceupper)r/r3s r0 __getitem__zEnvironHeaders.__getitem__Us.||DKK#++c3*?*E*E*GGHHr2c8tdt|DS)Nc3 K|]}dyw)N).0r8s r0 z)EnvironHeaders.__len__..Ys)1)s )sumiterr/s r0__len__zEnvironHeaders.__len__Xs)d4j)))r2c#K|jD]i}t|ts|jddj }|j |j sP|t|j dkywr6)r* isinstancestrr9r: startswithr+len)r/ks r0__iter__zEnvironHeaders.__iter__[si (Aa% #s#))+A<< ,C $&' ' (sA:A<)HTTP_)r,zGenerator[str, None, None]) __name__ __module__ __qualname__rHr1r;intrErLr?r2r0r)r)LsO$  IsII** (r2r)optionsOptional[Dict[str, Any]]r,c|yt|jdduxr(|jdduxs|jdduS)z} Returns True if either traces_sample_rate or traces_sampler is defined and enable_tracing is set and not false. NFenable_tracingtraces_sample_ratetraces_sampler)boolgetrRs r0has_tracing_enabledr[gsY   $%U2 KK, -T 9 9{{+,D8  r2Optional[dict[str, Any]]cV|y|jdxsijddk(S)NF _experimentstrace_lifecyclestream)rYrZs r0has_span_streaming_enabledraxs/ KK ' -2 2 23D E QQr2cursorrquery params_list paramstyle Optional[str] executemanyrecord_cursor_repr span_originz.Generator[sentry_sdk.tracing.Span, None, None]c#2Ktjjdjddr|r|dgk(rd}|dk(rd}nd}d}t ||}i}|||d<|||d<|rd|d <|r|||d <t 5tj |d | dddtjtj|| 5}|jD]\} } |j| | |dddy#1swYgxYw#1swYyxYww)Nr^record_sql_paramsFpyformatformat db.params db.paramstyleTdb.executemany db.cursorrcmessagecategorydataopnameorigin) sentry_sdk get_clientrRrY _format_sqlradd_breadcrumb start_spanr DBitemsset_data) rbrcrdrergrhriruspanrKvs r0record_sql_queriesrs?&&~6::;NPUVkdV3K  #!J   &E D'[ *_!% f0"[ $ &N!!%'MN    55   JJL DAq MM!Q   NNs6A>DC?.D/D 6 D?DD DDzUGenerator[Union[sentry_sdk.tracing.Span, sentry_sdk.traces.StreamedSpan], None, None]c#Ktj}|jdjddr|r|dgk(rd}|dk(rd}nd}d}t ||}i}|||d<|||d<|rd|d <|r|||d <t 5tj |d | dddt|jrDtjj|d n||tjd5} | dddytjtj||5} |jD]\} } | j| | | dddy#1swYxYw#1swYyxYw#1swYyxYww)Nr^rkFrlrmrnroTrprqrcrrz)z sentry.originz sentry.op)rx attributesrv)rzr{rRrYr|rr}ratracesr~r rrr) rbrcrdrergrhriclientrurrKrs r0'record_sql_queries_supporting_streamingrs " " $F ~~n%))*=uEkdV3K  #!J   &E D'[ *_!% f0"[ $ &N!!%'MN"&..1    ) )*/-&U!,UU*   J   " "uu    $1 a# $J  NN    sOBE>EAE>.E&3/E>"/E2 E>E#E>&E/+E>2E;7E>scopesentry_sdk.Scoperc\|jtjk(r*|j|jdd|j y|jtj k(rd}|jjtj}|r!d|cxkrdkrnnd}nd|cxkrdkrnnd}|r |jd d |j| y|jd d |j y|jd k(r*|jd d |j|jyy)Nredis)rstypertruiiWerroriiwarninghttphttplib)rrtrulevel)rrtru subprocess)rrtrsru) rwr DB_REDISr} description_tags HTTP_CLIENT_datarYr HTTP_STATUS_CODE)rrr status_codes r0"maybe_create_breadcrumbs_from_spanrs ww"++ $$7W4::   BNN "jjnnX%>%>? k(S( *s*!   idjj !   fytzz R L  !$$   !r2cN |jjS#t$rYywxYwr.)f_code co_filename Exception)frames r0_get_frame_module_abs_pathr s)||''' s  $$is_sentry_sdk_frame namespacein_app_includeOptional[list[str]]in_app_excludeabs_path project_rootct||}t|xs t||}| xr|xst||xr| Sr.)rrr)rrrrrrshould_be_includedshould_be_excludeds r0_should_be_includedrs[)NC,X6/>;# "  T , 7 R@R ++J7I  $ + h55yA""8#:#:IF-e4  $1)XF )h.A.A,.O&..|R@GGO & $ + h44kB*&&'7E !!LL00M  $$ + h44ell6J6JK""8#9#95<<;O;OP %S/ I 4 F  I 2 ! M !sGI9> J J%J/9 JJ JJ J,+J,/ J=<J=crtj}|jsyt|tr |j y|j y|jjdd}|sy|j #tjtjn |j }||j z }|jjdd}|tdz |kD}|syt||jd|jjd |jjd  y) zO Adds OTel compatible source code information to a database query span Nenable_db_query_sourceTdb_query_source_threshold_msrr> millisecondsrrrrrrrrzr{ is_activerGr timestampstart_timestamprRrYrnowrutcrr)rrshould_add_query_source end_timestampduration threshold slow_querys r0add_query_sourcerzs   " " $F    $ # >> !  #$nn001I4P "'+nn&< X\\"$..t333H""#A1EII155 AJ  ^^N3~~))*:;~~))*:; r2crtj}|jsyt|tr |j y|j y|jjdd}|sy|j #tjtjn |j }||j z }|jjdd}|tdz |kD}|syt||jd|jjd |jjd  y) z] Adds OTel compatible source code information to a span for an outgoing HTTP request Nenable_http_request_sourceT http_request_source_threshold_msrr>rrrrrr)rrshould_add_request_sourcerrrrs r0add_http_request_sourcers  " " $F    $ # >> !  # & 2 23OQU V $'+nn&< X\\"$..t333H""#EqIII155 AJ  ^^N3~~))*:;~~))*:; r2c@|sy|jdr|jdr|dd}tj|}|sy|j \}}}d}|rdj t |d}|rdj t |d}|r|d k7}|||d S) zL Given a `sentry-trace` header string, return a dictionary of data. Nz00-z-00z{:032x}z{:016x}0)trace_idparent_span_idparent_sampled)rIendswithSENTRY_TRACE_REGEXmatchgroupsrmrQ)headerrrr sampled_strrs r0extract_sentrytrace_datars  FOOE$:"  $ $V ,E ,1LLN)HnkN##C"$56"))#nb*AB$+(( r2sqlcd} t|drF|j|}t|tr%|j |j j }|xs t|S#t$rd}YwxYw)Nmogrify) hasattrrrGbytesdecode connectionencodingrr)rbrreal_sqls r0r|r|soH  69 %~~c*H(E*#??6+<+<+E+EF  %y~% sAA%% A32A3cHeZdZdZdZ d ddZe ddZede fdZ e jde ddfd Z ede fd Z e jde ddfd Z edd Z de fd ZddZddZddZ ddZde fdZddZddZy)PropagationContextzJ The PropagationContext represents the data of a trace in Sentry. ) _trace_id_span_idrrbaggagecustom_sampling_contextNr,c||_ ||_ ||_ ||_ ||_ ||t ||_d|_yr.)rrrrrBaggager)r/rspan_idrrdynamic_sampling_contextrs r0r1zPropagationContext.__init__se"/ :, H- Q PN ?7C"#;|jjSr.) get_baggagerrDs r0rz+PropagationContext.dynamic_sampling_contextis!::<)rmrrrrrrDs r0__repr__zPropagationContext.__repr__s<mtt NN MM       LL   r2c|jytt|jjj d}| d|cxkrdkrytt|jjj d}t |j |\}} t|j||f}|d |jjd<y#t$r)tjd|j d|d YywxYw) a Ensure that there is a valid sample_rand value in the baggage. If there is a valid sample_rand value in the baggage, we keep it. Otherwise, we generate a sample_rand value according to the following: - If we have a parent_sampled value and a sample_rate in the DSC, we compute a sample_rand value randomly in the range: - [0, sample_rate) if parent_sampled is True, - or, in the range [sample_rate, 1) if parent_sampled is False. - If either parent_sampled or sample_rate is missing, we generate a random value in the range [0, 1). The sample_rand is deterministically generated from the trace_id, if present. This function does nothing if there is no baggage. N sample_randrr> sample_rateintervalz5Could not backfill sample_rand, since parent_sampled=z and sample_rate=..6f) rrfloat sentry_itemsrY_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r/r(r)lowerr:s r0rz$PropagationContext._fill_sample_rands& <<  !%)B)B)F)F})UV  "qK';!'; (< "%)B)B)F)F})UV )$*=*={K u / PUWK7B#5F !!-0  LLGH[H[G\]##.-q2   sC/DDcf|jy|jjjdS)zAConvenience method to get the sample_rand value from the baggage.Nr()rr/rYrDs r0 _sample_randzPropagationContext._sample_rands* << ||((,,];;r2)NNNNNN)rrfrrfrrfrOptional[bool]rzOptional[Dict[str, str]]rOptional[Baggage]r,N)rDict[str, Any]r,r)r,rS)r,r)r,zIterator[Tuple[str, str]])r"r9r,N)rdict[str, Any]r,Nr,N)r,rf)rNrOrP__doc__ __slots__r1 classmethodr propertyrHrsetterrrrrrrr$r&rr6r?r2r0rrsvI%)#'*.+/?C'+H!H!H( H ) H #= H%H HB#,# ##:#__cd ^^ST==11 /?'7? ?  # -H^FNr()splitrstriprSENTRY_PREFIX_REGEXrr rH) rrr6r/rCrDitemr3val baggage_keys r0rzBaggage.from_incoming_headers   S) Wd?02W::CC cn|j tiStj|jS)zC Deprecated: use populate_from_propagation_context )_propagation_contextrr)rrs r0 from_optionszBaggage.from_options s1  % % -2; 889S9STTr2c i}d}d}tj}|js t|S|j}|j |d<|j dr|d|d<|j dr|d|d<|jrH|jj|d<|jjr|jj|d<|j drt|d|d <t|||S) NrFr environmentrelease public_keyorg_idrVr)) rzr{rrrRrrY parsed_dsnrVrWrH)rrr/rCrDrrRs r0rz)Baggage.populate_from_propagation_contexts*, &&(!<( (..#6#?#? Z ;;} %*1-*@L ' ;;y !&-i&8L #   )/):):)E)EL &  '')/):):)A)A X& ;;+ ,*-g6J.K*LL '|%6@@r2cXtj}i}|js t|S|jxsi}|j |d<|j d|d<|jdr|d|d<|jdr|d|d<|jrH|jj|d<|jjr|jj|d<|jr!|jtvr|j|d<|jt|j|d <|j |j rd nd |d <|j"r;|j"j$r%|j'|j"j$t|d S) Populate fresh baggage entry with sentry_items and make it immutable if this is the head SDK which originates traces. rr-r(rTrUrVrW transactionr)truefalsesampledFrD)rzr{rrrRrr6rYrXrVrWrxsourceLOW_QUALITY_TRANSACTION_SOURCESr)rHr^_baggager/r)rr[rr/rRs r0populate_from_transactionz!Baggage.populate_from_transaction4s&&()+ !<( (..&B#.#7#7 Z )4)A)A#(F ]# ;;} %*1-*@L ' ;;y !&-i&8L #   )/):):)E)EL &  '')/):):)A)A X&   ""*II*5*:*:L '  " " .*-k.E.E*FL '    *0;0C0CfL #   K$8$8$E$E    4 4 A A B|U33r2c~tj}i}|js t|S|jxsi}|j |d<|j d|d<|jdr|d|d<|jdr|d|d<|jrH|jj|d<|jjr|jj|d<|jjdtvr|jr|j|d <|jt|j|d <|j |j rd nd |d <|j"r;|j"j$r%|j'|j"j$t|dS)rZrr-r(rTrUrVrWzsentry.span.sourcer[r)r\r]r^Fr_)rzr{rrrRrr6rYrXrVrWget_attributesLOW_QUALITY_SEGMENT_SOURCES_name _sample_raterHr^rbr/r)rsegmentrr/rRs r0populate_from_segmentzBaggage.populate_from_segmentfs &&()+ !<( (..&B#*#3#3 Z )0)=)=c(B ]# ;;} %*1-*@L ' ;;y !&-i&8L #   )/):):)E)EL &  '')/):):)A)A X&  " " $ ( ()= >. /mm*1--L '    +*-g.B.B*CL ' ?? &07fWL #    0 0 = =    0 0 = = >|U33r2cd|_yNFr_rDs r0freezezBaggage.freezes  r2cXi}|jjD] \}}|||< |Sr.)r/r)r/rr3rMs r0rz Baggage.dynamic_sampling_contexts9**002 ICF3K  r2include_third_partycng}|jjD]]\}}t5tjt |zdzt t |z}|j|ddd_|r|j|jdj|S#1swYxYw)NrIrH) r/rrr SENTRY_PREFIXr rHappendrCjoin)r/rorr3rNrMs r0rzBaggage.serializes))//1 #HC,. #,,uSz9C?%C/Q T" # # #  LL// 0xx # #s AB++B4 cPdjd|jdDS)zRemove Sentry baggage from the given header. Given a Baggage header, return a new Baggage header with all Sentry baggage items removed. rHc3zK|]3}tjj|js|5ywr.)rrLrrK)r@rMs r0rAz/Baggage.strip_sentry_baggage..s2 2288F s9;)rsrJ)rs r0strip_sentry_baggagezBaggage.strip_sentry_baggages+ xx "LL-   r2cxtt|jjd}|d|cxkrdkr|Syy)zConvenience method to get the sample_rand value from the sentry_items. We validate the value and parse it as a float before returning it. The value is considered valid if it is a float in the range [0, 1). r(N?)rr.r/rY)r/r(s r0r6zBaggage._sample_randsD "%):):)>)>})MN  "sk'?C'? (@r2cHd|jdd|jdS)Nz )rrDrDs r0r&zBaggage.__repr__s,DNNtNDE[QUQ]Q]P^^_``r2)rT)rrfr6rfr,r)rzsentry_sdk.scope.Scoper,r8)rrr,r)r[zsentry_sdk.tracing.Transactionr,r)ri StreamedSpanr,rr;)r,rE)F)r,Optional[float])rNrOrPr<r=rqrecompilerLrHrXr1r>rrRrrcrjrmrr staticmethodrvr6r&r?r2r0rrsTAIM$"**Z0 "$ &  )- A A& A  A ADUUA"6A AA@/4:/4 /4/4b.4.4` T c   S  S     a#ar2rrzsentry_sdk.client.BaseClienturlcV|jd}t||ryt||dS)zy Returns True if url matches trace_propagation_targets configured in the given client. Otherwise, returns False. trace_propagation_targetsFT)substring_matching)rRrr)rrrs r0should_propagate_tracers1!'/J KVS! C!:t TTr2ci}|jD]@\}}|jdr|dd}|jddj}|||<B|S)z Normalizes incoming data so the keys are all lowercase with dashes instead of underscores and stripped from known prefixes. rMNr8r7)rrIr9r4)rrur3rs r0rrse D#))+ U >>' "ab'Ckk#s#))+S  Kr2rxrtemplatec0ddlmdfd }|S)a Create a span decorator that can wrap both sync and async functions. :param op: The operation type for the span. :type op: str or :py:class:`sentry_sdk.consts.OP` or None :param name: The name of the span. :type name: str or None :param attributes: Additional attributes to set on the span. :type attributes: dict or None :param template: The type of span to create. This determines what kind of span instrumentation and data collection will be applied. Use predefined constants from :py:class:`sentry_sdk.consts.SPANTEMPLATE`. The default is `SPANTEMPLATE.DEFAULT` which is the right choice for most use cases. :type template: :py:class:`sentry_sdk.consts.SPANTEMPLATE` r)should_send_default_piicltjdfd } tj|_tjdfd } tj|_tj r|S|S#t $rYbwxYw#t $rY3wxYw)D Decorator to create a span for the given function. c PKt}|/tjdt  |i|d{St |t r,t jdtd |i|d{S xs t} xst xsd}t||} }|j||5}|j xsit||| || |i|d{}t||||cdddS777!#1swYyxYwwNzbCannot create a child span for %s. Please start a Sentry transaction before calling this function.zBUse the @sentry_sdk.traces.trace decorator in span streaming mode. stacklevelr)rwrxget_current_spanrr3rrGr|warningswarnDeprecationWarning _get_span_op_get_span_name start_child update_data_set_input_attributes_set_output_attributesargskwargs current_spanspan_op function_name span_namesend_piirresultrfrxrwrrs r0 async_wrapperzDcreate_span_decorator..span_decorator..async_wrappersJ+-L# V*1- ////, 5 X&  ////2L2G C$:1$=CM&xGI.0H))*   !1r2%(HmQf !$1&11&tXxH  00 2  sL7D&D;D&6D7A D&2D6D7D D&D&DD#D&c  t}|'tjdt  |i|St |t r$t jdtd |i|S xs t} xst xsd}t||} }|j||5}|j xsit||| || |i|}t||||cdddS#1swYyxYwrrrs r0 sync_wrapperzCcreate_span_decorator..span_decorator..sync_wrapper3s'+-L# V*1- $)&)), 5 X&  $)&))2L2G C$:1$=CM&xGI.0H))*   !1r2%(HmQfD+F+&tXxH   s 2>C::Drrrrr,r functoolswrapsinspect signature __signature__riscoroutinefunction)rrrrrxrwrrs` r0span_decoratorz-create_span_decorator..span_decorators  % %  % N *1*;*;A*>M '  % %  % N )0):):1)=L &  & &q )  g   Z   s#B%B' B$#B$' B32B3rrr,r)sentry_sdk.scoper)rwrxrrrrs```` @r0create_span_decoratorrs,9b b H r2activecdfd }|S)zW Create a span creating decorator that can wrap both sync and async functions. cdtjdfd } tj|_tjdfd } tj|_tj r|S|S#t $rY`wxYw#t $rY3wxYw)rc"Ktj}t|jst j ddxst xsd}t|5|i|d{}|cdddS7#1swYyxYwwNzSUsing span streaming API in non-span-streaming mode. Use @sentry_sdk.trace instead.rrr)rxrrrzr{rarRrrrstart_streaming_span) rrrrrrrrrxs r0rzNcreate_streaming_span_decorator..span_decorator..async_wrappervs**,F-fnn= 1  ? 6q 9?RI%:f  !$1&11   2  s0A"B% B1B2B7 BBB Bctj}t|jst j ddxst xsd}t|5|i|cdddS#1swYyxYwrr)rrrrrrrrxs r0rzMcreate_streaming_span_decorator..span_decorator..sync_wrappers{**,F-fnn= 1  ? 6q 9?RI%:f *$)&) * * *s #A55A>rr)rrrrrrxs` r0rz7create_streaming_span_decorator..span_decoratorqs     " *1*;*;A*>M '   *  *  )0):):1)=L &  & &q )  9   ,   s#B!B# B B # B/.B/rr?)rxrrrs``` r0create_streaming_span_decoratorrhs5 n r2cN|xstj}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )rzget_current_scoper)rrs r0rrs&  3Z113E::L r2cddlm}m}m}t j }|s0t |jr|nt j}|t|trV|jtj|j*|jjtjyyt||r:|j|_|j"|j|j"_yyyy)z Set the status of the current or given span to INTERNAL_ERROR. Also sets the status of the transaction (root span) to INTERNAL_ERROR. r)r| SpanStatus_get_current_streamed_spanN)sentry_sdk.tracesr|rrrzr{rarRrrGr set_statusrINTERNAL_ERRORcontaining_transactionERRORstatus_segment)rr|rrrs r0set_span_erroredrs WV  " " $F *&..9 ' (,,.   dD ! OOJ55 6**6++66z7P7PQ7 l +$**DK}}('1'7'7 $), r2rxryr*rr+tuple[float, float]c|\}}||ks tdt|}t|dz}t|dz} |j||}|dz S#t$r |}Y|dz SwxYw)aGenerate a sample_rand value from a trace ID. The generated value will be pseudorandomly chosen from the provided interval. Specifically, given (lower, upper) = interval, the generated value will be in the range [lower, upper). The value has 6-digit precision, so when printing with .6f, the value will never be rounded up. The pseudorandom number generator is seeded with the trace ID. z/Invalid interval: lower must be less than upperi@B)r2rrQ randrange)rr+r4r:rng lower_scaled upper_scaledsample_rand_scaleds r0r1r1sLE5 5=JKK  Cuy()Luy()L* ]]<F  )) **  )) *sA A'&A'c$||y|durd|fS|dfS)z Compute the lower (inclusive) and upper (exclusive) bounds of the range of values that a generated sample_rand value must fall into, given the parent_sampled and sample_rate values. rTrxryr?)rr)s r0r0r0s0!4 4 KCr2r`r3r4cd}t|tr|j|}|St||r t ||}|S|S#t $rd}Y|SwxYw)z Gets a value from a source object. The source can be a dict or an object. It is checked for dictionary keys and object attributes. N)rGdictrYrgetattrr)r`r3rs r0 _get_valuersj E&$ 3 L 63  , L5L  L s A AAzUnion[str, SPANTEMPLATE]rc|}|tjk(rDd}|r3dD].}|j|st||ts)||}n|rd|}|Sd}|S|tj k(rd|}|S|tj k(rd|}|S)zF Get the name of the span based on the template and the name. N)model model_namezchat chatz invoke_agent z execute_tool )rAI_CHATrYrGrHAI_AGENTAI_TOOL)rrxrrrr3s r0rrsI<''' . ::c?z&+s'C"3KE  (-eE7O  39   \** *#D6*  \)) )#D6* r2ctjtjtjtj tj tji}|j|tj}t|S)z> Get the operation of the span based on the template. ) rrr GEN_AI_CHATrGEN_AI_INVOKE_AGENTrGEN_AI_EXECUTE_TOOLrYFUNCTIONrH)rmappingrws r0rr.s\ bnnr55b44AG Xr{{ +B r7Nr2rrztuple[Any, ...]r:c i|tjtjtjfvrtj t ftj t ftjt ftjt ftjtftjtftjtftjtftjtftjtfd dt ddddffd }t!|j#D]\}}|dk(rCt%|t r3j'tj(gj+d |d N|d k(rCt%|t r3j'tj(gj+d |d ||||tjk(r!|rt-||d tj.<tj(vr-t-tj(tj(<S)z; Get input attributes for the given span template. ) rragent agent_name max_tokensfrequency_penaltypresence_penalty temperaturetop_ptop_kr3rrr,NcJ|vr|\}}|t||r||<yyyyr.)rG)r3r attribute data_typerrs r0 _set_from_keyz,_get_input_attributes.._set_from_keyUs?g~)0&I$E9)E,1Jy)*F$r2promptuser)rolecontent system_promptsystem)rr)rrrrr GEN_AI_REQUEST_MODELrHGEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSrQ GEN_AI_REQUEST_FREQUENCY_PENALTYr.GEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_KlistrrG setdefaultGEN_AI_REQUEST_MESSAGESrrrGEN_AI_TOOL_INPUT) rrrrrr3rrrs @@r0_get_input_attributesr<s$&JL))<+?+?AUAUVV33S9#88#>00#6#55s;#==sC"*"K"KU!S!)!I!I5 Q$??G33U;33S9   2s 25 2T 2 v||~. &JCh:eS#9%%h&F&FKRR#6o%*UC*@%%h&F&FKRR%%8 #u % &<'''H1:V ,2 8--. '':57@ x77 88 8334 r2cidtddddffd }|tjd|tjd|tjd S) z Get usage attributes. rkeysztuple[str, ...]r,Nc^|D]'}t|}|t|ts#||<)yr.)rrGrQ)rrr3rrusages r0_set_from_keysz-_get_usage_attributes.._set_from_keys~s7 .Cuc*E Zs%;(- 9% .r2) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)rHr GEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)r r rs` @r0_get_usage_attributesrxsoJ.#.->.4. **)++.** r2rci}|tjtjtjfvrt 5|g}t |d}|j |t |d}t |d}|j ||D]}||jt|!t |d}|#t|tr||tj<t |d}|#t|tr||tj<ddd|tjk(r|rt||tj<|S#1swYE  Errci}|tjk(r#tjdtj|i}n||tj k(rtjdi}nV|tj k(rCtjdtj|i}|j}|||tj<|jt|||||j|xsiy)a Set span input attributes based on the given span template. :param span: The span to set attributes on. :param template: The template to use to set attributes on the span. :param send_pii: Whether to send PII data. :param f: The wrapped function. :param args: The arguments to the wrapped function. :param kwargs: The keyword arguments to the wrapped function. invoke_agentr execute_toolN) rrr GEN_AI_OPERATION_NAMErrrGEN_AI_TOOL_NAMEr<GEN_AI_TOOL_DESCRIPTIONrrr) rrrrxrrrr docstrings r0rrs&$&J<(((  * *N  & &  \)) )  * *F  \)) )  * *N  % %t II  ;DJx77 8+HhfMNZ%2&r2cD|jt|||xsiy)a% Set span output attributes based on the given span template. :param span: The span to set attributes on. :param template: The template to use to set attributes on the span. :param send_pii: Whether to send PII data. :param result: The result of the wrapped function. N)rr)rrrrs r0rrs! +HhGM2Nr2rr8cttj}|j}|r |jnd}|r|jj dnd}|$|"||k7rt jd|d|dy|jj dd}|r%||||t jd|d|dyy ) z Check if we should continue the incoming trace according to the strict_trace_continuation spec. https://develop.sentry.dev/sdk/telemetry/traces/#stricttracecontinuation NrWzKStarting a new trace because org IDs don't match (incoming baggage org_id: z, SDK org_id: )Fstrict_trace_continuationzvStarting a new trace because strict trace continuation is enabled and one org ID is missing (incoming baggage org_id: T) rzr{rXrWr/rYrr3rR)rrrX client_org_idbaggage_org_idr&s r0rrs  " " $F""J)3J%%M;BW))--h7N !  & ^ + YZhYiiwyFxGGH I &,nn&8&8#U'!  &=+@  "}'@ LLIJXIYYghugvvwx  r2headerszMutableMapping[str, str]sentry_baggagec|jtd}tj|}t |dkDrdnd}||z|z|t<y)zAdd the Sentry baggage to the headers. This function directly mutates the provided headers. The provided sentry_baggage is appended to the existing baggage. If the baggage already contains Sentry items, they are stripped out first. rrrHN)rYrrrvrJ)r)r*existing_baggagestripped_existing_baggage separators r0add_sentry_baggage_to_headersr/sV{{#6; ' < <=M N459rI "I->  r2zOptional[Attributes]z>@K"".)11>>@ +,?,H,HI &fnn&8&89I&JK /88"5"D"D"5"D"D2K I >.tf4TUV/ $K  !9F5F ,TF)F8DE.~~q&..::::  LL06KL M3K'G *4&12 #D6)jkvjwwx y   Kg 55r2ctj}|jjdxsijd}|sydddddtfd}|D]}t |t tfr |||s#y t |ts6d |vsd |vs?d }d }|xsi}d |vr ||d |}d |vr/|d jD]\}} ||vs || ||rd}n|s|sy y) z:Determine if a span fits one of the rules in ignore_spans.r^ ignore_spansFrulerrr,ct|tr+t|trt|j |Sy||k(Srl)rGr rHrX fullmatch)r?rs r0_matchesz!is_ignored_span.._matchess7 dG $%%DNN5122u}r2Trxr) rzr{rRrYrXrGrHr rr) rxrrr>rBr? name_matchesattributes_matchrrs r0is_ignored_spanrEs!  " " $FNN&&~6<"AA.QL uUt dS'N +d# d #4<4;OL# #)rJ~'V d; t#(,\(:(@(@(B$Iu 2(z)4;,1(  0/2 r2)rrarr)rfr|r~)Fmanual)rrrzsentry_sdk.tracing.Spanr,N)rr&r,rf) r>Union[sentry_sdk.tracing.Span, sentry_sdk.traces.StreamedSpan]rrfrrrrr,N)rrGr,N)rrfr,z+Optional[Dict[str, Union[str, bool, None]]])rr9r,r9) rwzOptional[Union[str, OP]]rxrfrr\rrr,r)NNTr.)rzOptional[sentry_sdk.Scope]r,zOptional[Span])rz#Optional[Union[Span, StreamedSpan]]r,N)rr7r)r}r,r)r rr,r:)l contextlibrrrr~rrcollections.abcrrrrrrandomr urllib.parser r rr ImportErrortypingrzsentry_sdk.constsr r rrsentry_sdk.utilsrrrrrrrrrrrrrsentry_sdk.tracingrrrrr r!r"r#r$r%typesr&sentry_sdk._typesr'rrbase64_strippedr)rXr[racontextmanagerrHrrrrrrrrrr|rrrrDEFAULTrrrrr.r1r0rrrrrrrrrr/r<rErrarrrfr|r~rr?r2r0rVs 322'  DD2  , RZZ(W(6!;"R(BRtR  %* * ** *  *  **6**^  %5 5 55 5  5  55]55p  %>   B**    "  &TQ JTQ!TQ*TQ* TQ  TQn( J( (V( J( (V 2B&&C&O&$H<H  $  3D    u3?(*.(  ' : 5 # 9(99 9  9  9x8%(%48%BG%%P)' )'()')'  )'  )'  )' )' )'X O  O6 OBF OPU O  O!$7!D!H '9< &U6 U6&U6 U6D U6p*#*+A*d*\  '_5sJ J/.J/