diff --git a/backend/main.py b/backend/main.py index 2da19c5c7..c37124f1e 100644 --- a/backend/main.py +++ b/backend/main.py @@ -173,13 +173,11 @@ https://github.com/open-webui/open-webui def run_migrations(): - env = os.environ.copy() - env["DATABASE_URL"] = DATABASE_URL - migration_task = subprocess.run( - ["alembic", f"-c{BACKEND_DIR}/alembic.ini", "upgrade", "head"], env=env - ) - if migration_task.returncode > 0: - raise ValueError("Error running migrations") + from alembic.config import Config + from alembic import command + + alembic_cfg = Config("alembic.ini") + command.upgrade(alembic_cfg, "head") @asynccontextmanager