Coverage for mindsdb / utilities / otel / tracer.py: 0%

1from opentelemetry import trace 

2from opentelemetry.sdk.resources import Resource 

3from opentelemetry.sdk.trace import TracerProvider 

4from opentelemetry.sdk.trace.export import BatchSpanProcessor, SpanExporter 

5from opentelemetry.sdk.trace.sampling import Sampler 

6 

7 

8def setup_tracer(resource: Resource, sampler: Sampler, exporter: SpanExporter) -> None: 

9 """ 

10 Setup OpenTelemetry tracing 

11 """ 

12 # Set the tracer provider with the custom resource 

13 trace.set_tracer_provider(TracerProvider(resource=resource, sampler=sampler)) 

14 

15 # Replace the default span processor with the custom one 

16 trace.get_tracer_provider().add_span_processor(BatchSpanProcessor(exporter))