g(>ddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZmZddlmZmZdd lmZdd lmZmZmZmZmZdd lmZeGd d ZeGddZeGddZ eGddZ!ddZ"GddeZ#y)) annotations) dataclass)cached_property)Lock)CallableLiteral)context) ReadableSpanSpan SpanProcessor)SamplerTraceIdRatioBased)Self)ATTRIBUTES_LOG_LEVEL_NUM_KEY LEVEL_NUMBERSNUMBER_TO_LEVELONE_SECOND_IN_NANOSECONDS LevelName)WrapperSpanProcessorc^eZdZUdZded< ed dZd dZdZddZ ddZ dd Z dd Z y ) SpanLevelaA convenience class for comparing span/log levels. Can be compared to log level names (strings) such as 'info' or 'error' using `<`, `>`, `<=`, or `>=`, so e.g. `level >= 'error'` is valid. Will raise an exception if compared to a non-string or an invalid level name. intnumberc@tj|jS)zIThe human-readable name of the level, or `None` if the number is invalid.)rgetrselfs T/var/www/openai/venv/lib/python3.12/site-packages/logfire/sampling/_tail_sampling.pynamezSpanLevel.name&s""4;;//ct|tr|j|k(St|tr|j|k(St|t r|j|jk(St SN) isinstancerrstrrrNotImplementedrothers r__eq__zSpanLevel.__eq__+sW eS !;;%' ' eS !99% % eY ';;%,,. .r c,t|jSr")hashrrs r__hash__zSpanLevel.__hash__4sDKK  r c.|jt|kSr"rrr&s r__lt__zSpanLevel.__lt__7{{]5111r c.|jt|kDSr"r-r&s r__gt__zSpanLevel.__gt__:r/r c.|jt|k\Sr"r-r&s r__ge__zSpanLevel.__ge__={{mE222r c.|jt|kSr"r-r&s r__le__zSpanLevel.__le__@r4r N)returnLevelName | None)r'object)r'r) __name__ __module__ __qualname____doc____annotations__propertyrr(r+r.r1r3r6r rrrsE K00!2233r rcJeZdZUdZded<ded<ed dZed dZy) TraceBufferzArguments of `SpanProcessor.on_start` and `SpanProcessor.on_end` for spans in a single trace. These are stored until either the trace is included by tail sampling or it's completed and discarded. z)list[tuple[Span, context.Context | None]]startedzlist[ReadableSpan]endedc&|jddS)Nr)rCrs r first_spanzTraceBuffer.first_spanNs||Aq!!r cN|jj}|J|jSr")rFr trace_id)r span_contexts rrHzTraceBuffer.trace_idRs).. '''$$$r N)r7r )r7r)r:r;r<r=r>rrFrHr@r rrBrBDs; 76 ""%%r rBcdeZdZUdZded< ded< ded< ded <ed d Zedd Zy )TailSamplingSpanInfoz`Argument passed to the [`SamplingOptions.tail`][logfire.sampling.SamplingOptions.tail] callback.r spancontext.Context | Noner zLiteral['start', 'end']eventrBbufferc|jjxsi}|jt}t |t s t d}t|S)zThe log level of the span.info)rL attributesrrr#rrr)rrRlevels rrSzTailSamplingSpanInfo.levelosFYY))/R ;<%%!&)Er c|jjxs|jjxsd|jjjxs t dz t z S)zRThe time in seconds between the start of the trace and the start/end of this span.rinf)rLend_time start_timerOrFfloatrrs rdurationzTailSamplingSpanInfo.durationxsS YY   <499#7#7 <1AWAWAbAbArfklqfr s %& &r N)r7r)r7rX)r:r;r<r=r>r?rSrYr@r rrKrKYsXj $# #"    &&r rKcbeZdZUdZdZded< dZded< eddd d d  d d Zy)SamplingOptionszOptions for [`logfire.configure(sampling=...)`][logfire.configure(sampling)]. See the [sampling guide](https://logfire.pydantic.dev/docs/guides/advanced/sampling/). ?float | SamplerheadNz.Callable[[TailSamplingSpanInfo], float] | Nonetailnoticeg@)r^level_thresholdduration_thresholdbackground_ratect|ttfr|nd}dcxkr|cxkrdkstdtddfd }|||S)aReturns a `SamplingOptions` instance that tail samples traces based on their log level and duration. If a trace has at least one span/log that has a log level greater than or equal to `level_threshold`, or if the duration of the whole trace is greater than `duration_threshold` seconds, then the whole trace will be included. Otherwise, the probability is `background_rate`. The `head` parameter is the same as in the `SamplingOptions` constructor. r\razEInvalid sampling rates, must be 0.0 <= background_rate <= head <= 1.0cP|jkDry|jk\ryS)Nr\)rYrS) span_infordrcrbs rget_tail_sample_ratez?SamplingOptions.level_or_duration..get_tail_sample_rates4!-)2D2DGY2Y*y//Q" "r )r^r_)rgrKr7rX)r#rXr ValueError)clsr^rbrcrdhead_sample_raterhs ``` rlevel_or_durationz!SamplingOptions.level_or_durations`$$.dUCL#A4sA*:AcAde eBde e ##788r ) r^r]rbr8rcz float | NonerdrXr7r) r:r;r<r=r^r>r_ classmethodrlr@r rr[r[s  D/<@D 8? !$,4+.!$ 99* 9 ) 9  9 99r r[cT|tjztj|kSr")rTRACE_ID_LIMITget_bound_for_rate)rHrates rcheck_trace_id_ratiorrs& (77 7;L;_;_`d;e eer cZeZdZdZdfd Zd d fd Zd fd Zd dZd dZd fd Z xZ S)TailSamplingProcessorzYPasses spans to the wrapped processor if any span in a trace meets the sampling criteria.c^t||||_i|_t |_yr")super__init__rhtracesrlock)r processorrh __class__s rrwzTailSamplingProcessor.__init__s, #$8! /1 F r c d}d}|j5|jr|jj}|jt gg|j |<|j j |}|:|jj||f|jt||d|}ddd|t|1||y|r|j|yy#1swY1xYw)NFstart)ryr rHparentrBrxrrCappend check_spanrKrvon_start push_buffer)rrLparent_contextdroppedrOrHr{s rrzTailSamplingProcessor.on_starts YY||<<00;;&,7B,?DKK)2%NN))4*@A"oo.B4Y`bh.ijG$ > G T> 2    V $)Ys BC!!C*c d}d}|j5|jr|jj}|jj |}|`|j j ||jt|dd|}|j|jj|dddd|t|1|y|r|j|yy#1swY0xYw)NFend)ryr rHrxrrDrrrKr~poprvon_endr)rrLrrOrHr{s rrzTailSamplingProcessor.on_ends YY||<<002%LL''-"oo.B4uV\.]^G{{* $7 > GN4    V $Ys B C!!C*c|j|}t|jj|x}r|j |j|S)zbIf the span meets the sampling criteria, drop the buffer and return True. Otherwise, return False.)rhrrrOrH drop_buffer)rrg sample_ratesampleds rrz TailSamplingProcessor.check_span sI// : *9+;+;+D+DkR R7 R   Y-- .r c2|j|j=yr")rxrH)rrOs rrz!TailSamplingProcessor.drop_buffers KK (r c|jD]}t|||jD]}t||yr")rCrvrrDr)rrOrCrLr{s rrz!TailSamplingProcessor.push_buffers8~~G G g &&LLD GN4 !r )rzr rhz'Callable[[TailSamplingSpanInfo], float]r7Noner")rLr rrMr7r)rLr r7r)rgrKr7bool)rOrBr7r) r:r;r<r=rwrrrrr __classcell__)r{s@rrtrts)c %6%.)!!r rtN)rHrrqrXr7r)$ __future__r dataclassesr functoolsr threadingrtypingrr opentelemetryr opentelemetry.sdk.tracer r r opentelemetry.sdk.trace.samplingr rtyping_extensionsrlogfire._internal.constantsrrrrr#logfire._internal.exporters.wrapperrrrBrKr[rrrtr@r rrs"!%$!EEG"E )3)3 )3X %% %( $&$& $&N =9=9 =9@f Q!0Q!r