scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

fix(proxy): harden prisma reconnect cycle semantics

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-20 17:21:47 -08:00
parent 67809b8981
commit edcb6bd14d
1 changed files with 11 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
litellm/proxy/utils.py
View file

@ -3553,30 +3553,26 @@ class PrismaClient:
) -> None:
"""
Run a reconnect cycle. When timeout_seconds is set, use direct db operations
with per-step timeouts to avoid long retries on hot paths (e.g. auth).
with a single overall timeout budget to avoid long retries on hot paths
(e.g. auth).
"""
if timeout_seconds is None:
async def _do_direct_reconnect() -> None:
try:
await self.disconnect()
await self.db.disconnect()
except Exception as disconnect_err:
verbose_proxy_logger.debug(
"Prisma DB disconnect before reconnect failed (ignored): %s",
disconnect_err,
)
await self.connect()
await self.health_check()
await self.db.connect()
await self.db.query_raw("SELECT 1")
if timeout_seconds is None:
await _do_direct_reconnect()
return
try:
await self.db.disconnect()
except Exception as disconnect_err:
verbose_proxy_logger.debug(
"Prisma DB disconnect before reconnect failed (ignored): %s",
disconnect_err,
)
await asyncio.wait_for(self.db.connect(), timeout=timeout_seconds)
await asyncio.wait_for(self.db.query_raw("SELECT 1"), timeout=timeout_seconds)
await asyncio.wait_for(_do_direct_reconnect(), timeout=timeout_seconds)
async def attempt_db_reconnect(
self,