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

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):
db = connect(db_url)
if isinstance(db, PostgresqlDatabase):
db = PooledPostgresqlExtDatabase(
db.database,
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
# Directly use PostgresqlDatabase without pooling
db.autoconnect = True
elif isinstance(db, SqliteDatabase):
db = PooledSqliteDatabase(
db.database,
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
# Directly use SqliteDatabase without pooling
db.autoconnect = True
else:
raise ValueError('Unsupported database connection')
return db