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

fix(_service_logging.py): only trigger otel if in service_callback

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4511
This commit is contained in:
Krrish Dholakia 2024-07-03 09:48:19 -07:00
parent 2116dbcdc1
commit 606d04b05b
4 changed files with 64 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/_service_logger.py
View file

@ -75,16 +75,16 @@ class ServiceLogging(CustomLogger):
await self.prometheusServicesLogger.async_service_success_hook(
payload=payload
)
elif callback == "otel":
from litellm.proxy.proxy_server import open_telemetry_logger
from litellm.proxy.proxy_server import open_telemetry_logger
if parent_otel_span is not None and open_telemetry_logger is not None:
await open_telemetry_logger.async_service_success_hook(
payload=payload,
parent_otel_span=parent_otel_span,
start_time=start_time,
end_time=end_time,
)
if parent_otel_span is not None and open_telemetry_logger is not None:
await open_telemetry_logger.async_service_success_hook(
payload=payload,
parent_otel_span=parent_otel_span,
start_time=start_time,
end_time=end_time,
)
async def async_service_failure_hook(
self,