gdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z mZmZddlmZddlZdd lmZe eefZd Zdd Zed d ddZeGddeZGddeZeGddeZy)aThis module provides a thin wrapper around the OpenTelemetry propagate API to allow the OpenTelemetry contexts (and therefore Logfire contexts) to be transferred between different code running in different threads, processes or even services. In general, you should not need to use this module since Logfire will automatically patch [`ThreadPoolExecutor`][concurrent.futures.ThreadPoolExecutor] and [`ProcessPoolExecutor`][concurrent.futures.ProcessPoolExecutor] to carry over the context. And existing plugins exist to propagate the context with [requests](https://pypi.org/project/opentelemetry-instrumentation-requests/) and [httpx](https://pypi.org/project/opentelemetry-instrumentation-httpx/). ) annotations)contextmanager) dataclass)AnyIteratorMapping)context propagatetrace)TextMapPropagatorN)warn_at_user_stacklevel) get_contextattach_contextContextCarrierc4i}tj||S)ajCreate a new empty carrier dict and inject context into it. Returns: A new dict with the context injected into it. Usage: ```py from logfire.propagate import get_context, attach_context logfire_context = get_context() ... # later on in another thread, process or service with attach_context(logfire_context): ... ``` You could also inject context into an existing mapping like headers with: ```py from logfire.propagate import get_context existing_headers = {'X-Foobar': 'baz'} existing_headers.update(get_context()) ... ``` )r injectcarriers F/var/www/openai/venv/lib/python3.12/site-packages/logfire/propagate.pyrrs<!G W NF) third_partyc#Ktj}tj}|s9t |t t fr#|j}t |t t fr#|j|} tj|dtj|y#tj|wxYww)aAttach a context as generated by [`get_context`][logfire.propagate.get_context] to the current execution context. Since `attach_context` is a context manager, it restores the previous context when exiting. Set `third_party` to `True` if using this inside a library intended to be used by others. This will respect the [`distributed_tracing` argument of `logfire.configure()`][logfire.configure(distributed_tracing)], so users will be warned about unintentional distributed tracing by default and they can suppress it. See [Unintentional Distributed Tracing](https://logfire.pydantic.dev/docs/how-to-guides/distributed-tracing/#unintentional-distributed-tracing) for more information. rN) otel_context get_currentr get_global_textmap isinstance#WarnOnExtractTraceContextPropagatorNoExtractTraceContextPropagatorwrappedextractattach)rr old_context propagator new_contexts rrrBs**,K--/J &IKj%kl#++J&IKj%kl$$W$5K)K( K( K(s$A#C&C9B(C(B??Cc<eZdZUdZded<ddZd dZedZy) WrapperPropagatorz-Helper base class to wrap another propagator.r rc:|jj|i|SN)rr selfargskwargss rr zWrapperPropagator.extractas#t||##T4V44rc:|jj|i|Sr()rrr)s rrzWrapperPropagator.injectds"t||""D3F33rc.|jjSr()rfields)r*s rr/zWrapperPropagator.fieldsgs||"""rN)r+rr,rreturnotel_context.Context)r+rr,r) __name__ __module__ __qualname____doc____annotations__r rpropertyr/rrr&r&[s)7 54##rr&c:eZdZdZ d dfd ZxZS)rzA propagator that ignores any trace context that was extracted by the wrapped propagator. Used when `logfire.configure(distributed_tracing=False)` is called. ct|||g|i|}||k(r|Stjtj||Sr()superr r set_span_in_contextget_current_span)r*rr r+r,result __class__s rr z'NoExtractTraceContextPropagator.extractrsJ'CDCFC W M(()?)?)H&QQrr( rrr zotel_context.Context | Noner+rr,rr0r1)r2r3r4r5r __classcell__r?s@rrrlsN04 R R- R R  R  R RrrcJeZdZUdZdZded< d dfd ZxZS)rzA propagator that warns the first time that trace context is extracted by the wrapped propagator. Used when `logfire.configure(distributed_tracing=None)` is called. This is the default behavior. Fboolwarnedct|||g|i|}|js^||k7rYtj|tj|k7r.d|_d}t |t tj||S)NTzFound propagated trace context. See https://logfire.pydantic.dev/docs/how-to-guides/distributed-tracing/#unintentional-distributed-tracing.) r;r rEr r=r RuntimeWarninglogfirewarn)r*rr r+r,r>messager?s rr z+WarnOnExtractTraceContextPropagator.extracts{'CDCFC{{v0U5K5KG5TX]XnXnouXv5vDKz  $G^ < LL ! rr(r@)r2r3r4r5rEr6r rArBs@rrrsS FD 04-     rr)r0r)rrrrDr0zIterator[None])r5 __future__r contextlibr dataclassesrtypingrrr opentelemetryr rr r !opentelemetry.propagators.textmapr rHlogfire._internal.stack_infor strr__all__rrr&rrr8rrrTs #%!))CC?@c" ; FCH))0  #) #  # R&7R( *; r