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

fix: Pass all parsed options to ReconnectingPostgresqlDatabase

Browse Source
This commit is contained in:
Zhaofeng Li 2024-08-12 08:52:16 -06:00
parent 8d257ed596
commit a53c2a8c6b
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/apps/webui/internal/wrappers.py
View File

@ -54,13 +54,7 @@ def register_connection(db_url):
connection = parse(db_url) connection = parse(db_url)
# Use our custom database class that supports reconnection # Use our custom database class that supports reconnection
db = ReconnectingPostgresqlDatabase( db = ReconnectingPostgresqlDatabase(**connection)
connection["database"],
user=connection["user"],
password=connection["password"],
host=connection["host"],
port=connection["port"],
)
db.connect(reuse_if_open=True) db.connect(reuse_if_open=True)
elif isinstance(db, SqliteDatabase): elif isinstance(db, SqliteDatabase):
# Enable autoconnect for SQLite databases, managed by Peewee # Enable autoconnect for SQLite databases, managed by Peewee