scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(proxy-extras): schema-qualify the _prisma_migrations logs lookup for non-public Prisma schemas

Browse source
This commit is contained in:
mateo-berri 2026-09-01 14:04:12 -07:00
parent 692f3b513c
commit 5192b2162c
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -292,14 +292,22 @@ class ProxyExtrasDBManager:
return ""
cleaned_url = ProxyExtrasDBManager._strip_prisma_query_params(database_url)
ledger_table = psycopg.sql.SQL("{}.{}").format(
psycopg.sql.Identifier(
ProxyExtrasDBManager._prisma_schema_param(database_url) or "public"
),
psycopg.sql.Identifier("_prisma_migrations"),
)
try:
with psycopg.connect(
cleaned_url, connect_timeout=10, autocommit=True
) as conn:
row = conn.execute(
"SELECT logs FROM _prisma_migrations "
"WHERE migration_name = %s AND finished_at IS NULL "
"AND rolled_back_at IS NULL",
psycopg.sql.SQL(
"SELECT logs FROM {} "
"WHERE migration_name = %s AND finished_at IS NULL "
"AND rolled_back_at IS NULL"
).format(ledger_table),
(migration_name,),
).fetchone()
except (psycopg.OperationalError, psycopg.DatabaseError):