fix: alembic

This commit is contained in:
Timothy J. Baek 2024-07-08 12:55:27 -07:00
parent d3ef3a7494
commit 4b6ee584c2
2 changed files with 8 additions and 5 deletions

View File

@ -173,11 +173,14 @@ https://github.com/open-webui/open-webui
def run_migrations(): def run_migrations():
from alembic.config import Config try:
from alembic import command from alembic.config import Config
from alembic import command
alembic_cfg = Config("alembic.ini") alembic_cfg = Config("alembic.ini")
command.upgrade(alembic_cfg, "head") command.upgrade(alembic_cfg, "head")
except Exception as e:
print(f"Error: {e}")
@asynccontextmanager @asynccontextmanager

View File

@ -43,7 +43,7 @@ target_metadata = Auth.metadata
DB_URL = DATABASE_URL DB_URL = DATABASE_URL
if DB_URL: if DB_URL:
config.set_main_option("sqlalchemy.url", DB_URL) config.set_main_option("sqlalchemy.url", DB_URL.replace("%", "%%"))
def run_migrations_offline() -> None: def run_migrations_offline() -> None: