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

otel - use correct raw request name

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-08 14:22:44 -07:00
parent 47f1e16533
commit 9478c98fff
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/opentelemetry.py
View file

@ -23,7 +23,8 @@ 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"),
} }
RAW_REQUEST_SPAN_NAME = "RAW_GENAI_REQUEST" RAW_REQUEST_SPAN_NAME = "raw_gen_ai_request"
LITELLM_REQUEST_SPAN_NAME = "litellm_request"
@dataclass @dataclass
@ -412,7 +413,7 @@ class OpenTelemetry(CustomLogger):
return int(dt.timestamp() * 1e9) return int(dt.timestamp() * 1e9)
def _get_span_name(self, kwargs): def _get_span_name(self, kwargs):
return f"litellm-{kwargs.get('call_type', 'completion')}" return LITELLM_REQUEST_SPAN_NAME
def _get_span_context(self, kwargs): def _get_span_context(self, kwargs):
from opentelemetry.trace.propagation.tracecontext import ( from opentelemetry.trace.propagation.tracecontext import (