scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-15 09:12:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 311

fix(proxy): read the statement_timeout probe row without an untyped default

Browse source
This commit is contained in:
devin-ai-integration[bot] 2026-11-04 16:21:40 +00:00
parent f0d33aae24
commit df144e6a65
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/utils.py
View file

@ -5337,7 +5337,7 @@ async def _lower_statement_timeout(transaction: _RawReadTransaction, timeout_ms:
rows: Final = await transaction.query_raw(
"SELECT setting::int AS ms FROM pg_settings WHERE name = 'statement_timeout'"
)
current_ms: Final = _STATEMENT_TIMEOUT_MS.validate_python(next(iter(rows), {}).get("ms", 0))
current_ms: Final = _STATEMENT_TIMEOUT_MS.validate_python(rows[0].get("ms", 0) if rows else 0)
if 0 < current_ms <= timeout_ms:
return current_ms
await transaction.execute_raw(f"SET LOCAL statement_timeout = {timeout_ms}")