1
0
Эх сурвалжийг харах

refactor: format

Signed-off-by: Adam Tao <tcx4c70@gmail.com>
Adam Tao 2 сар өмнө
parent
commit
7bd7559bfe

+ 1 - 3
backend/open_webui/env.py

@@ -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 - 0
backend/open_webui/internal/db.py

@@ -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 - 0
backend/open_webui/utils/misc.py

@@ -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