phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix OTEL semantic conventions does not exist

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-11 12:28:38 -07:00
parent 842b8337e6
commit 1af3d34d8b
3 changed files with 51 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/opentelemetry.py
View file

@ -247,7 +247,7 @@ class OpenTelemetry(CustomLogger):
span.end(end_time=self._to_ns(end_time))
def set_tools_attributes(self, span: Span, tools):
from opentelemetry.semconv.ai import SpanAttributes
from litellm.proxy._types import SpanAttributes
import json
if not tools:
@ -272,7 +272,7 @@ class OpenTelemetry(CustomLogger):
pass
def set_attributes(self, span: Span, kwargs, response_obj):
from opentelemetry.semconv.ai import SpanAttributes
from litellm.proxy._types import SpanAttributes
optional_params = kwargs.get("optional_params", {})
litellm_params = kwargs.get("litellm_params", {}) or {}
@ -407,7 +407,7 @@ class OpenTelemetry(CustomLogger):
)
def set_raw_request_attributes(self, span: Span, kwargs, response_obj):
from opentelemetry.semconv.ai import SpanAttributes
from litellm.proxy._types import SpanAttributes
optional_params = kwargs.get("optional_params", {})
litellm_params = kwargs.get("litellm_params", {}) or {}