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

otel log event_metadata

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-05 20:01:58 -07:00
parent 3905825609
commit 774807d2c4
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/integrations/opentelemetry.py
View file

@ -153,6 +153,11 @@ class OpenTelemetry(CustomLogger):
if event_metadata: if event_metadata:
for key, value in event_metadata.items(): for key, value in event_metadata.items():
if isinstance(value, dict):
try:
value = str(value)
except Exception:
value = "litllm logging error - could_not_json_serialize"
service_logging_span.set_attribute(key, value) service_logging_span.set_attribute(key, value)
service_logging_span.set_status(Status(StatusCode.OK)) service_logging_span.set_status(Status(StatusCode.OK))
service_logging_span.end(end_time=_end_time_ns) service_logging_span.end(end_time=_end_time_ns)
@ -163,6 +168,7 @@ class OpenTelemetry(CustomLogger):
parent_otel_span: Optional[Span] = None, parent_otel_span: Optional[Span] = None,
start_time: Optional[Union[datetime, float]] = None, start_time: Optional[Union[datetime, float]] = None,
end_time: Optional[Union[float, datetime]] = None, end_time: Optional[Union[float, datetime]] = None,
event_metadata: Optional[dict] = None,
): ):
from datetime import datetime from datetime import datetime
@ -193,6 +199,15 @@ class OpenTelemetry(CustomLogger):
service_logging_span.set_attribute( service_logging_span.set_attribute(
key="service", value=payload.service.value key="service", value=payload.service.value
) )
if event_metadata:
for key, value in event_metadata.items():
if isinstance(value, dict):
try:
value = str(value)
except Exception:
value = "litllm logging error - could_not_json_serialize"
service_logging_span.set_attribute(key, value)
service_logging_span.set_status(Status(StatusCode.ERROR)) service_logging_span.set_status(Status(StatusCode.ERROR))
service_logging_span.end(end_time=_end_time_ns) service_logging_span.end(end_time=_end_time_ns)