UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-26 17:20:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-22 16:34:12 +02:00
parent 670672c067
commit 25de3e753d
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/webui/internal/db.py
View File

@ -54,11 +54,10 @@ else:
# Workaround to handle the peewee migration # Workaround to handle the peewee migration
# This is required to ensure the peewee migration is handled before the alembic migration # This is required to ensure the peewee migration is handled before the alembic migration
def handle_peewee_migration(DATABASE_URL): def handle_peewee_migration(DATABASE_URL):
# db = None
try: try:
# Replace the postgresql:// with postgres:// to handle the peewee migration # Replace the postgresql:// with postgres:// to handle the peewee migration
db = register_connection( db = register_connection(DATABASE_URL.replace("postgresql://", "postgres://"))
DATABASE_URL.replace("postgresql://", "postgres://"), unquote_password=True
)
migrate_dir = BACKEND_DIR / "apps" / "webui" / "internal" / "migrations" migrate_dir = BACKEND_DIR / "apps" / "webui" / "internal" / "migrations"
router = Router(db, logger=log, migrate_dir=migrate_dir) router = Router(db, logger=log, migrate_dir=migrate_dir)
router.run() router.run()
@ -67,7 +66,6 @@ def handle_peewee_migration(DATABASE_URL):
except Exception as e: except Exception as e:
log.error(f"Failed to initialize the database connection: {e}") log.error(f"Failed to initialize the database connection: {e}")
raise raise
finally: finally:
# Properly closing the database connection # Properly closing the database connection
if db and not db.is_closed(): if db and not db.is_closed():