ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-11 20:05:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-06-21 19:12:43 +04:00
parent 38e8209d71
commit aef0ad2d10
1 changed files with 16 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
backend/open_webui/internal/db.py
View file

@ -84,24 +84,23 @@ if "sqlite" in SQLALCHEMY_DATABASE_URL:
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False} SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
) )
else: else:
if DATABASE_POOL_SIZE > 0: if isinstance(DATABASE_POOL_SIZE, int):
engine = create_engine( if DATABASE_POOL_SIZE > 0:
SQLALCHEMY_DATABASE_URL, engine = create_engine(
pool_size=DATABASE_POOL_SIZE, SQLALCHEMY_DATABASE_URL,
max_overflow=DATABASE_POOL_MAX_OVERFLOW, pool_size=DATABASE_POOL_SIZE,
pool_timeout=DATABASE_POOL_TIMEOUT, max_overflow=DATABASE_POOL_MAX_OVERFLOW,
pool_recycle=DATABASE_POOL_RECYCLE, pool_timeout=DATABASE_POOL_TIMEOUT,
pool_pre_ping=True, pool_recycle=DATABASE_POOL_RECYCLE,
poolclass=QueuePool, pool_pre_ping=True,
) poolclass=QueuePool,
elif DATABASE_POOL_SIZE == 0: )
engine = create_engine( else:
SQLALCHEMY_DATABASE_URL, pool_pre_ping=True, poolclass=NullPool engine = create_engine(
) SQLALCHEMY_DATABASE_URL, pool_pre_ping=True, poolclass=NullPool
)
else: else:
engine = create_engine( engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_pre_ping=True)
SQLALCHEMY_DATABASE_URL, pool_pre_ping=True
)
SessionLocal = sessionmaker( SessionLocal = sessionmaker(