From 174d6eaf136b3c11825412c88503ceccbfd561d0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Jun 2024 11:50:16 -0700 Subject: [PATCH 1/2] 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 fa7be1d574..e28fb9bc11 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) From c91b254e254c4f29dd355b3aebdfd3bdb803a8a5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Jul 2024 15:15:49 -0700 Subject: [PATCH 2/2] fix OTEL TracerProvider --- litellm/integrations/opentelemetry.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/litellm/integrations/opentelemetry.py b/litellm/integrations/opentelemetry.py index e28fb9bc11..6b72f75452 100644 --- a/litellm/integrations/opentelemetry.py +++ b/litellm/integrations/opentelemetry.py @@ -29,8 +29,6 @@ else: 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" @@ -73,9 +71,7 @@ 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, **LITELLM_ENVIRONMENT}) - ) + provider = TracerProvider(resource=Resource(attributes=LITELLM_RESOURCE)) provider.add_span_processor(self._get_span_processor()) trace.set_tracer_provider(provider)