scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

prom svc logger init if it's None

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-07 09:02:03 -07:00
parent 7e0cd39de7
commit 315bba34e6
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/_service_logger.py
View file

@ -73,6 +73,7 @@ class ServiceLogging(CustomLogger):
)
for callback in litellm.service_callback:
if callback == "prometheus_system":
await self.init_prometheus_services_logger_if_none()
await self.prometheusServicesLogger.async_service_success_hook(
payload=payload
)
@ -88,6 +89,11 @@ class ServiceLogging(CustomLogger):
event_metadata=event_metadata,
)
async def init_prometheus_services_logger_if_none(self):
if self.prometheusServicesLogger is None:
self.prometheusServicesLogger = self.prometheusServicesLogger()
return
async def async_service_failure_hook(
self,
service: ServiceTypes,
@ -120,8 +126,7 @@ class ServiceLogging(CustomLogger):
)
for callback in litellm.service_callback:
if callback == "prometheus_system":
if self.prometheusServicesLogger is None:
self.prometheusServicesLogger = self.prometheusServicesLogger()
await self.init_prometheus_services_logger_if_none()
await self.prometheusServicesLogger.async_service_failure_hook(
payload=payload
)