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

[Fix] OTEL - Don't log messages when callback settings disable message logging (#5875)

Browse source 
* fix otel dont log messages

* otel fix redis failure hook logging
This commit is contained in:
Ishaan Jaff 2024-09-24 18:29:52 -07:00 • committed by GitHub
parent d37c8b5c6b
commit 3ccdb42d26
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 60 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

31
litellm/_service_logger.py
View file

@ -196,20 +196,27 @@ class ServiceLogging(CustomLogger):
end_time=end_time,
event_metadata=event_metadata,
)
elif callback == "otel":
from litellm.integrations.opentelemetry import OpenTelemetry
from litellm.proxy.proxy_server import open_telemetry_logger
from litellm.proxy.proxy_server import open_telemetry_logger
await self.init_otel_logger_if_none()
if not isinstance(error, str):
error = str(error)
if open_telemetry_logger is not None:
await self.otel_logger.async_service_failure_hook(
payload=payload,
parent_otel_span=parent_otel_span,
start_time=start_time,
end_time=end_time,
event_metadata=event_metadata,
error=error,
)
if not isinstance(error, str):
error = str(error)
if (
parent_otel_span is not None
and open_telemetry_logger is not None
and isinstance(open_telemetry_logger, OpenTelemetry)
):
await self.otel_logger.async_service_success_hook(
payload=payload,
parent_otel_span=parent_otel_span,
start_time=start_time,
end_time=end_time,
event_metadata=event_metadata,
)
async def async_post_call_failure_hook(
self,