OSSForks/ollama-webui
1
0
Fork 0
mirror of https://github.com/zebrajr/ollama-webui.git synced 2025-12-05 12:20:26 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: format

Browse Source 
Signed-off-by: Adam Tao <tcx4c70@gmail.com>
This commit is contained in:
Adam Tao 2025-07-19 19:20:53 +08:00
parent 635cb8e3ff
commit 7bd7559bfe
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/env.py
View File

@ -336,9 +336,7 @@ else:
except Exception:
DATABASE_POOL_RECYCLE = 3600
DATABASE_ENABLE_SQLITE_WAL = (
os.environ.get("DATABASE_ENABLE_SQLITE_WAL", "False").lower() == "true"
)
DATABASE_ENABLE_SQLITE_WAL = (os.environ.get("DATABASE_ENABLE_SQLITE_WAL", "False").lower() == "true")
DATABASE_DEDUPLICATE_INTERVAL = (
os.environ.get("DATABASE_DEDUPLICATE_INTERVAL", 0.)

2
backend/open_webui/internal/db.py
View File

@ -115,6 +115,7 @@ elif "sqlite" in SQLALCHEMY_DATABASE_URL:
engine = create_engine(
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
)
def on_connect(dbapi_connection, connection_record):
cursor = dbapi_connection.cursor()
if DATABASE_ENABLE_SQLITE_WAL:
@ -122,6 +123,7 @@ elif "sqlite" in SQLALCHEMY_DATABASE_URL:
else:
cursor.execute("PRAGMA journal_mode=DELETE")
cursor.close()
event.listen(engine, "connect", on_connect)
else:
if isinstance(DATABASE_POOL_SIZE, int):

2
backend/open_webui/utils/misc.py
View File

@ -500,6 +500,7 @@ def deduplicate(interval: float = 10.0):
:param interval: Duration in seconds to wait before allowing the function to be called again.
"""
def decorator(func):
last_calls = {}
lock = threading.Lock()
@ -514,6 +515,7 @@ def deduplicate(interval: float = 10.0):
return None
last_calls[key] = now
return func(*args, **kwargs)
return wrapper
return decorator