tracer provider

python/openinference-instrumentation/src/openinference/instrumentation/config.py

@@ -30,6 +30,7 @@
     detach,
     set_value,
 )
+from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.id_generator import IdGenerator
 from opentelemetry.trace import (
     INVALID_SPAN_ID,
@@ -668,6 +669,25 @@ def start_span(
         return span
 
 
+class OpenInferenceTracerProvider(TracerProvider):
+    def __init__(
+        self,
+        *args: Any,
+        config: Optional[TraceConfig] = None,
+        **kwargs: Any,
+    ) -> None:
+        super().__init__(*args, **kwargs)
+        self._oi_trace_config = config or TraceConfig()
+
+    def get_tracer(
+        self,
+        *args: Any,
+        **kwargs: Any,
+    ) -> OITracer:
+        tracer = super().get_tracer(*args, **kwargs)
+        return OITracer(tracer, config=self._oi_trace_config)
+
+
 def is_base64_url(url: str) -> bool:
     if not isinstance(url, str):
         return False