tniddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZddlZddlZddlmZmZmZmZddlmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"ddl#m$Z$e$r0dd l#m%Z%dd l#m&Z&dd l#m'Z'dd l#m(Z(dd l#m)Z)ddl#m*Z*ddl#m+Z+ddl,m-Z-ej.dZ/dZ0GddeZ1ddde2fdZ3ej4 dzdddddddd d!e2d"e2d#e5dd$fd%Z6 d{d*Z7d|d-Z8d.e2d/d d0d1d2d1d3d d4d de2fd5Z9 d}d6Z:d~d7Z;d~d8Z< dd;Z=dddGd>d?Z?Gd@dAZ@dBdCdDe5de2fdEZAddHZBdddejCf ddPZDdddSZEdddTZFdUdVdWd dXdYdeGfdZZH dd_ZId`ddae5ddbfdcZJ ddNdddKe5dedMde5fdfZKdNddde5fdgZLdNdddhe2didjdedkddkf dlZMddnZNdNdddhe2dodddkfdpZOd(dqdNdddhe2dKe5drddidjdedkddfdsZPd(dqdNdddhe2dodddf dtZQdudvde2fdwZRddxlSmTZTmUZUmVZVe$rddylSmWZWdSdS)N)Mapping) timedelta)Random)quoteunquote)OPSPANDATA SPANSTATUS SPANTEMPLATE) capture_internal_exceptionsfilename_for_moduleloggermatch_regex_listqualname_from_function safe_repr to_string try_convert is_sentry_url_is_external_source_is_in_project_root_module_in_list) TYPE_CHECKING)Any)Dict) Generator)Optional)Union)Iterator)Tuple) FrameTypez5^[ ]*([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 U/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/sentry_sdk/tracing_utils.py__init__zEnvironHeaders.__init__Bs   key Optional[Any]cx|j|j|ddzSN-_)r$r%replaceupper)r)r-s r* __getitem__zEnvironHeaders.__getitem__Js0|DK#++c3*?*?*E*E*G*GGHHr,cNtdt|DS)Nc3K|]}dVdS)N).0r2s r* z)EnvironHeaders.__len__..Ns"))1))))))r,)sumiterr)s r*__len__zEnvironHeaders.__len__Ms%))d4jj))))))r,Generator[str, None, None]c# K|jD]y}t|ts|dd}||js[|t|jdVzdSr0)r$ isinstancestrr3r4 startswithr%len)r)ks r*__iter__zEnvironHeaders.__iter__Ps ( (Aa%%  #s##))++A<< ,, C $$&&' ' ' ' ' ( (r,)r#)r&r@) __name__ __module__ __qualname__rCr+r5intr?rGr9r,r*r"r"As$  IsIIIII***** ( ( ( ( ( (r,r"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)boolget)rLs r*has_tracing_enabledrT\si u  $%%U2 KK, - -T 9 9{{+,,D8   r,Fmanualcursorrquery params_list paramstyle Optional[str] executemanyrecord_cursor_repr span_originz.Generator[sentry_sdk.tracing.Span, None, None]c#TKtjjdddr|r|dgkrd}|dkrd}nd}d}t ||}i}|||d<|||d<|rd|d <|r|||d <t 5tj|d | dddn #1swxYwYtjtj || 5}| D]\} } | | | |VddddS#1swxYwYdS)N _experimentsrecord_sql_paramsFpyformatformatz db.paramsz db.paramstyleTzdb.executemanyz db.cursorrW)messagecategorydata)opnameorigin) sentry_sdk get_clientrLrS _format_sqlr add_breadcrumb start_spanrDBitemsset_data) rVrWrXrYr[r\r]respanrFvs r*record_sql_queriesrsms&~6::;NPUVV kdV33K  # #!J   & &E D'[ *_&!% #f0"[ $ & &NN!%'MMMMNNNNNNNNNNNNNNN   5     JJLL  DAq MM!Q     s$ B//B36B35DD!$D!scopesentry_sdk.Scoperqsentry_sdk.tracing.Spanc |jtjkr%||jdd|jdS|jtjkrd}|jtj }|r%d|cxkrdkrnnd}nd|cxkrdkrnnd}|r |d d |j| dS|d d |j dS|jd kr%|d d |j|jdSdS)Nredis)rctyperdreiiWerroriiwarninghttphttplib)ryrdrelevel)ryrdre subprocess)ryrdrcre) rfrDB_REDISrl description_tags HTTP_CLIENT_datarSr HTTP_STATUS_CODE)rtrqr~ status_codes r*"maybe_create_breadcrumbs_from_spanrsy w"+ $7W4:       BN " "jnnX%>??  "k((((S((((( ****s*****!  S  idj !       fytz R R R R R L  !$       ! r,framer c> |jjS#t$rYdSwxYwr()f_code co_filename Exception)rs r*_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| Sr()rrr)rrrrrrshould_be_includedshould_be_excludeds r*_should_be_includedrsi)NCC,X66/>;;# "  T , 7 7 R@R > > !!L0MM ! ! ! MMM !  $ MM(0%,2F G G G G G?< % $sGA AAB B*)B*C00 C?>C?6 G GGctj}|sdS|j|jdS|jdd}|sdS|j|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_msrr8 millisecondsrrrrqrrr rirj is_active timestampstart_timestamprLrSrr)rqclientshould_add_query_sourceduration threshold slow_querys r*add_query_sourcers  " $ $F      ~!5!=$n001I4PP "~ 44H""#A1EEII15555 AJ  ^N3~))*:;;~))*:;; r,ctj}|sdS|j|jdS|jdd}|sdS|j|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_msrr8rrrrrr)rqrshould_add_request_sourcerrrs r*add_http_request_sourcer=s  " $ $F      ~!5!= & 2 23OQU V V $~ 44H""#EqIIII15555 AJ  ^N3~))*:;;~))*:;; r,header+Optional[Dict[str, Union[str, bool, None]]]c|sdS|dr|dr |dd}t|}|sdS|\}}}d}|r#dt |d}|r#dt |d}|r|d k}|||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)rDendswithSENTRY_TRACE_REGEXmatchgroupsrbrK)rrrr sampled_strrs r*extract_sentrytrace_datar[s t FOOE$:$:"  $ $V , ,E t,1LLNN)HnkN7##C"$5$566C"))#nb*A*ABB,$+((  r,sqlcd} t|drI||}t|tr||jj}n#t$rd}YnwxYw|pt|S)Nmogrify) hasattrrrBbytesdecode connectionencodingrr)rVrreal_sqls r*rkrk|sH  69 % % G~~c**H(E** G#??6+<+EFF   %y~~%sAA A-,A-cJeZdZdZdZ d$ d%dZe d&dZede fdZ e j de ddfdZ ede fdZ e j de ddfdZ ed'dZ de fdZd(dZd)dZd*d Zde fd!Zd+d"Zd,d#ZdS)-PropagationContextzJ The PropagationContext represents the data of a trace in Sentry. ) _trace_id_span_idrrbaggageNrrZspan_idrrOptional[bool]dynamic_sampling_contextOptional[Dict[str, str]]rOptional[Baggage]r&c||_ ||_ ||_ ||_ ||_ ||t ||_dSdSdSr()rrrrrBaggage)r)rrrrrrs r*r+zPropagationContext.__init__sq"/ :, H- Q PN ?7C"#;<s r*rzPropagationContext.trace_ids&~ .!Z\\-DN~r,valuec||_dSr()rr)rs r*rzPropagationContext.trace_ids r,ch|js%tjjdd|_|jS)z+The span id of the currently executed span.rN)rrrrr>s r*rzPropagationContext.span_ids/} 2 JLL,RSS1DM}r,c||_dSr()rrs r*rzPropagationContext.span_ids  r,rMcN|Sr() get_baggagerr>s r*rz+PropagationContext.dynamic_sampling_contexts !!::<<s r*to_traceparentz!PropagationContext.to_traceparents-00$,000r,rc\|jt||_|jSr()rr!populate_from_propagation_contextr>s r*rzPropagationContext.get_baggages' < "DDTJJDL|r,Iterator[Tuple[str, str]]c#Kt|fV|}|r t|fVdSdS)zw Creates a generator which returns the propagation_context's ``sentry-trace`` and ``baggage`` headers. N)rrr serializer)r)rs r* iter_headerszPropagationContext.iter_headerssk'(;(;(=(=====""$$..00  /%w. . . . . . / /r, other_dictc~|D]'\}} t|||#t$rY$wxYwdS)zU Updates the PropagationContext with data from the given dictionary. N)rosetattrAttributeError)r)r r-rs r*rzPropagationContext.updatesg%**,,  JC c5))))!      s - ::cfd|j|j|j|j|jS)Nz\)rbrrrrrr>s r*__repr__zPropagationContext.__repr__s5mtt N M     L    r,c|jdStt|jjd}|d|cxkrdkrnndStt|jjd}t |j|\}} t|j||f}n1#t$r$tj d|jd|d YdSwxYw|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_randrr8 sample_rateintervalz5Could not backfill sample_rand, since parent_sampled=z and sample_rate=..6f) rrfloat sentry_itemsrS_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r)rrlowerr4s r*rz$PropagationContext._fill_sample_rands7& <  F!%)B)F)F})U)UVV  "qK';';';';!';';';';'; F"%)B)F)F})U)UVV )$*={KK u / PUWWWKK    L2H[22#.222    FF 7B3G3G !-000sB44*C"!C"cR|jdS|jjdS)zAConvenience method to get the sample_rand value from the baggage.Nr)rrrSr>s r* _sample_randzPropagationContext._sample_randIs( < 4|(,,];;;r,)NNNNNN)rrZrrZrrZrrrrrrr&N)rrr&r)r&rM)r&r)r&r)r rr&Nr&N)r&rZ)rHrIrJ__doc__ __slots__r+ classmethodrpropertyrCrsetterrrrrr rrrr r9r,r*rrs.I%)#'*.+/?C'+=!=!=( = ) = #= =%= ====>#,# ###[#:#X_cd_X ^ST^===X=11111 //// #    -H-H-H-H^<<<<<rC)rrrr)r*rrLs r*rz)Baggage.populate_from_propagation_contexts*, &((!! )<(( (.#6#? Z ;;} % % A*1-*@L ' ;;y ! ! 9&-i&8L #   B)/):)EL & ' B)/):)A X& ;;+ , , M*-g6J.K*L*LL '|%6@@@r, transactionsentry_sdk.tracing.Transactionctj}i}|st|S|jpi}|j|d<|jd|d<|dr |d|d<|dr |d|d<|jr*|jj |d<|jj r|jj |d<|j r|j tvr |j |d<|jt|j|d <|j|jrd nd |d <|jr+|jjr||jjt|dS)z Populate fresh baggage entry with sentry_items and make it immutable if this is the head SDK which originates traces. rrrr;r<r=r>r@NrtruefalsesampledFr*)rirjrrrLrr rSr?r=r>rgsourceLOW_QUALITY_TRANSACTION_SOURCESrrCrE_baggagerr)rr@rrrLs r*populate_from_transactionz!Baggage.populate_from_transactions&(()+ !! )<(( (.&B#.#7 Z )4)A&G&G ]# ;;} % % A*1-*@L ' ;;y ! ! 9&-i&8L #   B)/):)EL & ' B)/):)A X&   ;"*III*5*:L '  " .*-k.E*F*FL '   *0;0C&PffL #   CK$8$E C    4 A B B B|U3333r,cd|_dS)NFrFr>s r*freezezBaggage.freezes  r,cRi}|jD] \}}|||< |Sr()rro)r)rr-r3s r*rz Baggage.dynamic_sampling_contexts9*0022  ICF3KK r,Finclude_third_partycg}|jD]|\}}t5tjt |zdzt t |z}||dddn #1swxYwY}|r||jd |S)Nr/r.) rror r SENTRY_PREFIXrrCappendr)join)r)rNror-r4r3s r*r 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. r.c3|K|]7}tj|3|V8dSr()rr2rr1)r:r3s r*r;z/Baggage.strip_sentry_baggage..sU  288FF       r,)rRr0)rs r*strip_sentry_baggagezBaggage.strip_sentry_baggagesB xx  "LL--      r,Optional[float]ctt|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). rN?)rrrrS)r)rs r*r zBaggage._sample_randsV "%):)>)>})M)MNN  "sk'?'?'?'?C'?'?'?'?'? tr,cFd|dd|jdS)Nz )r r*r>s r*rzBaggage.__repr__s-`DNNtNDD``QUQ]````r,)rT)rrZr rZr&r)rtr6r&r)rrr&r)r@rAr&rr!)r&r+)F)r&rV)rHrIrJr"r#rPrecompiler2rCrRr+r$rr9rrJrLrr  staticmethodrUr rr9r,r*rrQsAIM$"*Z00 "$ &  )- A A A& A  A A A[ ADUUU[UA"6A AAA[A@/4:/4 /4/4/4[/4b  T c      S  S    \      a#aaaaaar,rrzsentry_sdk.client.BaseClienturlcd|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)rLrr)rr_ras r*should_propagate_tracerc s?!'/J KVS!!u C!:t T T TTr,rrci}|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#Nr2r1)rorDr3r)rrer-rs r*rr,sx D#))++ U >>' " " abb'Ckk#s##))++S Kr,rfOptional[Union[str, OP]]rg attributesOptional[dict[str, Any]]templater c0ddlmdfd }|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_piifrr&cvtjdfd } tj|_n#t $rYnwxYwtjdfd } tj|_n#t $rYnwxYwtjr|S|S) zD Decorator to create a span for the given function. argsrkwargsr&c Kt}|0tjdt  |i|d{VS pt } pt pd}t ||} }|||5}| pit||| || |i|d{V}t||||cdddS#1swxYwYdSNzbCannot create a child span for %s. Please start a Sentry transaction before calling this function.r)rfrg get_current_spanrrr _get_span_op_get_span_name start_child update_data_set_input_attributes_set_output_attributesrnro current_spanspan_op function_name span_namesend_piirqresultrgrlrgrfrkris r* async_wrapperzDcreate_span_decorator..span_decorator..async_wrapperXs+--L# V*1-- Q//////////2L22G C$:1$=$=CM&xGGI..00H))*   !1r222%(HmQf !q$1&11111111&tXxHHH                  sAC55C9<C9c t}|*tjdt  |i|S pt } pt pd}t ||} }|||5}| pit||| || |i|}t||||cdddS#1swxYwYdSrqrrrzs r* sync_wrapperzCcreate_span_decorator..span_decorator..sync_wrapper}s+--L# V*1-- q$)&)))2L22G C$:1$=$=CM&xGGI..00H))*   !1r222%(HmQfD+F++&tXxHHH                  sAC''C+.C+N)rnrrorr&r) functoolswrapsinspect signature __signature__riscoroutinefunction)rlrrrgrgrfrkris` r*span_decoratorz-create_span_decorator..span_decoratorSs7                > *1*;A*>*>M ' '    D                 > )0):1)=)=L & &    D   &q ) )   s#A AA9B B B )rlrr&r)sentry_sdk.scoperk)rfrgrgrirrks```` @r*create_span_decoratorr;s`,988888R R R R R R R R R R h r,Optional[sentry_sdk.Scope]Optional[Span]c>|ptj}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )riget_current_scoperq)rtr{s r*rsrss$  3Z133E:L r,c|p t}|L|tj|j(|jtjdSdSdS)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. N)rs set_statusr INTERNAL_ERRORcontaining_transaction)rqs r*set_span_erroredrsg  %#%%D   1222  & 2  ' 2 2:3L M M M M M 2 2r,rXrYrrrtuple[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)rrrK randrange)rrrr4rng lower_scaled upper_scaledsample_rand_scaleds r*rrsLE5 5==JKKK   Cuy())Luy())L* ]]<FF *****  ))sA&& A54A5rrrrVc&||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. NrTrXrYr9)rrs r*rrs5!4x 4  KCr,rGr-r.cd}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)rBdictrSrgetattrr)rGr-rs r* _get_valuers} E&$ 3 63    ,,     LsA AAzUnion[str, SPANTEMPLATE]roc|}|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 )r AI_CHATrSrBrCAI_AGENTAI_TOOL)rirgror~rr-s r*rurusI<'''  .  ::c??z&+s'C'C"3KEE',8OEOOO& \* * **D** \) ) )*D** r,ctjtjtjtjtjtji}||tj }t|S)z> Get the operation of the span based on the template. ) r rr GEN_AI_CHATrGEN_AI_INVOKE_AGENTrGEN_AI_EXECUTE_TOOLrSFUNCTIONrC)rimappingrfs r*rtrtsN bnr5b4AG Xr{ + +B r77Nr,rrnztuple[Any, ...]dict[str, Any]c pi|tjtjtjfvrtjt ftjt ftjt ftjt ftjtftj tftj tftj tftjtftjtfd dt ddddffd }t!|D]\}}|dkrLt%|t r7tjgd |d W|d krLt%|t r7tjgd |d ||||tjkr!|rt-||d tj<tjvr,t-tjtj<S)z; Get input attributes for the given span template. ) rragent agent_name max_tokensfrequency_penaltypresence_penalty temperaturetop_ptop_kr-rrr&Nc`|vr$|\}}|t||r ||<dSdSdSdSr()rB)r-r attribute data_typergrs r* _set_from_keyz,_get_input_attributes.._set_from_keyCsRg~~)0&I$E9)E)E$,1Jy)))~$$$$r,promptuser)rolecontent system_promptsystem)rnro)r rrrr GEN_AI_REQUEST_MODELrCGEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSrK GEN_AI_REQUEST_FREQUENCY_PENALTYrGEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_KlistrorB setdefaultGEN_AI_REQUEST_MESSAGESrQrGEN_AI_TOOL_INPUT) rirrnrorr-rrgrs @@r*_get_input_attributesr*s($&JL)<+?AUVVV3S9#8#>0#6#5s;#=sC"*"KU!S!)!I5 Q$?G3U;3S9    2s 25 2T 2 2 2 2 2 2 2 v||~~.. & &JCh:eS#9#9%%h&FKKRR#66o%%*UC*@*@%%%h&FKKRR%%88 M#u % % % %<'''H'1:V , ,2 2 8-. ':557@ x7 88 8 834 r,usagecidtddddffd }|tjd|tjd|tjd S) z Get usage attributes. rkeysztuple[str, ...]r&Ncj|D].}t|}|t|tr||</dSr()rrBrK)rrr-rrgrs r*_set_from_keysz-_get_usage_attributes.._set_from_keyslsK . .Cuc**E Zs%;%; (- 9% . .r,) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)rCr GEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)rrrgs` @r*_get_usage_attributesrfsJ.#.->.4....... N*)N+.N* r,rci}|tjtjtjfvrt 5|g}t |d}||t |d}t |d}|||D]&}|"|t|'t |d}|$t|tr||tj <t |d}|$t|tr||tj <dddn #1swxYwY|tjkr|rt||tj<|S)z< Get output attributes for the given span template. rmetadataNrr)r rrrr rrQrrrBrCr GEN_AI_RESPONSE_MODELrGEN_AI_TOOL_OUTPUT) rirrrgusage_candidatesrmetausage_candidaters r*_get_output_attributesrs $&JL)<+?AUVVV ( * * 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<'''H'2;F2C2C 8./ sC3D66D:=D:Spanrlci}|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 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) r rr GEN_AI_OPERATION_NAMErrrGEN_AI_TOOL_NAMEr"GEN_AI_TOOL_DESCRIPTIONrrrw) rqrirrgrlrnrorg docstrings r*rxrxs&$&J<(((  *N  &  \) ) )  *F  \) ) )  *N  %t I  ;DJx7 8+HhfMMNNNZ%2&&&&&r,cR|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 send_pii: Whether to send PII data. :param result: The result of the wrapped function. N)rwr)rqrirrs r*ryrys0 +HhGGM2NNNNNr,rrc^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 Nr>zKStarting 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) rirjr?r>rrSrrrL)rrr? client_org_idbaggage_org_idrs r*rrs=  " $ $F"J)3=J%%M;BLW)--h777N !  & ^ + +  IZh I IyF I I I   u&,n&8&8#U''!  &=+@  "}'@ LxJXxxhuxxx   5 4r,)rrHr)r)FrU)rtrurqrvr&N)rr r&rZ) rqrvrrZrrrrr&N)rqrvr&N)rrZr&r)rrr&r) rfrfrgrZrgrhrir r&rr()rtrr&r)rqrr&N)rrrrVr&r)rrr&r)X contextlibrrrr\rcollections.abcrdatetimerrandomr urllib.parserrrrisentry_sdk.constsrr r r sentry_sdk.utilsr r rrrrrrrrrrtypingrrrrrrrrtypesr r]rbase64_strippedr"rRrTcontextmanagerrCrsrrrrrrrrkrrrcrDEFAULTrrsrrrrrrurtrrrrxryrsentry_sdk.tracingrrHrrr9r,r*rsC ######'''''''' DDDDDDDDDDDD                            !          RZ(((((W(((6!;"  %** * ** *  *  **6****Z  %>      B**    "  &FH #FH!FH*FH* FH  FHFHFHFHR<< 2B&&C&O&&&&$@<@<@<@<@<@<@<@  $  3D       u3?(*.(  ' : 5 #    9(99 9  9  9999x8%(%48%BG%%%%%P)' )'()')'  )'  )'  )' )' )')')')'X O  O6 OBF OPU O  O O O O!$7!D!!!!J (''''''''((r,