From 174d6eaf136b3c11825412c88503ceccbfd561d0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Jun 2024 11:50:16 -0700 Subject: [PATCH] OTEL allow setting deployment environment --- litellm/integrations/opentelemetry.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/opentelemetry.py b/litellm/integrations/opentelemetry.py index fa7be1d574c..e28fb9bc11c 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)