diff --git a/litellm/integrations/opentelemetry.py b/litellm/integrations/opentelemetry.py index 480e1d6b8..7af354694 100644 --- a/litellm/integrations/opentelemetry.py +++ b/litellm/integrations/opentelemetry.py @@ -33,6 +33,8 @@ class OpenTelemetry(CustomLogger): from opentelemetry.sdk.trace import TracerProvider self.config = config + self.OTEL_ENDPOINT = self.config.endpoint + self.OTEL_BEARER_TOKEN = self.config.bearer_token provider = TracerProvider(resource=Resource(attributes=LITELLM_RESOURCE)) provider.add_span_processor(self._get_span_processor()) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c154bd8e9..6d6a1e36b 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2399,6 +2399,12 @@ class ProxyConfig: and callback in known_compatible_callbacks ): imported_list.append(callback) + elif isinstance(callback, str) and callback == "otel": + from litellm.integrations.opentelemetry import ( + OpenTelemetry, + ) + + imported_list.append(OpenTelemetry()) elif isinstance(callback, str) and callback == "presidio": from litellm.proxy.hooks.presidio_pii_masking import ( _OPTIONAL_PresidioPIIMasking,