jddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZmZddlZ ddlmZddlZddlmZmZmZmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd lm)Z)e)r6dd lm*Z*dd lm+Z+dd lm,Z,dd lm-Z-ddlm.Z.ddlm/Z/ddlm0Z0ddl1m2Z2ddl3m4Z4ejjdZ6dZ7GddeZ8ddde9fdZ:ddde9fdZ;ejx duddddd dd!d"d#e9d$e9d%e=dd&fd'Z> dvd+Z?dwd,Z@d-e9d.d"d/d0d1d0d2d"d3d"de9fd4ZA dxd5ZBdyd6ZCdyd7ZD dzd8ZEddd9e=dd"fd:ZFGd;d<ZGGd=d>ZHd?d@dAe=de9fdBZId{dCZJdddejf d|dGZL d}dDd"dEddHe9ddfdIZM d~ ddJZNd~ddKZOdLdMdNd"dOdPdePfdQZQ ddRZRdSddTe=ddUfdVZS d~dFdWdDe=dXdde=fdYZTdFdWde=fdZZUdFdWd[e9d\d]dXd^dd^f d_ZVdd`ZWdFdWd[e9daddd^fdbZXd*dcdFdWd[e9dDe=dddd\d]dXd^ddfdeZYd*dcdFdWd[e9dadddf dfZZdgdhde9fdiZ[djdkdle=ddfdmZ\dDe=dEdnd(d)ddofdpZ]dDe=dEdnde9fdqZ^ddrl_m`Z`maZambZbmcZcddsldmeZemfZfmgZhe)rddtl_mcZcyy#e$r ddlmZY2wxYw)N)MappingMutableMapping) timedelta)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) 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 q/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/sentry_sdk/tracing_utils.py__init__zEnvironHeaders.__init__Ls   key Optional[Any]cv|j|j|jddjzSN-_)r'r(replaceupper)r,r0s r- __getitem__zEnvironHeaders.__getitem__Ts.||DKK#++c3*?*E*E*GGHHr/c8tdt|DS)Nc3 K|]}dyw)N).0r5s r- z)EnvironHeaders.__len__..Xs)j1js )sumiterr,s r-__len__zEnvironHeaders.__len__Ws)d4j)))r/c#K|jD]i}t|ts|jddj }|j |j sP|t|j dkywr3)r' isinstancestrr6r7 startswithr(len)r,ks r-__iter__zEnvironHeaders.__iter__ZsdAa% #s#))+A<< ,C $&' 'sA:A<)HTTP_)r)zGenerator[str, None, None]) __name__ __module__ __qualname__rEr.r8intrBrIr<r/r-r&r&KsO$  IsII** (r/r&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)boolgetrOs r-has_tracing_enabledrXfsY   $%U2 KK, -T 9 9{{+,D8  r/Optional[dict[str, Any]]cV|y|jdxsijddk(S)NF _experimentstrace_lifecyclestream)rVrWs r-has_span_streaming_enabledr^ws/ KK ' -2 2 23D E QQr/cursorrquery 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_paramsFpyformatformatz db.paramsz db.paramstyleTzdb.executemanyz db.cursorr`)messagecategorydata)opnameorigin) sentry_sdk get_clientrOrV _format_sqlradd_breadcrumb start_spanr DBitemsset_data) r_r`rarbrdrerfrmspanrHvs r-record_sql_queriesr{~s*&&~6::;NPUVkdV3K  #!J   &E D'[ *_!% f0"[ $ &!!%'M '    55   JJLDAq MM!Q !    ' &  s6A>DC?.D/D 6 D?DD DDscopesentry_sdk.Scoperyc\|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)rktyperlrmiiWerroriiwarninghttphttplib)rrlrmlevel)rrlrm subprocess)rrlrkrm) rnr DB_REDISrt description_tags HTTP_CLIENT_datarVr HTTP_STATUS_CODE)r|ryr status_codes r-"maybe_create_breadcrumbs_from_spanrs ww"++ $$7W4::   BNN "jjnnX%>%>? k(S( *s*!   idjj !   fytzz R L  !$$   !r/cN |jjS#t$rYywxYwr+)f_code co_filename Exception)frames r-_get_frame_module_abs_pathrs)||''' 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 r-_should_be_includedrs[)NC,X6/>;# "  T , 7 R@R !!LL00M  $ MM(00%,,2F2F G %=/ I 4 F  I " ! M !sGG> G7G)6G; GG G&%G&) G87G8; H H ctj}|jsy|j |jy|j j dd}|sy|j|jz }|j j dd}|tdz |kD}|syt||j d|j j d |j j d  y) zO Adds OTel compatible source code information to a database query span Nenable_db_query_sourceTdb_query_source_threshold_msrr; millisecondsrrrryrrr rqrr is_active timestampstart_timestamprOrVrr)ryclientshould_add_query_sourceduration threshold slow_querys r-add_query_sourcer0s " " $F     ~~!5!5!=$nn001I4P "~~ 4 44H""#A1EII155 AJ  ^^N3~~))*:;~~))*:; r/ctj}|jsy|j |jy|j j dd}|sy|j|jz }|j j dd}|tdz |kD}|syt||j d|j j d |j j d  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)ryrshould_add_request_sourcerrrs r-add_http_request_sourcerNs " " $F     ~~!5!5!= & 2 23OQU V $~~ 4 44H""#EqIII155 AJ  ^^N3~~))*:;~~))*:; r/c@|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)rFendswithSENTRY_TRACE_REGEXmatchgroupsrjrN)headerrrr sampled_strrs r-extract_sentrytrace_datarls  FOOE$:"  $ $V ,E ,1LLN)HnkN##C"$56"))#nb*AB$+(( r/sqlcd} t|drF|j|}t|tr%|j |j j }|xs t|S#t$rd}YwxYw)Nmogrify) hasattrrrDbytesdecode connectionencodingrr)r_rreal_sqls r-rsrssoH  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 r-r.zPropagationContext.__init__se"/ :, H- Q PN ?7C"#;|jjSr+) get_baggagerrAs r-rz+PropagationContext.dynamic_sampling_contexts!::<)rjrrrrrrAs r-__repr__zPropagationContext.__repr__*s<mtt NN MM       LL   r/c|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_itemsrV_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r,rrlowerr7s r-rz$PropagationContext._fill_sample_rand3s& <<  !%)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)rrrVrAs r- _sample_randzPropagationContext._sample_randbs* << ||((,,];;r/)NNNNNN)rrcrrcrrcrOptional[bool]rzOptional[Dict[str, str]]rOptional[Baggage]r)N)rDict[str, Any]r)r)r)rP)r)r)r)zIterator[Tuple[str, str]])rr'r)N)rdict[str, Any]r)Nr)N)r)rc)rKrLrM__doc__ __slots__r. classmethodrpropertyrErsetterrrrrr rrrrr$r<r/r-rrsvI%)#'*.+/?C'+H!H!H( H ) H #= H%H HB#,# ##:#__cd ^^ST==11 /?'7? ?  # -H^CC cn|j tiStj|jS)zC Deprecated: use populate_from_propagation_context )_propagation_contextrr )rr|s r- from_optionszBaggage.from_optionss1  % % -2; 889S9STTr/c 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_idrSr) rqrrrrrOrrV parsed_dsnrDrErE)rrrr1r2rrOs r-r z)Baggage.populate_from_propagation_contexts*, &&(!<( (..#6#?#? Z ;;} %*1-*@L ' ;;y !&-i&8L #   )/):):)E)EL &  '')/):):)A)A X& ;;+ ,*-g6J.K*LL '|%6@@r/cXtj}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. rrrrBrCrDrE transactionrtruefalsesampledFr2)rqrrrrrOrr$rVrFrDrErosourceLOW_QUALITY_TRANSACTION_SOURCESrrErL_baggagerr)rrIrrrOs r-populate_from_transactionz!Baggage.populate_from_transactions&&()+ !<( (..&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|U33r/c~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)rHrrrrBrCrDrEzsentry.span.sourcerIrrJrKrLFrM)rqrrrrrOrr$rVrFrDrEget_attributesLOW_QUALITY_SEGMENT_SOURCES_name _sample_raterErLrPrr)rsegmentrrrOs r-populate_from_segmentzBaggage.populate_from_segments &&()+ !<( (..&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 = = >|U33r/cd|_yNFrMrAs r-freezezBaggage.freeze3s  r/cXi}|jjD] \}}|||< |Sr+)rrw)r,rr0r;s r-rz Baggage.dynamic_sampling_context6s4**002ICF3K3 r/include_third_partycng}|jjD]]\}}t5tjt |zdzt t |z}|j|ddd_|r|j|jdj|S#1swYxYw)Nr7r6) rrwrr SENTRY_PREFIXrrEappendr1join)r,r]rwr0r<r;s r-r zBaggage.serialize>s))//1HC,.,,uSz9C?%C/Q T"/.2  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. r6c3zK|]3}tjj|js|5ywr+)rr:rr9)r=r;s r-r>z/Baggage.strip_sentry_baggage..Rs2 -D2288F-s9;)rar8)rs r-strip_sentry_baggagezBaggage.strip_sentry_baggageKs+ xx "LL-   r/cxtt|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). rN?)rrrrV)r,rs r-r$zBaggage._sample_randYsD "%):):)>)>})MN  "sk'?C'? (@r/cHd|jdd|jdS)Nz )r r2rAs r-rzBaggage.__repr__fs,DNNtNDE[QUQ]Q]P^^_``r/)rT)rrcr$rcr)r)r|zsentry_sdk.scope.Scoper)r&)rrr)r)rIzsentry_sdk.tracing.Transactionr)r)rW StreamedSpanr)rr))r)r3)F)r)Optional[float])rKrLrMr*r+r_recompiler:rErUr.r,rr@r rQrXr[rr  staticmethodrdr$rr<r/r-rrjsTAIM$"**Z0 "$ &  )- A A& A  A ADUUA"6A AA@/4:/4 /4/4b.4.4` T c   S  S     a#ar/rrzsentry_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)rOrr)rrorqs r-should_propagate_tracersjs1!'/J KVS! C!:t TTr/ci}|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. rJNr5r4)rwrFr6r")rrmr0rs r-rrvs` D#))+ U >>' "ab'Ckk#s#))+S , Kr/ro attributestemplatec0ddlmdfd }|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)rnroget_current_spanrr!rrDrjwarningswarnDeprecationWarning _get_span_op_get_span_name start_child update_data_set_input_attributes_set_output_attributesargskwargs current_spanspan_op function_name span_namesend_piiryresultrvfrornryrws r- async_wrapperzDcreate_span_decorator..span_decorator..async_wrappersE+-L# V*1- ////, 5 X&  ////2L2G C$:1$=CM&xGI.0H))*  !1r2%(HmQf !$1&11&tXxH00 2sL7D&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#1swYyxYwr}rrs r- sync_wrapperzCcreate_span_decorator..span_decorator..sync_wrappers"+-L# V*1- $)&)), 5 X&  $)&))2L2G C$:1$=CM&xGI.0H))*  !1r2%(HmQfD+F+&tXxHs 2>C::Drrrrr)r functoolswrapsinspect signature __signature__riscoroutinefunction)rrrrvrornryrws` r-span_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.scopery)rnrorvrwrrys```` @r-create_span_decoratorrs,9b b H r/activecdfd }|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)r{c"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.r~rr)rorvrrqrrr^rOrrrstart_streaming_span) rrrrrrrvrros r-rzNcreate_streaming_span_decorator..span_decorator..async_wrappers**,F-fnn= 1  ? 6q 9?RI%:f !$1&11 2s0A"B% B1B2B7 BBB Bctj}t|jst j ddxst xsd}t|5|i|cdddS#1swYyxYwrr)rrrrrrvrros r-rzMcreate_streaming_span_decorator..span_decorator..sync_wrapper)sv**,F-fnn= 1  ? 6q 9?RI%:f$)&)s #A55A>rr)rrrrrvros` r-rz7create_streaming_span_decorator..span_decorator s     " *1*;*;A*>M '   *  *  )0):):1)=L &  & &q )  9   ,   s#B!B# B B # B/.B/rr<)rorvrrs``` r-create_streaming_span_decoratorrs5 n r/cN|xstj}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )rqget_current_scopery)r|rs r-rrGs&  3Z113E::L r/cddlm}m}|xs t}|t |t rV|j tj|j*|jj tjyyt ||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)rj SpanStatusN) sentry_sdk.tracesrjrrrDSpan set_statusr INTERNAL_ERRORcontaining_transactionERRORstatus_segment)ryrjrs r-set_span_erroredrRs ;  %#%D  dD ! OOJ55 6**6++66z7P7PQ7 l +$**DK}}('1'7'7 $), r/rfrgrrrtuple[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)r rrN randrange)rrr"r7rng lower_scaled upper_scaledsample_rand_scaleds r-rrfsLE5 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. rTrfrgr<)rrs r-rrs0!4 4 KCr/rNr0r1cd}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)rDdictrVrgetattrr)rNr0rs r- _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 )r AI_CHATrVrDrEAI_AGENTAI_TOOL)rwrorrrr0s r-rrsI<''' .::c?z&+s'C"3KE/ (-eE7O  39   \** *#D6*  \)) )#D6* r/ctjtjtjtj tj tji}|j|tj}t|S)z> Get the operation of the span based on the template. ) r rr GEN_AI_CHATrGEN_AI_INVOKE_AGENTrGEN_AI_EXECUTE_TOOLrVFUNCTIONrE)rwmappingrns r-rrs\ bnnr55b44AG Xr{{ +B r7Nr/rrztuple[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_kr0rrr)NcJ|vr|\}}|t||r||<yyyyr+)rD)r0r attribute data_typervrs r- _set_from_keyz,_get_input_attributes.._set_from_keys?g~)0&I$E9)E,1Jy)*F$r/promptuser)rolecontent system_promptsystem)rr)r rrrr GEN_AI_REQUEST_MODELrEGEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSrN GEN_AI_REQUEST_FREQUENCY_PENALTYrGEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_KlistrwrD setdefaultGEN_AI_REQUEST_MESSAGESr`rGEN_AI_TOOL_INPUT) rwrrrrr0rrvrs @@r-_get_input_attributesrs$&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 r/cidtddddffd }|tjd|tjd|tjd S) z Get usage attributes. rkeysztuple[str, ...]r)Nc^|D]'}t|}|t|ts#||<)yr+)rrDrN)rrr0rrvusages r-_set_from_keysz-_get_usage_attributes.._set_from_keyss2Cuc*E Zs%;(- 9%r/) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)rEr GEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)rrrvs` @r-_get_usage_attributesr soJ.#.->.4. **)++.** r/rci}|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) r rr GEN_AI_OPERATION_NAMErrrGEN_AI_TOOL_NAMEr*GEN_AI_TOOL_DESCRIPTIONrrr) ryrwrrorrrrv docstrings r-rrQs&$&J<(((  * *N  & &  \)) )  * *F  \)) )  * *N  % %t II  ;DJx77 8+HhfMNZ%2&r/cD|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 )ryrwrrs r-rr}s! +HhGM2Nr/rr&cttj}|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 NrEzKStarting 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) rqrrrFrErrVrr!rO)rrrF client_org_idbaggage_org_idrs r-rrs  " " $F""J)3J%%M;BW))--h7N !  & ^ + YZhYiiwyFxGGH I &,nn&8&8#U'!  &=+@  "}'@ LLIJXIYYghugvvwx  r/headerszMutableMapping[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. rrr6N)rVrrrdrG)rrexisting_baggagestripped_existing_baggage separators r-add_sentry_baggage_to_headersrsV{{#6; ' < <=M N459rI "I->  r/zOptional[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 55r/ctj}|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(SrZ)rDr rErU fullmatch)r.rs r-_matchesz!is_ignored_span.._matches"s7 dG $%%DNN5122u}r/Trorv) rqrrrOrVrUrDrEr rrw) rorvrr-r1r. name_matchesattributes_matchrrs r-is_ignored_spanr4s  " " $FNN&&~6<"AA.QL uUt dS'N +d# d #4<4;OL# #)rJ~'V d; t#(,\(:(@(@(B$Iu 2(z)4;,1( )C 0/2 r/)rrOrr)rTrjru)r)Fmanual)r|r}rysentry_sdk.tracing.Spanr)N)rr#r)rc) ryr6rrcrrrrr)N)ryr6r)N)rrcr)z+Optional[Dict[str, Union[str, bool, None]]])rr'r)r') rnzOptional[Union[str, OP]]rorcrvrYrwr r)r)NNTr+)r|zOptional[sentry_sdk.Scope]r)#Optional[Union[Span, StreamedSpan]])ryr7r)N)rr%rrkr)r)rrr)r()i contextlibrrrrlrrcollections.abcrrdatetimerrandomr urllib.parserrrr ImportErrortypingrqsentry_sdk.constsr r r r sentry_sdk.utilsrrrrrrrrrrrrrrrrrrr r!r"typesr#sentry_sdk._typesr$rmrbase64_strippedr&rUrXr^contextmanagerrEr{rrrrrrrrsrrrsrDEFAULTrrrrrrrrrrrrr rrrrr+r4sentry_sdk.tracingrrOrrrrTrjrurr<r/r-rGsA 3'  DD ! , RZZ(W(6!;"R(BRtR  %* * ** *  *  **6**Z  %>   B**    "  &FH #FH!FH*FH* FH  FHR<< 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*\  'I2sI(( I76I7