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(auth): bound reconnect latency on key lookup path

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-20 15:53:05 -08:00
parent e89c5b7b60
commit 14f529a0e9
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/auth/auth_checks.py
View file

@ -2003,8 +2003,14 @@ async def _fetch_key_object_from_db_with_reconnect(
if PrismaDBExceptionHandler.is_database_connection_error(e):
did_reconnect = False
if hasattr(prisma_client, "attempt_db_reconnect"):
auth_reconnect_timeout = getattr(
prisma_client, "_db_auth_reconnect_timeout_seconds", 2.0
)
if not isinstance(auth_reconnect_timeout, (int, float)):
auth_reconnect_timeout = 2.0
did_reconnect = await prisma_client.attempt_db_reconnect(
reason="auth_get_key_object_lookup_failure"
reason="auth_get_key_object_lookup_failure",
timeout_seconds=auth_reconnect_timeout,
)
if did_reconnect:
return await prisma_client.get_data(