]j/HSSKJr SSKrSSKJs Jr SSKJ r SSK J r SSK J r SSKJrJr SSKJr SSKJrJrJr SS KJrJr SS KJr SS KJrJr SS KJ r SS K!J"r" SSK#J$r$ \ "SS55r%\ "SS55r&\ "SS55r'SSjr("SS\ 5r)g)) annotationsN) dataclass)cached_property)Lock)CallableLiteral)context) ReadableSpanSpan SpanProcessor)SamplerTraceIdRatioBased)Self)ONE_SECOND_IN_NANOSECONDS LevelName)WrapperSpanProcessor)suppress_instrumentation) SpanLevelcL\rSrSr%SrS\S'S\S'S\S'\S S j5rS rg ) 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]endedr first_spancURRnSo!ULo3(d[R"SU4SX45S[R "5;d[R "U5(a[R"U5OS[R"U5S.-nSSU0-n[[R"U55eS=p2UR$)N)is not)z%(py0)s is not %(py3)s span_context)py0py3zassert %(py5)spy5) rr @pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _safereprAssertionError_format_explanationtrace_id)selfr @py_assert2 @py_assert1 @py_format4 @py_format6s \/home/mande/repo/quber/.venv/lib/python3.13/site-packages/logfire/sampling/_tail_sampling.pyr)TraceBuffer.trace_id!sx.. #''4'''''|''''''|''''|'''4'''''''$$$N)returnint) __name__ __module__ __qualname____firstlineno____doc____annotations__rr)__static_attributes__r2r1r/rrs- 76 %%r1rcp\rSrSr%SrS\S'S\S'S\S'S \S '\SS j5r\SS j5rS r g)TailSamplingSpanInfo(z`Argument passed to the [`SamplingOptions.tail`][logfire.sampling.SamplingOptions.tail] callback.r spancontext.Context | Noner zLiteral['start', 'end']eventrbuffercB[R"UR5$)zThe log level of the span.)r from_spanr?r*s r/levelTailSamplingSpanInfo.level>s""499--r1cURR=(d URR=(d SURRR=(d [ S5- [ - $)zRThe time in seconds between the start of the trace and the start/end of this span.rinf)r?end_time start_timerBrfloatrrEs r/durationTailSamplingSpanInfo.durationCsS YY   <499#7#7 <1AWAWAbAbArfklqfr s %& &r1r2N)r3r)r3rL) r5r6r7r8r9r:propertyrFrMr;r2r1r/r=r=(sXj $# #"  ..&&r1r=cn\rSrSr%SrSrS\S'SrS\S'\SS S S S .SS jj5r Sr g)SamplingOptionsLzOptions 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@)rUlevel_thresholdduration_thresholdbackground_ratec^^^[U[[45(aUOSnSTs=::a Us=::aS::d O [S5eSUUU4SjjnU"XS9$)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. rSrXzEInvalid sampling rates, must be 0.0 <= background_rate <= head <= 1.0cX>TbURT:agTbURT:agT$)NrS)rMrF) span_infor[rZrYs r/get_tail_sample_rate?SamplingOptions.level_or_duration..get_tail_sample_rates4!-)2D2DGY2Y*y//Q" "r1)rUrV)r^r=r3rL) isinstancerLr4 ValueError)clsrUrYrZr[head_sample_rater_s ``` r/level_or_duration!SamplingOptions.level_or_durationjsS$$.dUCL#A#A4sA*:AcAde e # #88r1r2) rUrTrYzLevelName | NonerZz float | Noner[rLr3r) r5r6r7r8r9rUr:rV classmethodrer;r2r1r/rQrQLs  D/<@D 8? !$,4+.!$ 99* 9 ) 9  9 99r1rQcVU[R-[R"U5:$N)rTRACE_ID_LIMITget_bound_for_rate)r)rates r/check_trace_id_ratiorms& (77 7;L;_;_`d;e eer1c^\rSrSrSrS S U4SjjjrS SU4SjjjrSU4SjjrSSjrSSjr SU4Sjjr SU4S jjr SSU4S jjjr S r U=r$)TailSamplingProcessoraIBuffers spans until a span in the trace meets the sampling criteria. The wrapped `processor` receives `on_start` calls immediately (most processors just set attributes, which is safe before the sampling decision). Only `on_end` is buffered. The optional `deferred_processor` has both `on_start` and `on_end` buffered until the sampling decision. Use this for processors like `PendingSpanProcessor` that create new spans in `on_start`. Advanced users can mark their own processors for deferral by setting the `tail_sampling_defer_on_start` attribute to `True` on the processor class. ch>[TU]U5 X lX0l0Ul[ 5Ulgri)super__init__r_deferred_processortracesrlock)r* processorr_rt __class__s r/rsTailSamplingProcessor.__init__s2 #$8!"4 /1 F r1c b>SnSnUR UR(aURRnURc[ //US9UR U'UR R U5nUbEURbURRX45 UR[XSU55nSSS5 [TU]5X5 Uc*URbURRX5 ggU(aURU5 gg!,(df  Nd=f)NF)rrrstart)rvr r)parentrrugetrtrappend check_spanr=rron_start push_buffer)r*r?parent_contextdroppedrBr)rxs r/rTailSamplingProcessor.on_starts YY||<<00;;&,7"Y],^DKK)2%..:--t.DE"oo.B4Y`bh.ijG* . >&&2''00F3    V $9Ys B0D  D.c P>SnSnUR UR(aURRnURR U5nUbaUR R U5 UR[USSU55nURcURRUS5 SSS5 Uc9[TU]1U5 URbURRU5 ggU(aURU5 gg!,(df  Nd=f)NFend)rvr r)rur}rr~rr=r|poprron_endrtr)r*r?rrBr)rxs r/rTailSamplingProcessor.on_ends YY||<<002%LL''-"oo.B4uV\.]^G{{* $7 > GN4 &&2''..t43    V $+Ys B'D D%cURU5n[URRU5=n(aUR UR5 U$)zIf the span meets the sampling criteria, drop the buffer and return True. Otherwise, return False. Must be called within self.lock since it modifies self.traces via drop_buffer. )r_rmrBr) drop_buffer)r*r^ sample_ratesampleds r/r TailSamplingProcessor.check_spansK // : *9+;+;+D+DkR R7 R   Y-- .r1c2URUR gri)rur))r*rBs r/r!TailSamplingProcessor.drop_buffers KK (r1c>URbZURHnURR"U6 M URHnURR U5 M URHn[ TU]U5 M gri)rtrrrrrr)r*rBrr?rxs r/r!TailSamplingProcessor.push_bufferso  " " .!>>''00':* ''..t4% LLD GN4 !r1c>[TU]5 URb.[5 URR5 SSS5 gg!,(df  g=fri)rrshutdownrtr)r*rxs r/rTailSamplingProcessor.shutdownsJ   " " .*+''002,+ /,+s A  Ac>[TU]U5nURb9[5 URRU5=(a UnSSS5 U$U$!,(df  U$=fri)rr force_flushrtr)r*timeout_millisresultrxs r/r!TailSamplingProcessor.force_flushs\$^4  " " .)+00<<^LWQW, v ,+ s %A A()rtr_rvruri)rwr r_z'Callable[[TailSamplingSpanInfo], float]rtzSpanProcessor | Noner3None)r?r rr@r3r)r?r r3r)r^r=r3bool)rBrr3r)r3r)i0u)rr4r3r)r5r6r7r8r9rsrrrrrrrr; __classcell__)rxs@r/rorosl 48  F1   &!%!%F%> ) !3r1ro)r)r4rlrLr3r)* __future__rbuiltinsr#_pytest.assertion.rewrite assertionrewriter! dataclassesr functoolsr threadingrtypingrr opentelemetryr opentelemetry.sdk.tracer r r opentelemetry.sdk.trace.samplingr rtyping_extensionsrlogfire._internal.constantsrr#logfire._internal.exporters.wrapperrlogfire._internal.utilsr logfire.typesrrr=rQrmror2r1r/rs"!%$!EEG"E<# %% %"  & &  &F =9=9 =9@f K0Kr1