scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 174

fix(alerts): isolate optional flush tracking failures

Browse source
This commit is contained in:
moe-berri 2026-11-07 12:16:12 +08:00
parent a75d6335b5
commit fb50c5714c
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/repositories/usage_alert_health.py
View file

@ -87,8 +87,8 @@ class UsageAlertFlushRecorder:
prior: Final = self._user_hour if source == "user" else self._team_hour
if prior == hour:
return
database: Final = cast(MetricsDatabase, prisma_client.writer_db) # cast-ok: generated Prisma delegate
try:
database: Final = cast(MetricsDatabase, prisma_client.writer_db) # cast-ok: generated Prisma delegate
if not await UsageAlertHealthRepository(database).record(source, now):
return
except Exception: # noqa: BLE001 # observability must not fail or requeue a committed spend flush

2
tests/unit/proxy/test_proxy_server_endpoints_and_startup.py
View file

@ -1656,6 +1656,8 @@ class MockPrisma:
self.database_url = database_url
self.proxy_logging_obj = proxy_logging_obj
self.http_client = http_client
self.writer_db = MagicMock()
self.writer_db.litellm_config.find_unique = AsyncMock(return_value=None)
async def connect(self):
pass