tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-26 10:28:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files
litellm/_service_logger.py
+7 -2
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
)