phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix OTEL TracerProvider

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-06 15:15:49 -07:00
parent 4f409e647f
commit 9738aaebf6
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_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"), "deployment.environment": os.getenv("OTEL_ENVIRONMENT_NAME", "production"),
} }
RAW_REQUEST_SPAN_NAME = "raw_gen_ai_request" RAW_REQUEST_SPAN_NAME = "raw_gen_ai_request"
@ -73,9 +71,7 @@ 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( provider = TracerProvider(resource=Resource(attributes=LITELLM_RESOURCE))
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)