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

stop even using pooled DBs in peewee

Browse Source
This commit is contained in:
perf3ct 2024-06-17 09:56:31 -07:00
parent 981866eb93
commit 5c655f298b
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
1 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,23 +22,11 @@ class PeeweeConnectionState(object):
def register_connection(db_url): def register_connection(db_url):
db = connect(db_url) db = connect(db_url)
if isinstance(db, PostgresqlDatabase): if isinstance(db, PostgresqlDatabase):
db = PooledPostgresqlExtDatabase( # Directly use PostgresqlDatabase without pooling
db.database, db.autoconnect = True
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
elif isinstance(db, SqliteDatabase): elif isinstance(db, SqliteDatabase):
db = PooledSqliteDatabase( # Directly use SqliteDatabase without pooling
db.database, db.autoconnect = True
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
else: else:
raise ValueError('Unsupported database connection') raise ValueError('Unsupported database connection')
return db return db