phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

OTEL allow setting deployment environment

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-26 11:50:16 -07:00
parent ae431eb85d
commit 174d6eaf13
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/integrations/opentelemetry.py
View file

@ -30,6 +30,9 @@ LITELLM_TRACER_NAME = os.getenv("OTEL_TRACER_NAME", "litellm")
LITELLM_RESOURCE = { LITELLM_RESOURCE = {
"service.name": os.getenv("OTEL_SERVICE_NAME", "litellm"), "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" RAW_REQUEST_SPAN_NAME = "raw_gen_ai_request"
LITELLM_REQUEST_SPAN_NAME = "litellm_request" LITELLM_REQUEST_SPAN_NAME = "litellm_request"
@ -70,7 +73,9 @@ class OpenTelemetry(CustomLogger):
self.OTEL_EXPORTER = self.config.exporter self.OTEL_EXPORTER = self.config.exporter
self.OTEL_ENDPOINT = self.config.endpoint self.OTEL_ENDPOINT = self.config.endpoint
self.OTEL_HEADERS = self.config.headers 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()) provider.add_span_processor(self._get_span_processor())
trace.set_tracer_provider(provider) trace.set_tracer_provider(provider)