diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py index 288b08dd16..6ec03493f7 100644 --- a/backend/open_webui/env.py +++ b/backend/open_webui/env.py @@ -604,12 +604,10 @@ if WEBSOCKET_REDIS_OPTIONS == "": WEBSOCKET_REDIS_OPTIONS = None else: try: - WEBSOCKET_REDIS_OPTIONS = json.loads( - WEBSOCKET_REDIS_OPTIONS - ) + WEBSOCKET_REDIS_OPTIONS = json.loads(WEBSOCKET_REDIS_OPTIONS) except Exception: log.warning("Invalid WEBSOCKET_REDIS_OPTIONS, defaulting to None") - WEBSOCKET_REDIS_OPTIONS = None + WEBSOCKET_REDIS_OPTIONS = None WEBSOCKET_REDIS_URL = os.environ.get("WEBSOCKET_REDIS_URL", REDIS_URL) WEBSOCKET_REDIS_CLUSTER = ( @@ -625,14 +623,18 @@ except ValueError: WEBSOCKET_SENTINEL_HOSTS = os.environ.get("WEBSOCKET_SENTINEL_HOSTS", "") WEBSOCKET_SENTINEL_PORT = os.environ.get("WEBSOCKET_SENTINEL_PORT", "26379") -WEBSOCKET_SERVER_LOGGING = os.environ.get("WEBSOCKET_SERVER_LOGGING", "False").lower() == "true" -WEBSOCKET_SERVER_ENGINEIO_LOGGING = os.environ.get("WEBSOCKET_SERVER_LOGGING", "False").lower() == "true" +WEBSOCKET_SERVER_LOGGING = ( + os.environ.get("WEBSOCKET_SERVER_LOGGING", "False").lower() == "true" +) +WEBSOCKET_SERVER_ENGINEIO_LOGGING = ( + os.environ.get("WEBSOCKET_SERVER_LOGGING", "False").lower() == "true" +) WEBSOCKET_SERVER_PING_TIMEOUT = os.environ.get("WEBSOCKET_SERVER_PING_TIMEOUT", "20") try: WEBSOCKET_SERVER_PING_TIMEOUT = int(WEBSOCKET_SERVER_PING_TIMEOUT) except ValueError: WEBSOCKET_SERVER_PING_TIMEOUT = 20 - + WEBSOCKET_SERVER_PING_INTERVAL = os.environ.get("WEBSOCKET_SERVER_PING_INTERVAL", "25") try: WEBSOCKET_SERVER_PING_INTERVAL = int(WEBSOCKET_SERVER_PING_INTERVAL) diff --git a/backend/open_webui/models/users.py b/backend/open_webui/models/users.py index b9e40abf0b..256d3bc75e 100644 --- a/backend/open_webui/models/users.py +++ b/backend/open_webui/models/users.py @@ -345,7 +345,9 @@ class UsersTable: with get_db() as db: current_timestamp = int(datetime.datetime.now().timestamp()) today_midnight_timestamp = current_timestamp - (current_timestamp % 86400) - query = db.query(User).filter(User.last_active_at > today_midnight_timestamp) + query = db.query(User).filter( + User.last_active_at > today_midnight_timestamp + ) return query.count() def update_user_role_by_id(self, id: str, role: str) -> Optional[UserModel]: diff --git a/backend/open_webui/retrieval/vector/dbs/pgvector.py b/backend/open_webui/retrieval/vector/dbs/pgvector.py index 3ffad4fc48..85c2ef009d 100644 --- a/backend/open_webui/retrieval/vector/dbs/pgvector.py +++ b/backend/open_webui/retrieval/vector/dbs/pgvector.py @@ -209,9 +209,7 @@ class PgvectorClient(VectorDBBase): index_method = "ivfflat" if index_method == "hnsw": - index_options = ( - f"WITH (m = {PGVECTOR_HNSW_M}, ef_construction = {PGVECTOR_HNSW_EF_CONSTRUCTION})" - ) + index_options = f"WITH (m = {PGVECTOR_HNSW_M}, ef_construction = {PGVECTOR_HNSW_EF_CONSTRUCTION})" else: index_options = f"WITH (lists = {PGVECTOR_IVFFLAT_LISTS})" diff --git a/backend/open_webui/retrieval/vector/dbs/s3vector.py b/backend/open_webui/retrieval/vector/dbs/s3vector.py index bada99cde0..e2a7adfd8b 100644 --- a/backend/open_webui/retrieval/vector/dbs/s3vector.py +++ b/backend/open_webui/retrieval/vector/dbs/s3vector.py @@ -116,19 +116,18 @@ class S3VectorClient(VectorDBBase): return filtered_metadata def has_collection(self, collection_name: str) -> bool: - """ - Check if a vector index exists using direct lookup. - This avoids pagination issues with list_indexes() and is significantly faster. - """ - try: - self.client.get_index( - vectorBucketName=self.bucket_name, - indexName=collection_name - ) - return True - except Exception as e: - log.error(f"Error checking if index '{collection_name}' exists: {e}") - return False + """ + Check if a vector index exists using direct lookup. + This avoids pagination issues with list_indexes() and is significantly faster. + """ + try: + self.client.get_index( + vectorBucketName=self.bucket_name, indexName=collection_name + ) + return True + except Exception as e: + log.error(f"Error checking if index '{collection_name}' exists: {e}") + return False def delete_collection(self, collection_name: str) -> None: """ diff --git a/backend/open_webui/routers/prompts.py b/backend/open_webui/routers/prompts.py index c056489e0a..6a957f2547 100644 --- a/backend/open_webui/routers/prompts.py +++ b/backend/open_webui/routers/prompts.py @@ -53,7 +53,9 @@ async def create_new_prompt( user.id, "workspace.prompts", request.app.state.config.USER_PERMISSIONS ) or has_permission( - user.id, "workspace.prompts_import", request.app.state.config.USER_PERMISSIONS + user.id, + "workspace.prompts_import", + request.app.state.config.USER_PERMISSIONS, ) ): raise HTTPException( diff --git a/backend/open_webui/socket/main.py b/backend/open_webui/socket/main.py index 2ef556f987..f79d1dd958 100644 --- a/backend/open_webui/socket/main.py +++ b/backend/open_webui/socket/main.py @@ -70,7 +70,9 @@ if WEBSOCKET_MANAGER == "redis": redis_options=WEBSOCKET_REDIS_OPTIONS, ) else: - mgr = socketio.AsyncRedisManager(WEBSOCKET_REDIS_URL, redis_options=WEBSOCKET_REDIS_OPTIONS) + mgr = socketio.AsyncRedisManager( + WEBSOCKET_REDIS_URL, redis_options=WEBSOCKET_REDIS_OPTIONS + ) sio = socketio.AsyncServer( cors_allowed_origins=SOCKETIO_CORS_ORIGINS, async_mode="asgi", @@ -81,7 +83,7 @@ if WEBSOCKET_MANAGER == "redis": logger=WEBSOCKET_SERVER_LOGGING, ping_interval=WEBSOCKET_SERVER_PING_INTERVAL, ping_timeout=WEBSOCKET_SERVER_PING_TIMEOUT, - engineio_logger=WEBSOCKET_SERVER_ENGINEIO_LOGGING, + engineio_logger=WEBSOCKET_SERVER_ENGINEIO_LOGGING, ) else: sio = socketio.AsyncServer( @@ -93,7 +95,7 @@ else: logger=WEBSOCKET_SERVER_LOGGING, ping_interval=WEBSOCKET_SERVER_PING_INTERVAL, ping_timeout=WEBSOCKET_SERVER_PING_TIMEOUT, - engineio_logger=WEBSOCKET_SERVER_ENGINEIO_LOGGING, + engineio_logger=WEBSOCKET_SERVER_ENGINEIO_LOGGING, ) diff --git a/src/lib/components/chat/Settings/Interface.svelte b/src/lib/components/chat/Settings/Interface.svelte index 1b0d978ae5..7c3939b02c 100644 --- a/src/lib/components/chat/Settings/Interface.svelte +++ b/src/lib/components/chat/Settings/Interface.svelte @@ -396,11 +396,7 @@
-