xj2vddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z m Z ddlmZddlmZmZ ddlmZn#e$r ddlmZYnwxYwddlmZddlZddlmZmZmZdd lmZdd lmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,er&dd l-m.Z.dd lm/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6dd l7m8Z8ddlm9Z9ej:dZ;dZ<Gdde Z=ddde>fdZ?ddde>fdZ@ddde>fdZAejB dddddddd d!d"e>d#e>d$eCd%d!dd&fd'ZDdd,ZE dd2ZFdd5ZGd6e>d7d!d8d9d:d9d;d!dfd=ZH dd@ZI ddAZJ ddBZK ddEZLdddFeCdd!fdGZMGdHdIZNGdJdKZOd-d.dLeCde>fdMZP ddOZQddRZRdddejSf ddYZT ddUd!dVdd[e>ddfd\ZU d dd`ZVdadbdcd!dddedeWfdfZX ddkZYdlddmeCddnfdoZZ ddWdpdUeCdqddeCfdrZ[dWdpdeCfdsZ\e]ej^ej_ej`hZadtdude>fdvZbdWdpdwe>dxdydqd+dd+f dzZcdd|ZddWdpd}e>d~ddd+fdZed>ddWdpdwe>dUeCdddxdydqd+ddfdZfd>ddWdpd}e>d~dddf dZgddde>fdZhdddeCddfdZidUeCdVdd]dddfdZjdUeCdVdde>fdZkddllmmZmmnZnddllmoZpddlqmrZrmsZsmtZtmuZuerddlqmuZudSdS)N)MappingMutableMapping)datetime timedeltatimezone)Random)quoteunquote)Pattern) TYPE_CHECKING)OPSPANDATA SPANTEMPLATE)0_apply_data_collection_filtering_to_query_string)_is_external_source_is_in_project_root_module_in_listcapture_internal_exceptionsfilename_for_modulehas_data_collection_enabled is_sentry_urlis_valid_sample_rateloggermatch_regex_listqualname_from_function safe_repr to_string try_convert) FrameType)AnyDict GeneratorIteratorLiteralOptionalTupleUnion) Attributes) ParsedUrlz5^[ ]*([0-9a-f]{32})?-?([0-9a-f]{16})?-?([01])?[ ]*$z(([a-zA-Z0-9+/]{4})*([a-zA-Z0-9+/]{2,3})?cJeZdZ ddddeddfdZdedd fd Zdefd Zdd ZdS)EnvironHeadersHTTP_environzMapping[str, str]prefixreturnNc"||_||_dSN)r-r.)selfr-r.s IC:\Projects\Historical\venv\Lib\site-packages\sentry_sdk/tracing_utils.py__init__zEnvironHeaders.__init__Ss   key Optional[Any]cx|j|j|ddzSN-_)r-r.replaceupper)r2r6s r3 __getitem__zEnvironHeaders.__getitem__[s0|DK#++c3*?*?*E*E*G*GGHHr5cNtdt|DS)Nc3K|]}dVdS)N).0r;s r3 z)EnvironHeaders.__len__.._s"))1))))))r5)sumiterr2s r3__len__zEnvironHeaders.__len__^s%))d4jj))))))r5Generator[str, None, None]c# K|jD]y}t|ts|dd}||js[|t|jdVzdSr9)r- isinstancestrr<r= startswithr.len)r2ks r3__iter__zEnvironHeaders.__iter__as ( (Aa%%  #s##))++A<< ,, C $$&&' ' ' ' ' ( (r5)r,)r/rI) __name__ __module__ __qualname__rLr4r>intrHrPrBr5r3r+r+Rs$  IsIIIII***** ( ( ( ( ( (r5r+optionsOptional[Dict[str, Any]]r/c|dSt|dduo-|ddup|dduS)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)boolgetrUs r3has_tracing_enabledr^msi u  $%%U2 KK, - -T 9 9{{+,,D8   r5Optional[dict[str, Any]]c|dS|dpiddk}|d|ddkS|S)NF _experimentstrace_lifecyclestream)r\)rUis_enabled_in_experiment_configs r3has_span_streaming_enabledre~squ'.{{>'B'B'Hb&M&M'' '#{{$%%1{{,--99 **r5cX|dS|dd ot| S)NTstream_gen_ai_spans)r\rer]s r3should_truncate_gen_ai_inputrhs?t{{t  2(11 12r5Fmanualcursorr query params_list paramstyle Optional[str] executemanyrecord_cursor_repr span_originspan_op_override_valuezUGenerator[Union[sentry_sdk.tracing.Span, sentry_sdk.traces.StreamedSpan], None, None]c#Ktj}t|jr,|jddr|r|dgkrd}|dkrd}n>d}d}n9|jdddr|r|dgkrd}|dkrd}nd}d}t ||}i} ||| d<||| d <|rd | d <|r||| d <t 5tj|d | dddn #1swxYwYt|jr`i} ||| d<tj |dn|||r|n tj d| 5} | VddddS#1swxYwYdStj ||n tj ||5} | D]\} } | | | | VddddS#1swxYwYdS)Ndata_collectiondatabase_query_datapyformatformatrarecord_sql_paramsFz db.paramsz db.paramstyleTzdb.executemanyz db.cursorrk)messagecategorydataz db.query.textz)z sentry.origin sentry.op)name attributes)opr}origin) sentry_sdk get_clientrrUr\ _format_sqlradd_breadcrumbretraces start_spanr DBitemsset_data)rjrkrlrmrorprqrrclientr{additional_attributesspanrOvs r3record_sql_queriesrsp " $ $F"6>22 >+ ,-B C  #+$"7"7" Z''% KJJ >. ) - -.A5 I I  #+$"7"7" Z''% KJ  & &E D'[ *_&!% #f0"[ $ & &NN!%'MMMMNNNNNNNNNNNNNNN"&.11 "  5: !/ 2   ) )*/-&&U!,)33U  ( *   JJJ                   ")?)K%%QSQV      $ $1 a####JJJ                  s6 C11C58C5E##E'*E'5GGG status_code Optional[int]r{dict[str, Any]cd}|r%d|cxkrdkrnnd}nd|cxkrdkrnnd}dd|d }|r||d <tjd i|dS) NiiWerroriiwarninghttphttplib)typerzr{levelrB)rr)rr{rkwargss r3add_http_breadcrumbrs E + $ $ $ $ $ $ $ $ $EE K & & & &3 & & & & &E(.ItTTF  w'''''''r5rsentry_sdk.client.BaseClient parsed_urlOptional[ParsedUrl]r(ci}||St|jr1d}|jr't|j|jdd}n|r|j}n|S|j}|r||t j<|d|zz }|jr!|j|t j <|d|jzz }||t j <|S)aBuild the `url.*` span attributes for an outgoing HTTP request. The query string is only included when the user has opted into collecting it, either through `data_collection` (in which case the configured filtering is applied) or through the legacy `send_default_pii`. Nrturl_query_params) query_string behaviour?#) rrUrkrshould_send_default_piiurlr URL_QUERYfragment URL_FRAGMENTURL_FULL)rrr~rkurl_fulls r3get_url_attributesrs "J#6>22    D'- .):; |jjS#t$rYdSwxYwr1)f_code co_filename Exception)rs r3_get_frame_module_abs_pathrs4|'' tts  is_sentry_sdk_frame namespacein_app_includeOptional[list[str]]in_app_excludeabs_path project_rootct||}t|pt||}| o|pt||o| Sr1)rrr)rrrrrrshould_be_includedshould_be_excludeds r3_should_be_includedr si)NCC,X66/>;;# "  T , 7 7 R@RUnion[sentry_sdk.tracing.Span, sentry_sdk.traces.StreamedSpan]ctj}|wt|} |jd}n#t $rd}YnwxYw|duo|d}t||||||}|rn |j}|wd}|= |j } n#t $rd} YnwxYw| Lt|tr!| tj| n|d|  |jd}n#t $rd}YnwxYw|Vt|tr!| tj|n |tj|t|} | |t#|| } nM|I| |r4| |dt(j} n| } t|tr!| tj| n| |d|  |jj} n#t $rd} YnwxYw| ot|tr,| tj|jjdS|tj|jjdSdSdS)zB Adds OTel compatible source code information to the span NrQz sentry_sdk.)rrrrrrzcode.line.numberzcode.file.path)sys _getframer f_globalsr\rrMrf_backf_linenorKSpanrr CODE_LINENO set_attributeCODE_NAMESPACErr<lstripossep CODE_FILEPATHrco_name CODE_FUNCTION) rrrrrrrrrlinenofilepath in_app_path code_functions r3 add_sourcer3sE'*mooE  -e44 ).)<)>> ++J77II   III   $%% G h5yAAAA""8#:IFFF-e44  $1)XFF )h.A.A,.O.O)&..|R@@GGOO & $%% F h4kBBBB*&&'7EEE !!L0MM ! ! ! MMM !  $$%% Q h4el6JKKKKK""8#95<;OPPPPP[R % $sGA AAB B*)B*<D D&%D& I IIctj}|sdSt|tr |jdS|jdS|jdd}|sdSt|tr|j }n|j}|ptj tj}||jz }|jdd}|tdz |k}|sdSt!||jd|jd |jd  dS) zO Adds OTel compatible source code information to a database query span Nenable_db_query_sourceTdb_query_source_threshold_msrrA millisecondsrrrrrrrrr is_activerKr timestampstart_timestamprUr\ StreamedSpan end_timestamprnowrutcrr)rrshould_add_query_sourcerduration threshold slow_querys r3add_query_sourcers[  " $ $F     $ > ! F #$n001I4PP "$ %%'*  !?X\(,%?%?Mt33H""#A1EEII15555 AJ  ^N3~))*:;;~))*:;; r5ctj}|sdSt|tr |jdS|jdS|jdd}|sdSt|tr|j }n|j}|ptj tj}||jz }|jdd}|tdz |k}|sdSt!||jd|jd |jd  dS) z] Adds OTel compatible source code information to a span for an outgoing HTTP request Nenable_http_request_sourceT http_request_source_threshold_msrrArrrrrr)rrshould_add_request_sourcerrrrs r3add_http_request_sourcers\  " $ $F     $ > ! F # & 2 23OQU V V $$ %%'*  !?X\(,%?%?Mt33H""#EqIIII15555 AJ  ^N3~))*:;;~))*:;; r5header+Optional[Dict[str, Union[str, bool, None]]]c|sdSd|vr9d|dD}td|Dd}|sdS|dr|dr |dd }t|}|sdS|\}}}d}|r#d t|d }|r#d t|d }|r|d k}|||dS)zL Given a `sentry-trace` header string, return a dictionary of data. N,c6g|]}|SrB)striprCparts r3 z,extract_sentrytrace_data..s <<<$<<.s'666t666666r5rz00-z-00z{:032x}z{:016x}0)trace_idparent_span_idparent_sampled) splitnextrMendswithSENTRY_TRACE_REGEXmatchgroupsrwrT)rpartsrrr sampled_strrs r3extract_sentrytrace_datarsJ t f}}<<&,,s*;*;<<<66666;; t FOOE$:$:"  $ $V , ,E t,1LLNN)HnkN7##C"$5$566C"))#nb*A*ABB,$+((  r5sqlcd} t|drI||}t|tr||jj}n#t$rd}YnwxYw|pt|S)Nmogrify) hasattrrrKbytesdecode connectionencodingrr)rjrreal_sqls r3rrsH  69 % % G~~c**H(E** G#??6+<+EFF   %y~~%sAA A-,A-cbeZdZdZdZ d) d*dZe d+dZede fdZ e j de ddfdZ ede fdZ e j de ddfdZ ed,dZ de fdZd-dZd.dZd/d Z d0d#Zde fd$Zd1d%Zd2d&Zd3d(ZdS)4PropagationContextzJ The PropagationContext represents the data of a trace in Sentry. ) _trace_id_span_idrrbaggagecustom_sampling_contextNrrnspan_idrrOptional[bool]dynamic_sampling_contextOptional[Dict[str, str]]rOptional[Baggage]r/c||_ ||_ ||_ ||_ ||_ ||t ||_d|_dSr1)rrrrrBaggager)r2rrrrrrs r3r4zPropagationContext.__init__/sn"/ :, H- Q PN ?7C"#;<)rwrrrrrrGs r3__repr__zPropagationContext.__repr__s5mtt N M     L    r5c|jdStt|jjd}|d|cxkrdkrnndS|}t |j|\}} t|j ||f}n1#t$r$tj d|jd|dYdSwxYw|d |jjd<dS) 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_randrrAintervalz5Could not backfill sample_rand, since parent_sampled=z and sample_rate=..6f) rrfloat sentry_itemsr\ _sample_rate_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r2rI sample_ratelowerr=s r3r%z$PropagationContext._fill_sample_rands&& <  F!%)B)F)F})U)UVV  "qK';';';';!';';';';'; F'')) )$*={KK u / PUWWWKK    L2H[22#.222    FF 7B3G3G !-000s=B*CCcR|jdS|jjdS)zAConvenience method to get the sample_rand value from the baggage.NrI)rrOr\rGs r3 _sample_randzPropagationContext._sample_rands( < 4|(,,];;;r5Optional[float]cx|jdStt|jjdS)zAConvenience method to get the sample_rate value from the baggage.NrU)rrrNrOr\rGs r3rPzPropagationContext._sample_rates2 < 45$,";"?"? "N"NOOOr5)NNNNNN)rrnrrnrrnrrrrrrr/N)rrr/r)r/rV)r/r)r/r<)r@rr/N)rrr/Nr/N)r/rnr/rY)rQrRrS__doc__ __slots__r4 classmethodr,propertyrLrsetterrrr9r7r?r$rErGr%rXrPrBr5r3rr!sxI%)#'*.+/?C'+H!H!H( H ) H #= H%H HHHHB#,# ###[#:#X_cd_X ^ST^===X=11111 ////?'7? ????  #    -H-H-H-H^<<<<PPPPPPr5rcBeZdZdZdZdZejdZ d)ddd e d e fd Z e d d  d*dZ e d+dZe d,dZe d-dZe d.dZd/d Zd0d!Zd1d#e de fd$Zede de fd%Zd2d'Zde fd(Zd S)3raF The W3C Baggage header information (see https://www.w3.org/TR/baggage/). Before mutating a `Baggage` object, calling code must check that `mutable` is `True`. Mutating a `Baggage` object that has `mutable` set to `False` is not allowed, but it is the caller's responsibility to enforce this restriction. rOthird_party_itemsmutablezsentry-z^sentry-rTrODict[str, str]rdrec0||_||_||_dSr1rc)r2rOrdres r3r4zBaggage.__init__s )!2 r5N)rXrrnrXr/ci}d}d}|r|dD]}d|vrt5|}|dd\}}tj|r=t |dd} t ||| <d}n ||rdnd|zz }dddn #1swxYwY|t||d <d}t|||S) zF freeze if incoming header already has sentry baggage rTr=rAr:FNrI)rrrrSENTRY_PREFIX_REGEXrr rL) r&rrXrOrdreitemr6val baggage_keys r3r"zBaggage.from_incoming_header sw   W S)) W Wd??022WW::< > >|U3333r5cd|_dSNFrrGs r3freezezBaggage.freezes  r5cRi}|jD] \}}|||< |Sr1)rOr)r2rr6rks r3rz Baggage.dynamic_sampling_contexts9*0022  ICF3KK r5Finclude_third_partycg}|jD]|\}}t5tjt |zdzt t |z}||dddn #1swxYwY}|r||jd |S)Nrir) rOrrr SENTRY_PREFIXr rLappendrdjoin)r2rrr6rlrks r3r>zBaggage.serializes)//11 # #HC,.. # #,uSzz9C?%C//Q T""" # # # # # # # # # # # # # # #  1 LL/ 0 0 0xxsAB  B B cfdd|dDS)zRemove Sentry baggage from the given header. Given a Baggage header, return a new Baggage header with all Sentry baggage items removed. rc3|K|]7}tj|3|V8dSr1)rrjrr)rCrks r3rDz/Baggage.strip_sentry_baggage..sU  288FF       r5)rr)rs r3strip_sentry_baggagezBaggage.strip_sentry_baggagesB xx  "LL--      r5rYctt|jd}|d|cxkrdkrnn|SdS)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). rIN?)rrNrOr\)r2rIs r3rXzBaggage._sample_randsV "%):)>)>})M)MNN  "sk'?'?'?'?C'?'?'?'?'? tr5cFd|dd|jdS)Nz )r>rerGs r3rGzBaggage.__repr__s-`DNNtNDD``QUQ]````r5)rT)rrnrXrnr/r)rnror/r)r'rr/r)ryrzr/r)rrr/rr[)r/rf)Fr\)rQrRrSr]r^rrecompilerjrLr[r4r_r"rrr;rrrrr> staticmethodrrXrGrBr5r3rrs)AIM$"*Z00 "$ &  )- A A A& A  A A A[ ADUUU[UA"6A AAA[A@/4:/4 /4/4/4[/4b/4/4/4[/4b  T c      S  S    \      a#aaaaaar5rrcd|jd}t||rdSt||dS)zy Returns True if url matches trace_propagation_targets configured in the given client. Otherwise, returns False. trace_propagation_targetsFT)substring_matching)rUrr)rrrs r3should_propagate_tracers?!'/J KVS!!u C!:t T T TTr5requestc t|drt|dstjddSt|t |jsdSt jD]S\}}tj d|d|d|jd|tkrt|j |I||j |<TdS) ai Attach Sentry trace propagation headers (``sentry-trace``/``baggage``) from the current scope's propagation context to an outgoing request, if the request's URL matches the configured ``trace_propagation_targets``. ``request`` is expected to expose ``url`` and a mutable ``headers`` mapping (e.g. an ``httpx``/``httpx2`` ``Request``). rheaderszPUnable to propagate trace headers in request - missing url or headers attributesNz[Tracing] Adding `z ` header z to outgoing request to rL) r rrrrLrrget_current_scopeiter_trace_propagation_headersrTr!add_sentry_baggage_to_headersr)rrr6r1s r3propagate_trace_headersrs 7E " "''9*E*E ^     !&#gk*:*: ; ; 244SSUU)) U \ \ \u \ \gk \ \ \    % % % )'/5 A A A A#(GOC ))r5rrci}|D]Q\}}|dr |dd}|dd}|||<R|S)z Normalizes incoming data so the keys are all lowercase with dashes instead of underscores and stripped from known prefixes. r,Nr;r:)rrMr<rV)rr{r6r1s r3rrsx D#))++ U >>' " " abb'Ckk#s##))++S Kr5rOptional[Union[str, OP]]r}r~templaterc dfd }|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` fr r/crtjdfd } tj|_n#t $rYnwxYwtjdfd } tj|_n#t $rYnwxYwtjr|S|S) D Decorator to create a span for the given function. argsr rr/c Kt}|0tjdt  |i|d{VSt |t r*t jdtd |i|d{VS pt} pt pd}t||}td}td}| ||5}| pit||| || |i|d{V} t||| | cdddS#1swxYwYdS NzbCannot 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. stacklevelrinputsoutputs)rr}get_current_spanrrTrrKrwarningswarnDeprecationWarning _get_span_op_get_span_name_should_collect_gen_ai start_child update_data_set_input_attributes_set_output_attributesrr current_spanspan_op function_name span_namecollect_inputscollect_outputsrresultr~rr}rrs r3 async_wrapperzDcreate_span_decorator..span_decorator..async_wrapperJs+--L# V*1-- Q//////////, 55 0 X&  Q//////////2L22G C$:1$=$=CM&xGGI3H==N4Y??O))*   !1r222%(NM1dF !q$1&11111111&tXOOO                  s-AEE E c t}|*tjdt  |i|St |t r$t jdtd |i|S pt} pt pd}t||}td}td}| ||5}| pit||| || |i|} t||| | cdddS#1swxYwYdSrrrs r3 sync_wrapperzCcreate_span_decorator..span_decorator..sync_wrapperxs+--L# V*1-- q$)&))), 55 * X&  q$)&)))2L22G C$:1$=$=CM&xGGI3H==N4Y??O))*   !1r222%(NM1dFD+F++&tXOOO                  sAD44D8;D8Nrr rr r/r ) functoolswrapsinspect signature __signature__riscoroutinefunction)rrrr~r}rrs` r3span_decoratorz-create_span_decorator..span_decoratorEs/   & & & & & & & & &   & P *1*;A*>*>M ' '    D    & & & & & & & & &   & P )0):1)=)=L & &    D   &q ) )   s#A AA7B BBrr r/r rB)rr}r~rrs```` r3create_span_decoratorr.sE.d d d d d d d d d L r5Tactivecdfd }|S)zW Create a span creating decorator that can wrap both sync and async functions. rr r/crtnidvrtjd<tjd fd } t j|_n#t$rYnwxYwtjd fd } t j|_n#t$rYnwxYwt j r|S|S) rr|rr rr/cHKtj}|r*t|jst jddptpd}t|5|i|d{V}|cdddS#1swxYwYdSNzSUsing span streaming API in non-span-streaming mode. Use @sentry_sdk.trace instead.rrr)r}r~r rrrrerUrrrstart_streaming_span) rrrrrrrr}new_attributess r3rzNcreate_streaming_span_decorator..span_decorator..async_wrappers!*,,F!! *DV^*T*T  1  ? 6q 9 9?RI%>&   !q$1&11111111                   s:BBBc4tj}|r*t|jst jddptpd}t|5|i|cdddS#1swxYwYdSrr)rrrrrrr}rs r3rzMcreate_streaming_span_decorator..span_decorator..sync_wrappers*,,F!! *DV^*T*T  1  ? 6q 9 9?RI%>& * *q$)&)) * * * * * * * * * * * * * * * * * *s8B  BBNr) dictr FUNCTIONrrrrrrr)rrrrrr~r}s` @r3rz7create_streaming_span_decorator..span_decoratorsR.8?j)))R n , ,*,+N; '              " *1*;A*>*>M ' '    D     * * * * * * * *   *  )0):1)=)=L & &    D   &q ) )   s$A(( A54A5B66 CCrrB)r}r~rrs``` r3create_streaming_span_decoratorrs68 8 8 8 8 8 8 8 t r5rnOptional[sentry_sdk.Scope]Optional[Span]c>|ptj}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )rrr)rnrs r3rrs$  3Z133E:L r5rrrJrrKtuple[float, float]c|\}}||kstdt|}t|dz}t|dz} |||}n#t$r|}YnwxYw|dz S)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)rSrrT randrange)rrKrVr=rng lower_scaled upper_scaledsample_rand_scaleds r3rRrRsLE5 5==JKKK   Cuy())Luy())L* ]]<FF *****  ))sA&& A54A5rrrUrYc&||dS|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. NrTrrrB)rrUs r3rQrQs5!4x 4  KCr5rr6r7cd}t|tr||}n4t||r$ t ||}n#t $rd}YnwxYw|S)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)rKrr\r getattrr)rr6r1s r3 _get_valuer.s} E&$ 3 63    ,,     LsA AAzUnion[str, SPANTEMPLATE]rc|}|tjkrMd}|r?dD]<}||r%t||tr ||}n=|rd|nd}n+|tjkrd|}n|tjkrd|}|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_CHATr\rKrLAI_AGENTAI_TOOL)rr}rrrr6s r3rr?sI<'''  .  ::c??z&+s'C'C"3KEE',8OEOOO& \* * **D** \) ) )*D** r5ctjtjtjtjtjtji}||tj }t|S)z> Get the operation of the span based on the template. ) rrr GEN_AI_CHATrGEN_AI_INVOKE_AGENTrGEN_AI_EXECUTE_TOOLr\rrL)rmappingrs r3rr\sN bnr5b4AG Xr{ + +B r77Nr5kindzLiteral["inputs", "outputs"]ctj}t|jr&t |jdd|S|S)Nrtgen_ai)rrrrUr[r)rrs r3rrosU  " $ $F"6>22GFN#45h?EFFF  ) ) + ++r5rrztuple[Any, ...]c  i |tvrtjtftjtftjtftjtftjt ftjtftj tftj tftj tftj t fd dtddddf fd }d}ttjjr|}d d d }t#|D]i\}}||vrT|rQt'|tr< tjg|||d ]|||j|t.jkr!|rt3||d  tj<tj vr,t3 tj tj< S)z; Get input attributes for the given span template. ) rragent agent_name max_tokensfrequency_penaltypresence_penalty temperaturetop_ptop_kr6r1r r/Nc`|vr$|\}}|t||r ||<dSdSdSdSr1)rK)r6r1 attribute data_typer~rs r3 _set_from_keyz,_get_input_attributes.._set_from_keysRg~~)0&I$E9)E)E$,1Jy)))~$$$$r5Tusersystem)prompt system_prompt)rolecontent)rr) _AI_TEMPLATESrGEN_AI_REQUEST_MODELrLGEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSrT GEN_AI_REQUEST_FREQUENCY_PENALTYrNGEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_KrrrrUlistrrK setdefaultGEN_AI_REQUEST_MESSAGESrrrrGEN_AI_TOOL_INPUT) rrrrrcollect_messagesrolesr6r1r~rs @@r3_get_input_attributesr'ws$&J=  3S9#8#>0#6#5s;#=sC"*"KU!S!)!I5 Q$?G3U;3S9    2s 25 2T 2 2 2 2 2 2 2  &z'<'>'>'F G G .- !H==v||~~.. & &JCe||# 5#(>(>))(*JBOOVV!&s>> M#u % % % %<'''N'1:V , ,2 2 8-. ':557@ x7 88 8 834 r5usagecidtddddffd }|tjd|tjd|tjd S) z Get usage attributes. rkeysztuple[str, ...]r/Ncj|D].}t|}|t|tr||</dSr1)rrKrT)rr*r6r1r~r(s r3_set_from_keysz-_get_usage_attributes.._set_from_keyssK . .Cuc**E Zs%;%; (- 9% . .r5) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)rLrGEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)r(r,r~s` @r3_get_usage_attributesr5sJ.#.->.4....... N*)N+.N* r5rrci}|tvrt5|g}t|d}||t|d}t|d}|||D]&}|"|t |'t|d}|$t |tr||tj <t|d}|$t |tr||tj <dddn #1swxYwY|tj kr|rt||tj <|S)z< Get output attributes for the given span template. r(metadataNrr)rrrrr$r5rKrLrGEN_AI_RESPONSE_MODELrrrGEN_AI_TOOL_OUTPUT) rrrr~usage_candidatesr(metausage_candidaters r3_get_output_attributesr=s $&J=  ( * * H H &x vw//E  # #E * * *fj11DtW--E  # #E * * *#3 N N".%%&;O&L&LMMM$FG44J%*Z*E*E%=G 89:#FL99J%*Z*E*E%=G 89:- H H H H H H H H H H H H H H H2<'''O'2;F2C2C 8./ sC3DD!Drrci}|tjkrtjdtj|i}na|tjkrtjdi}nB|tjkr2tjdtj|i}|j}|||tj <| t||||| |pidS)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 collect_inputs: Whether gen_ai inputs may be collected. :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) rrrGEN_AI_OPERATION_NAMErrrGEN_AI_TOOL_NAMEr]GEN_AI_TOOL_DESCRIPTIONr$r'r) rrrr}rrrr~ docstrings r3rrs&$&J<(((  *N  &  \) ) )  *F  \) ) )  *N  %t I  ;DJx7 8+HndFSSTTTZ%2&&&&&r5cR|t|||pidS)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 collect_outputs: Whether gen_ai outputs may be collected. :param result: The result of the wrapped function. N)rr=)rrrrs r3rr&s1 +HovNNTRTUUUUUr5rrc^tj}|j}|r|jnd}|r|jdnd}|%|#||krt jd|d|ddS|jdd}|r%||||t jd|d|ddSd S) 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) rrrxrwrOr\rrTrU)rrrx client_org_idbaggage_org_idrHs r3r#r#7s=  " $ $F"J)3=J%%M;BLW)--h777N !  & ^ + +  IZh I IyF I I I   u&,n&8&8#U''!  &=+@  "}'@ LxJXxxhuxxx   5 4r5rzMutableMapping[str, str]sentry_baggagec|td}t|}t |dkrdnd}||z|z|t<dS)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. rrrN)r\r!rrrN)rrKexisting_baggagestripped_existing_baggage separators r3rr[sj{{#6;; ' < <=M N N45599rI "I->    r5zOptional[Attributes]zsentry_sdk.ScopezFtuple[Optional[bool], Optional[float], Optional[float], Optional[str]]c tj}|}t|js |jdddfSd}|j|j}|t|j }t|j d}|r|t|}ni}|j d|j | d||dd| dd|d |jd}|jr||j |jd|}nC#t $rt#jdd Yd SwxYw|j|j}n |jd }t'|d st#jd|ddSt)|}|s'|rd} nd} t#jd|d| dd|dfS|} |jr.|d|jjzz}|st#jd|ddS||k} | rt#jd|d} nD| |kr!|| krt#jd|dd} nt#jd|d|dd} | ||| fS)z Decide whether a span should be sampled. Returns a tuple with: 1. the sampling decision (sampled, unsampled, deferred) 2. the effective sample rate 3. the sample rand 4. the reason for not sampling the span, if unsampled NrZr|r) rrrrr} descriptionrrrrr{)transaction_contextrz1[Tracing] traces_sampler raised; unsampling traceT)exc_info)FNNcallback_errorrYTracing)rz[Tracing] Discarding z because of invalid sample rate.)FNNrUzYtraces_sampler returned 0 or False, or is using a fallback sample rate that is 0 or Falseztraces_sample_rate is set to 0z because FrrUrz because backpressure)FrN backpressurez[Tracing] Starting rVzA because it's not included in the random sample (sampling rate = rG)rrget_active_propagation_contextr^rUrrrXrRrcallabler\rrrr$rrrrrNrTmonitordownsample_factor) r}r~rnrr'rItraces_sampler_definedsampling_contextrUreasonsample_rate_before_backpressureroutcomes r3_make_sampling_decisionr`nsg " $ $F>>@@ v~ . .D"14tCCK".)1>>@@ +,?,HII &fn&8&89I&J&JKK&?  !j))JJJ08"5"D nn[11##'!$..)EFF" $ $2@  "  6 Q  # #$7$O P P P 7:&.)9:;KLLKK 7 7 7 NC    766  7  - 9-K I > > >0UtUUUVVV// $$K 6 ! 6pFF5F DTDDFDDEEEc; 55'2# ~4q&.:::  4 LLLLL M M M33K'G$ 1411222 ,{ : :=== LLLLL M M M$GG L}}}oz}}}   $G Kg 55s$D;; EEcTtj}|jdd}|jdpid}||n|}|sdSdddddtfd }|D]}t |t tfr|||rd S.t |trkd |vsd |vrcd }d } d |vr||d |}d |vr>|pi}|d  D]\} } | |vs|| || sd} n |r| rd SdS) z:Determine if a span fits one of the rules in ignore_spans. ignore_spansNraFruler r1r/ct|tr9t|tr"t||SdS||kSr)rKr rLr[ fullmatch)rcr1s r3_matchesz!is_ignored_span.._matchessO dG $ $ %%% DNN511222uu}r5Tr}r~) rrrUr\r[rKrLr rr) r}r~ris_ignored_at_top_levelis_ignored_in_experiment_configrbrfrc name_matchesattributes_matchrr1s r3is_ignored_spanrks  " $ $F$n00FF'-~'9'9.'I'I'OR&T&T''# # .  , uuUt dS'N + + xd## tt d # # 4<4;O;OL# ~~'xV d;; t##'-2 (,\(:(@(@(B(B$Iu 22((z)4;;2,1( 3   0 tt 5r5)rr)r)r!rr r)r)FriN)rrr{rr/N)rrrrr/r()rrr/rn) rrrrnrrrrr/N)rrr/N)rrnr/r)rrrr r/N)rrr/r) rrr}rnr~r_rrr/r )NNTr1)rnrr/r)rrrUrYr/r)r(r r/r)v contextlibrrrrrr.rcollections.abcrrrrrrandomr urllib.parser r r ImportErrortypingr rsentry_sdk.constsr rrsentry_sdk.data_collectionrsentry_sdk.utilsrrrrrrrrrrrrrrtypesrr r!r"r#r$r%r&r'sentry_sdk._typesr(r)rrbase64_strippedr+r[r^rerhcontextmanagerrLrrrrrrrrrrrrrrrDEFAULTrrrrNrRrQrrr frozensetrrrrrr'r5r=rrr#rr`rksentry_sdk.tracesrrrrsentry_sdk.tracingr!rr rrBr5r3r}s-  333333332222222222''''''''! 8888888888"+                    -,,,,,****** RZ(((((W(((6!;" +(B +t + + + +2*D22222  %.2II I II I  I  II,I]IIIIX ( ( ( (% *%8M%%%%%P**    "  &TQ JTQ!TQ*TQ* TQ  TQTQTQTQn+ J+ ++++\+ J+ ++++\& &2&&&&R&&C&O&&&&$OPOPOPOPOPOPOPOPd~a~a~a~a~a~a~a~aB U#A U UPT U U U U) *)5:) ))))>     &* -1+3 }"} }+} }  }}}}B!-1CC C*C C CCCCN+/ ''1****$* ****>  $  3D       u3?(*.(  ' : 5 #     L0,2FG ,!?,D,,,,<(<< <  <  <<<<~8%(%;?%IN%%%%%P)' )'()')'  )'  )'  )' )' )')')')'XV V(VV  V  VVVV"!$7!D!!!!H '9< &x6 x6&x6 x6N x6x6x6x6v3#3+A3d3333n(''''''''((sA AA