fix: replace --skip_db_migration_check with --enforce_prisma_migration_check

Krrish Dholakia · claude · Krrish Dholakia · commit a53ecd532cb3 · 2026-03-14T18:54:21.000-07:00
Flip the default behavior: database migration failures now warn and
continue by default. Only when --enforce_prisma_migration_check (or
ENFORCE_PRISMA_MIGRATION_CHECK=true) is explicitly set will the proxy
exit on migration failure.

Co-Authored-By: Claude Opus 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/litellm/proxy/proxy_cli.py b/litellm/proxy/proxy_cli.py
@@ -557,11 +557,11 @@ def _maybe_setup_prometheus_multiproc_dir(
     envvar="MAX_REQUESTS_BEFORE_RESTART",
 )
 @click.option(
-    "--skip_db_migration_check",
+    "--enforce_prisma_migration_check",
     is_flag=True,
     default=False,
-    help="Warn and continue instead of exiting when database migration fails.",
-    envvar="SKIP_DB_MIGRATION_CHECK",
+    help="Exit with error if database migration fails on startup.",
+    envvar="ENFORCE_PRISMA_MIGRATION_CHECK",
 )
 def run_server(  # noqa: PLR0915
     host,
@@ -602,7 +602,7 @@ def run_server(  # noqa: PLR0915
     skip_server_startup,
     keepalive_timeout,
     max_requests_before_restart,
-    skip_db_migration_check: bool,
+    enforce_prisma_migration_check: bool,
 ):
     args = locals()
     if local:
@@ -868,18 +868,17 @@ def run_server(  # noqa: PLR0915
                     if not PrismaManager.setup_database(
                         use_migrate=not use_prisma_db_push
                     ):
-                        if skip_db_migration_check:
+                        if enforce_prisma_migration_check:
                             print(  # noqa
-                                "\033[1;33mLiteLLM Proxy: Database migration failed but continuing startup "
-                                "because --skip_db_migration_check is set.\033[0m"
+                                "\033[1;31mLiteLLM Proxy: Database setup failed after multiple retries. "
+                                "The proxy cannot start safely. Please check your database connection and migration status.\033[0m"
                             )
+                            sys.exit(1)
                         else:
                             print(  # noqa
-                                "\033[1;31mLiteLLM Proxy: Database setup failed after multiple retries. "
-                                "The proxy cannot start safely. Please check your database connection and migration status. "
-                                "Pass --skip_db_migration_check or set SKIP_DB_MIGRATION_CHECK=true to warn and continue instead of exiting.\033[0m"
+                                "\033[1;33mLiteLLM Proxy: Database migration failed but continuing startup. "
+                                "Set --enforce_prisma_migration_check or ENFORCE_PRISMA_MIGRATION_CHECK=true to exit on failure.\033[0m"
                             )
-                            sys.exit(1)
             else:
                 print(  # noqa
                     f"Unable to connect to DB. DATABASE_URL found in environment, but prisma package not found."  # noqa
