Spaces:
Running
Running
File size: 612 Bytes
d631808 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from .provider import TraceProvider
GLOBAL_TRACE_PROVIDER: TraceProvider | None = None
def set_trace_provider(provider: TraceProvider) -> None:
"""Set the global trace provider used by tracing utilities."""
global GLOBAL_TRACE_PROVIDER
GLOBAL_TRACE_PROVIDER = provider
def get_trace_provider() -> TraceProvider:
"""Get the global trace provider used by tracing utilities."""
if GLOBAL_TRACE_PROVIDER is None:
raise RuntimeError("Trace provider not set")
return GLOBAL_TRACE_PROVIDER
|