UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-07 00:59:52 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: db

Browse Source
This commit is contained in:
Timothy J. Baek 2024-07-08 10:46:35 -07:00
parent 36c0f7f273
commit 39d3dcd032
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/migrations/env.py
View File

@ -18,6 +18,8 @@ from apps.webui.models.users import User
from apps.webui.models.files import File
from apps.webui.models.functions import Function
from config import DATABASE_URL
# this is the Alembic Config object, which provides
# access to the values within the .ini file in use.
config = context.config
@ -38,9 +40,13 @@ target_metadata = Auth.metadata
# my_important_option = config.get_main_option("my_important_option")
# ... etc.
database_url = os.getenv("DATABASE_URL", None)
if database_url:
config.set_main_option("sqlalchemy.url", database_url)
DB_URL = DATABASE_URL
# Replace the postgres:// with postgresql://
if "postgres://" in DB_URL:
DB_URL = DB_URL.replace("postgres://", "postgresql://")
if DB_URL:
config.set_main_option("sqlalchemy.url", DB_URL)
def run_migrations_offline() -> None: