scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Merge 4d4f97986f into f285229b51

Browse source
This commit is contained in:
Ansh Rohilla 2026-09-30 16:57:42 -04:00 • committed by GitHub
commit 6d5965d404
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm-proxy-extras/litellm_proxy_extras/utils.py
View file

@ -656,6 +656,9 @@ class ProxyExtrasDBManager:
if not parsed.query:
return url
libpq_params = {
"host",
"hostaddr",
"port",
"sslmode",
"sslcert",
"sslkey",

8
litellm-proxy-extras/tests/test_setup_database_fail_fast.py
View file

@ -71,6 +71,14 @@ def test_strip_prisma_query_params_passthrough_no_query():
assert ProxyExtrasDBManager._strip_prisma_query_params(url) == url
def test_strip_prisma_query_params_preserves_host_socket():
"""Unix-socket DATABASE_URLs using host parameter (e.g. Cloud SQL) must be preserved."""
url = "postgresql://user:pass@localhost/litellm?host=/cloudsql/project:region:instance&connection_limit=10"
stripped = ProxyExtrasDBManager._strip_prisma_query_params(url)
assert "host=%2Fcloudsql%2Fproject%3Aregion%3Ainstance" in stripped
assert "connection_limit" not in stripped
def test_migration_timestamp_extracts_leading_digits():
assert _migration_timestamp("20260101000000_add_foo") == 20260101000000
assert _migration_timestamp("20250326162113_baseline") == 20250326162113