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

fix OTEL TracerProvider

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-06 15:15:49 -07:00
parent 174d6eaf13
commit c91b254e25
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/opentelemetry.py
View file

@ -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)