o ai@sddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZmZddlZzddlmZWneySddlmZYnwddlZddlmZmZmZmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd lm)Z)e)rdd lm*Z*dd lm+Z+dd lm,Z,dd lm-Z-ddlm.Z.ddlm/Z/ddlm0Z0ddl1m2Z2ddl3m4Z4e5dZ6dZ7GdddeZ8ddde9fddZ:ddde9fddZ;ej<  dd!d"d#d"d$d"d%d&d'e9d(e9d)e=dd*fd+d,Z> - . / 0  dd1d2Z?dd5d6Z@d7e9d8d&d9d:d;d:dd?ZA / 0 = & 9 : ; :  dd@dAZBddBdCZCddDdEZD F &  GddHdIZEd!d"dJe=dd&fdKdLZFGdMdNdNZGGdOdPdPZHdQdRdSe=de9fdTdUZIddXdYZJdddejKf [ \ & ]  ^ _  "dd`daZL   bdd\d&d]ddce9dd"fdddeZM d f  gddhdiZNdddjdkZOdldmdnd&dodpdePfdqdrZQ s t u v  pddwdxZRdyd"dze=dd{fd|d}ZS dd^d~d\e=ddde=fddZTd^d~de=fddZUd^d~de9ddddddf ddZVdddZWd^d~de9dd"ddfddZXd/dd^d~de9d\e=dd"ddddddfddZYd/dd^d~de9dd"ddf ddZZddde9fddZ[ddde=ddfddZ\d\e=d]dd-d.ddfddZ]d\e=d]dde9fddZ^ddl_m`Z`maZambZbmcZcddldmeZemfZfmgZhe)rmddl_mcZcdSdS)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})?c@sPeZdZ ddddeddfddZd edd fd d Zdefd dZdddZdS)EnvironHeadersHTTP_environzMapping[str, str]prefixreturnNcCs||_||_dSN)r&r')selfr&r'r+V/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/sentry_sdk/tracing_utils.py__init__Ls zEnvironHeaders.__init__key Optional[Any]cCs|j|j|ddSN-_)r&r'replaceupper)r*r.r+r+r, __getitem__TzEnvironHeaders.__getitem__cCstddt|DS)Ncss|]}dVqdS)Nr+).0r2r+r+r, Xsz)EnvironHeaders.__len__..)sumiterr*r+r+r,__len__WszEnvironHeaders.__len__Generator[str, None, None]ccsP|jD]!}t|ts q|dd}||jsq|t|jdVqdSr0)r& isinstancestrr3r4 startswithr'len)r*kr+r+r,__iter__Zs   zEnvironHeaders.__iter__)r%)r(r>) __name__ __module__ __qualname__r@r-r5intr=rDr+r+r+r,r$Ks r$optionsOptional[Dict[str, Any]]r(cCs:|durdSt|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)boolgetrIr+r+r,has_tracing_enabledfs rQOptional[dict[str, Any]]cCs$|durdS|dp iddkS)NF _experimentstrace_lifecyclestream)rOrPr+r+r,has_span_streaming_enabledwsrVFmanualcursorrquery params_list paramstyle Optional[str] executemanyrecord_cursor_repr span_originz.Generator[sentry_sdk.tracing.Span, None, None]c cs tjdddr|r|dgkrd}|dkrd}nd}d}t||}i}|dur/||d<|dur7||d<|r=d|d <|rG|durG||d <ttj|d |d Wdn1s]wYtjtj ||d }| D] \} } | | | qp|VWddS1swYdS)NrSrecord_sql_paramsFpyformatformatz db.paramsz db.paramstyleTzdb.executemanyz db.cursorrY)messagecategorydata)opnameorigin) sentry_sdk get_clientrIrO _format_sqlr add_breadcrumb start_spanr DBitemsset_data) rXrYrZr[r]r^r_respanrCvr+r+r,record_sql_queries~s@   "rsscopesentry_sdk.Scoperqsentry_sdk.tracing.SpancCs|jtjkr|j|jdd|jddS|jtjkrZd}|jt j }|rAd|kr.dkr3nnd}nd|kr=dkrAnnd}|rO|jd d |j|d dS|jd d |jd dS|jd krl|jd d |j|jddSdS)Nredis)rctyperdreiiWerroriiwarninghttphttplib)rxrdrelevel)rxrdre subprocess)rxrdrcre) rfr DB_REDISrl description_tags HTTP_CLIENT_datarOr HTTP_STATUS_CODE)rtrqr} status_coder+r+r,"maybe_create_breadcrumbs_from_spans2       rframer"cCs z|jjWStyYdSwr))f_code co_filename Exception)rr+r+r,_get_frame_module_abs_paths   ris_sentry_sdk_frame namespacein_app_includeOptional[list[str]]in_app_excludeabs_path project_rootcCs6t||}t|p t||}| o|pt||o| Sr))rrr)rrrrrrshould_be_includedshould_be_excludedr+r+r,_should_be_includeds rc Cst}|dur=t|}z|jd}Wn tyd}Ynw|duo(|d}t||||||d}|r6n |j}|dusd}|durz|j } Wn tySd} Ynw| dur`| t j |j z|jd}Wn tysd}Ynw|dur| t j |t|} | dur|durt|| } n|dur| |r| |dtj} n| } | t j| z|jj} Wn tyd} Ynw| dur| t j|jjdSdSdS)zB Adds OTel compatible source code information to the span NrEz sentry_sdk.)rrrrrr)sys _getframer f_globalsrOrrArf_backf_linenorpr CODE_LINENOCODE_NAMESPACErr3lstripossep CODE_FILEPATHrco_name CODE_FUNCTION) rqrrrrrrrrlinenofilepath in_app_path code_functionr+r+r, add_sourcesp         rcCt}|s dS|jdus|jdurdS|jdd}|s!dS|j|j}|jdd}|tdd|k}|s;dSt||jd|jd |jd d dS) zO Adds OTel compatible source code information to a database query span Nenable_db_query_sourceTdb_query_source_threshold_msrr7 millisecondsrrrrqrrr rirj is_active timestampstart_timestamprIrOrr)rqclientshould_add_query_sourceduration threshold slow_queryr+r+r,add_query_source0&    rcCr) z] Adds OTel compatible source code information to a span for an outgoing HTTP request Nenable_http_request_sourceT http_request_source_threshold_msrr7rrrrrr)rqrshould_add_request_sourcerrrr+r+r,add_http_request_sourceNrrheader+Optional[Dict[str, Union[str, bool, None]]]cCs|sdS|dr|dr|dd}t|}|sdS|\}}}d}|r0dt|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)rAendswithSENTRY_TRACE_REGEXmatchgroupsrbrH)rrrr sampled_strrr+r+r,extract_sentrytrace_datals&  rsqlcCsXd}zt|dr||}t|tr||jj}Wn ty%d}Ynw|p+t|S)Nmogrify) hasattrrr?bytesdecode connectionencodingrr)rXrreal_sqlr+r+r,rks     rkc@seZdZdZdZ      d4        d5ddZe    d6ddZede fddZ e j de ddfddZ ede fddZ e j de ddfddZ ed7ddZ de fdd Zd8d"d#Zd9d%d&Zd:d(d)Z * +  d;d,d-Zde fd.d/ZdPropagationContextzJ The PropagationContext represents the data of a trace in Sentry. ) _trace_id_span_idrrbaggagecustom_sampling_contextNrr\span_idrrOptional[bool]dynamic_sampling_contextOptional[Dict[str, str]]rOptional[Baggage]r(cCsN||_ ||_ ||_ ||_ ||_ |dur"|dur"t||_d|_dSr))rrrrrBaggager)r*rrrrrrr+r+r,r-s   zPropagationContext.__init__ incoming_dataDict[str, Any]cCstt}t|}|t}t|}|dur|S|t}|r"t|nd}t|s*|S| ||r4||_ | |Sr)) rnormalize_incoming_datarOSENTRY_TRACE_HEADER_NAMErBAGGAGE_HEADER_NAMErfrom_incoming_header_should_continue_traceupdater_fill_sample_rand)clsrpropagation_contextnormalized_datasentry_trace_headersentrytrace_databaggage_headerrr+r+r,from_incoming_datas    z%PropagationContext.from_incoming_datacCs|js tj|_|jS)z!The trace id of the Sentry trace.)ruuiduuid4hexr<r+r+r,rs zPropagationContext.trace_idvaluecC ||_dSr))rr*rr+r+r,r cCs |js tjdd|_|jS)z+The span id of the currently executed span.rN)rrrrr<r+r+r,rszPropagationContext.span_idcCrr))rrr+r+r,rrrJcCs |Sr)) get_baggagerr<r+r+r,rs z+PropagationContext.dynamic_sampling_contextcCs|jd|jS)Nr1)rrr<r+r+r,to_traceparent sz!PropagationContext.to_traceparentrcCs|jdur t||_|jSr))rr!populate_from_propagation_contextr<r+r+r,r s  zPropagationContext.get_baggageIterator[Tuple[str, str]]ccs2t|fV|}|rt|fVdSdS)zw Creates a generator which returns the propagation_context's ``sentry-trace`` and ``baggage`` headers. N)rrr serializer)r*rr+r+r, iter_headerss  zPropagationContext.iter_headers other_dictc Cs8|D]\}}zt|||WqtyYqwdS)zU Updates the PropagationContext with data from the given dictionary. N)rosetattrAttributeError)r*rr.rr+r+r,rs zPropagationContext.updaterdict[str, Any]cCrr))r)r*rr+r+r,_set_custom_sampling_context%s z/PropagationContext._set_custom_sampling_contextcCsd|j|j|j|j|jS)Nz\)rbrrrrrr<r+r+r,__repr__*szPropagationContext.__repr__c Cs|jdurdStt|jjd}|dur!d|kr dkr!dStt|jjd}t|j|\}}z t|j||fd}Wnt yUt d|jd|d YdSw|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_randrr7 sample_rateintervalz5Could not backfill sample_rand, since parent_sampled=z and sample_rate=..6f) rrfloat sentry_itemsrO_sample_rand_ranger_generate_sample_randr ValueErrorrdebug)r*rrlowerr4r+r+r,r3s&    z$PropagationContext._fill_sample_randcCs|jdurdS|jjdS)zAConvenience method to get the sample_rand value from the baggage.Nr)rr rOr<r+r+r, _sample_randbs zPropagationContext._sample_rand)NNNNNN)rr\rr\rr\rrrrrrr(N)rrr(r)r(rJ)r(r)r(r)rrr(N)rrr(Nr(N)r(r\)rErFrG__doc__ __slots__r- classmethodrpropertyr@rsetterrrrrrrrrrrr+r+r+r,rsh  !        /rc@seZdZdZdZdZedZ  d5ddd e d e fd d Z e d d     d6ddZ e d7ddZe    d8ddZe    d9d d!Ze d:d$d%Zd;d&d'Zdd1d2Zde fd3d4Zd S)?raF 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. r third_party_itemsmutablezsentry-z^sentry-rTr Dict[str, str]rrcCs||_||_||_dSr)r)r*r rrr+r+r,r-xs zBaggage.__init__N)rrr\rr(c Csi}d}d}|rU|dD]G}d|vrq t5|}|d\}}tj|r;t|dd} t||| <d}n ||r@dnd|7}Wdn1sOwYq |durat||d <d}t|||S) zF freeze if incoming header already has sentry baggage rT,=r1r7FNr)splitr striprSENTRY_PREFIX_REGEXrrr@) rrrr rritemr.val baggage_keyr+r+r,rs,     zBaggage.from_incoming_headerrtsentry_sdk.scope.ScopercCs|jdur tiSt|jS)zC Deprecated: use populate_from_propagation_context N)_propagation_contextrr)rrtr+r+r, from_optionss  zBaggage.from_optionsrrcCsi}d}d}t}|st|S|j}|j|d<|dr%|d|d<|dr0|d|d<|jrC|jj|d<|jj rC|jj |d<|drPt |d|d <t|||S) NrFr environmentrelease public_keyorg_idrLr) rirjrrrIrrO parsed_dsnr(r)r@)rrr rrrrIr+r+r,rs&         z)Baggage.populate_from_propagation_context transactionsentry_sdk.tracing.TransactioncCst}i}|st|S|jpi}|j|d<|jd|d<|dr*|d|d<|dr5|d|d<|jrH|jj |d<|jj rH|jj |d<|j rU|j t vrU|j |d<|jd urat|j|d <|jd uro|jrkd nd |d <|jr}|jjr}||jjt|ddS) 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)rirjrrrIrrrOr*r(r)rgsourceLOW_QUALITY_TRANSACTION_SOURCESrr@r0_baggager r)rr+rr rIr+r+r,populate_from_transactions4             z!Baggage.populate_from_transactionsegment StreamedSpancCst}i}|st|S|jpi}|j|d<|jd|d<|dr*|d|d<|dr5|d|d<|jrH|jj |d<|jj rH|jj |d<| dt vrY|j rY|j |d <|jd uret|j|d <|jd urs|jrod nd |d<|jr|jjr||jjt|ddS)r-rrrr&r'r(r)zsentry.span.sourcer+Nrr.r/r0Fr1)rirjrrrIrrrOr*r(r)get_attributesLOW_QUALITY_SEGMENT_SOURCES_name _sample_rater@r0r4r r)rr6rr rIr+r+r,populate_from_segments8             zBaggage.populate_from_segmentcCs d|_dSNFr1r<r+r+r,freeze3s zBaggage.freezecCs$i}|jD]\}}|||<q|Sr))r ro)r*rr.r r+r+r,r6s z Baggage.dynamic_sampling_contextFinclude_third_partyc Csg}|jD]+\}}ttjt|dtt|}||Wdn1s-wYq|r;||jd |S)Nrr) r ror r SENTRY_PREFIXrr@appendrjoin)r*r?ror.r!r r+r+r,r>s   zBaggage.serializecCsddd|dDS)zRemove Sentry baggage from the given header. Given a Baggage header, return a new Baggage header with all Sentry baggage items removed. rcss$|] }tj|s|VqdSr))rrrr)r8r r+r+r,r9Rs z/Baggage.strip_sentry_baggage..)rBr)rr+r+r,strip_sentry_baggageKs zBaggage.strip_sentry_baggageOptional[float]cCs:tt|jd}|durd|krdkr|SdSdS)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?)rr r rO)r*rr+r+r,rYs zBaggage._sample_randcCsd|jddd|jdS)Nz )rrr<r+r+r,rfr6zBaggage.__repr__)rT)rr\rr\r(r)rtr#r(r)rrr(r)r+r,r(r)r6r7r(rr)r(r)F)r(rD)rErFrGrrr@recompilerr@rNr-rrr%rr5r<r>rr staticmethodrCrrr+r+r+r,rjs\    "     1  0    rrzsentry_sdk.client.BaseClienturlcCs&|jd}t||r dSt||ddS)zy Returns True if url matches trace_propagation_targets configured in the given client. Otherwise, returns False. trace_propagation_targetsFT)substring_matching)rIrr)rrKrLr+r+r,should_propagate_tracejs  rNrrcCsHi}|D]\}}|dr|dd}|dd}|||<q|S)z Normalizes incoming data so the keys are all lowercase with dashes instead of underscores and stripped from known prefixes. r%Nr2r1)rorAr3r)rrer.rr+r+r,rvs   rrfOptional[Union[str, OP]]rg attributestemplater cs&ddlmd fdd }|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(cstd fdd }zt|_Wn ty"Ynwtd fdd }zt|_Wn tyEYnwtrM|S|S) D Decorator to create a span for the given function. argsrkwargsr(c s t}|durtdt|i|IdHSt|tr1tjdtdd|i|IdHSp6t }p>tp>d}t ||}}|j ||d,}| pUit ||||||i|IdH}t||||WdS1s~wYdSNzbCannot 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)rfrgget_current_spanrrrr?r7warningswarnDeprecationWarning _get_span_op_get_span_name start_child update_data_set_input_attributes_set_output_attributes rVrW current_spanspan_op function_name span_namesend_piirqresultrQrTrgrfrSrRr+r, async_wrappers@   $zDcreate_span_decorator..span_decorator..async_wrapperc st}|durtdt|i|St|tr*tjdtdd|i|Sp/t }p7tp7d}t ||}}|j ||d)}| pNit ||||||i|}t||||WdS1stwYdSrXr\rgrnr+r, sync_wrappers>   $zCcreate_span_decorator..span_decorator..sync_wrapperNrVrrWrr(r functoolswrapsinspect signature __signature__riscoroutinefunctionrTrorprQrgrfrSrRrTr,span_decorators"' '  z-create_span_decorator..span_decoratorNrTrr(r)sentry_sdk.scoperS)rfrgrQrRr|r+rzr,create_span_decorators drTactivecsdfdd }|S)zW Create a span creating decorator that can wrap both sync and async functions. rTrr(cstd fdd }zt|_Wn ty Ynwtd fdd }zt|_Wn tyAYnwtrI|S|S) rUrVrrWr(cs|t}t|jstjdddptpd}t|d|i|IdH}|WdS1s7wYdSNzSUsing span streaming API in non-span-streaming mode. Use @sentry_sdk.trace instead.rYrZr)rgrQrrirjrVrIr^r_rstart_streaming_span)rVrWrrkrmrrQrTrgr+r,ros $zNcreate_streaming_span_decorator..span_decorator..async_wrappercspt}t|jstjdddptpd}t|d|i|WdS1s1wYdSrr)rVrWrrkrr+r,rp)s  $zMcreate_streaming_span_decorator..span_decorator..sync_wrapperNrqrrryrrQrgr{r,r| s"   z7create_streaming_span_decorator..span_decoratorNr}r+)rgrQrr|r+rr,create_streaming_span_decorators 7rOptional[sentry_sdk.Scope]#Optional[Union[Span, StreamedSpan]]cCs|pt}|j}|S)zU Returns the currently active span if there is one running, otherwise `None` )riget_current_scoperq)rtrhr+r+r,r]Gs r]cCsddlm}m}|p t}|durAt|tr,|tj|j dur*|j tjdSdSt||rC|j |_ |j durE|j |j _ dSdSdSdS)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)r7 SpanStatusN) sentry_sdk.tracesr7rr]r?Span set_statusr INTERNAL_ERRORcontaining_transactionERRORstatus_segment)rqr7rr+r+r,set_span_erroredRs      rrErFrrrtuple[float, float]cCsl|\}}||ks tdt|}t|d}t|d}z |||}W|dSty5|}Y|dSw)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 rrH randrange)rrrr4rng lower_scaled upper_scaledsample_rand_scaledr+r+r,r fs   r rrrrDcCs,|dus|dur 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. NrTrErFr+)rrr+r+r,r s r r2r.r/cCsTd}t|tr||}|St||r(zt||}W|Sty'd}Y|Sw|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)r?dictrOrgetattrr)r2r.rr+r+r, _get_values     rzUnion[str, SPANTEMPLATE]rWcCs|}|tjkr/d}|r"dD]}||r!t||tr!||}nq |r+d|}|Sd}|S|tjkr;d|}|S|tjkrEd|}|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_CHATrOr?r@AI_AGENTAI_TOOL)rRrgrWrkrr.r+r+r,rbs&   rbcCs2tjtjtjtjtjtji}||tj }t |S)z> Get the operation of the span based on the template. ) r rr GEN_AI_CHATrGEN_AI_INVOKE_AGENTrGEN_AI_EXECUTE_TOOLrOFUNCTIONr@)rRmappingrfr+r+r,ras rarlrVztuple[Any, ...]rc sJi|tjtjtjfvrtjtftjtftjtftjtftjt ftj t ftj t ftj t ftjt ftjt fd dtddddffdd }t|D]7\}}|d krft|trftjgd |d qK|d kr}t|tr}tjgd |d qK|||qK|tjkr|rt||dtj<tjvrttjtj<S)z; Get input attributes for the given span template. ) rragent agent_name max_tokensfrequency_penaltypresence_penalty temperaturetop_ptop_kr.rrr(Ncs>|vr|\}}|durt||r||<dSdSdSdSr))r?)r.r attribute data_typerQrr+r, _set_from_keys   z,_get_input_attributes.._set_from_keypromptuser)rolecontent system_promptsystem)rVrW)r rrrr GEN_AI_REQUEST_MODELr@GEN_AI_AGENT_NAMEGEN_AI_REQUEST_MAX_TOKENSrH GEN_AI_REQUEST_FREQUENCY_PENALTYr GEN_AI_REQUEST_PRESENCE_PENALTYGEN_AI_REQUEST_TEMPERATUREGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TOP_Klistror? setdefaultGEN_AI_REQUEST_MESSAGESrArGEN_AI_TOOL_INPUT)rRrlrVrWrr.rr+rr,_get_input_attributessF     rusagecsHidtddddffdd }|tjd|tjd |tjd S) z Get usage attributes. rkeysztuple[str, ...]r(Ncs2|D]}t|}|durt|tr||<qdSr))rr?rH)rrr.rrQrr+r,_set_from_keyss  z-_get_usage_attributes.._set_from_keys) prompt_tokens input_tokens)completion_tokens output_tokens) total_tokens)r@r GEN_AI_USAGE_INPUT_TOKENSGEN_AI_USAGE_OUTPUT_TOKENSGEN_AI_USAGE_TOTAL_TOKENS)rrr+rr,_get_usage_attributes srrmc Csi}|tjtjtjfvrqtZ|g}t|d}||t|d}t|d}|||D] }|dur;|t|q.t|d}|durOt |t rO||t j <t|d}|durbt |t rb||t j <Wdn1slwY|tjkr|rt ||t j<|S)z< Get output attributes for the given span template. rmetadataNrr)r rrrr rrArrr?r@r GEN_AI_RESPONSE_MODELrGEN_AI_TOOL_OUTPUT) rRrlrmrQusage_candidatesrmetausage_candidaterr+r+r,_get_output_attributes)s0         rrrTc Csi}|tjkrtjdtj|i}n$|tjkrtjdi}n|tjkr4tjdtj|i}|j}|dur4||tj <| t ||||| |pCidS)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_NAMErGEN_AI_TOOL_DESCRIPTIONrrrd) rqrRrlrgrTrVrWrQ docstringr+r+r,reQs"    recCs|t|||p idS)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)rdr)rqrRrlrmr+r+r,rf}s rfrrcCst}|j}|r |jnd}|r|jdnd}|dur2|dur2||kr2td|d|ddS|jdd}|rY|durC|dusK|durY|durYtd|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)r rOrrrI)rrr* client_org_idbaggage_org_idrr+r+r,rs,rheaderszMutableMapping[str, str]sentry_baggagecCs>|td}t|}t|dkrdnd}||||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)rOrrrCrB)rrexisting_baggagestripped_existing_baggage separatorr+r+r,add_sentry_baggage_to_headerss   rzOptional[Attributes]z._matchesTrgrQ) rirjrIrOrNr?r@rrro) rgrQrrrr name_matchesattributes_matchrrr+r+r,is_ignored_spans8   r)rr3rr)r9r7rm)r)FrW)rtrurqrvr(N)rr"r(r\) rqrvrr\rrrrr(N)rqrvr(N)rr\r(r)rrr(r) rfrPrgr\rQrRrRr r(r)NNTr))rtrr(r)rqrr(N)rrrrDr(r)rrr(r)i contextlibrsrurrHrr^collections.abcrrdatetimerrandomr urllib.parserrrr ImportErrortypingrisentry_sdk.constsr r r r sentry_sdk.utilsr rrrrrrrrrrrrrrrrrrr r!typesr"sentry_sdk._typesr#rIrbase64_strippedr$rNrQrVcontextmanagerr@rsrrrrrrrrkrrrNrDEFAULTrrr]rr r r rrbrarrrrerfrrrrsentry_sdk.tracingrr3rrrr9r7rmrr+r+r+r,s   <             - !  I  !L   D     < ( , $  X.