diff --git a/litellm/integrations/opentelemetry.py b/litellm/integrations/opentelemetry.py index fa7be1d57..e28fb9bc1 100644 --- a/litellm/integrations/opentelemetry.py +++ b/litellm/integrations/opentelemetry.py @@ -30,6 +30,9 @@ LITELLM_TRACER_NAME = os.getenv("OTEL_TRACER_NAME", "litellm") LITELLM_RESOURCE = { "service.name": os.getenv("OTEL_SERVICE_NAME", "litellm"), } +LITELLM_ENVIRONMENT = { + "deployment.environment": os.getenv("OTEL_ENVIRONMENT_NAME", "production"), +} RAW_REQUEST_SPAN_NAME = "raw_gen_ai_request" LITELLM_REQUEST_SPAN_NAME = "litellm_request" @@ -70,7 +73,9 @@ class OpenTelemetry(CustomLogger): self.OTEL_EXPORTER = self.config.exporter self.OTEL_ENDPOINT = self.config.endpoint self.OTEL_HEADERS = self.config.headers - provider = TracerProvider(resource=Resource(attributes=LITELLM_RESOURCE)) + provider = TracerProvider( + resource=Resource(attributes={**LITELLM_RESOURCE, **LITELLM_ENVIRONMENT}) + ) provider.add_span_processor(self._get_span_processor()) trace.set_tracer_provider(provider)