phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 97af7822aa into b82af5b826

Browse source
This commit is contained in:
Krish Dholakia 2025-04-24 00:57:35 -07:00 • committed by GitHub
commit cc185780d4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/db/check_migration.py
View file

@ -76,16 +76,13 @@ def check_prisma_schema_diff_helper(db_url: str) -> Tuple[bool, List[str]]:
sql_commands = extract_sql_commands(result.stdout) sql_commands = extract_sql_commands(result.stdout)
if sql_commands: if sql_commands:
print("Changes to DB Schema detected") # noqa: T201 verbose_logger.info("Detected changes to DB Schema")
print("Required SQL commands:") # noqa: T201
for command in sql_commands:
print(command) # noqa: T201
return True, sql_commands return True, sql_commands
else: else:
return False, [] return False, []
except subprocess.CalledProcessError as e: except subprocess.CalledProcessError as e:
error_message = f"Failed to generate migration diff. Error: {e.stderr}" error_message = f"Failed to generate migration diff. Error: {e.stderr}. This will not block server start."
print(error_message) # noqa: T201 verbose_logger.exception(error_message)
return False, [] return False, []
@ -97,8 +94,8 @@ def check_prisma_schema_diff(db_url: Optional[str] = None) -> None:
raise Exception("DATABASE_URL not set") raise Exception("DATABASE_URL not set")
has_diff, message = check_prisma_schema_diff_helper(db_url) has_diff, message = check_prisma_schema_diff_helper(db_url)
if has_diff: if has_diff:
verbose_logger.exception( error_message = "🚨🚨🚨 Prisma schema out of sync with db, and `DISABLE_PRISMA_SCHEMA_UPDATE` is enabled. Manual override is required. Consider running these sql_commands to sync the two - {}".format(
"🚨🚨🚨 prisma schema out of sync with db. Consider running these sql_commands to sync the two - {}".format(
message message
) )
) verbose_logger.exception(error_message)
raise Exception(error_message)