From f402957f056d6114a917cb929e1d402d759bbc1b Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 14 Jul 2025 14:35:13 +0400 Subject: [PATCH] refac --- backend/open_webui/socket/main.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/backend/open_webui/socket/main.py b/backend/open_webui/socket/main.py index 5e7988a226..861d9ae52c 100644 --- a/backend/open_webui/socket/main.py +++ b/backend/open_webui/socket/main.py @@ -107,7 +107,6 @@ if WEBSOCKET_MANAGER == "redis": # TODO: Implement Yjs document management with Redis DOCUMENTS = {} - DOCUMENT_USERS = {} clean_up_lock = RedisLock( redis_url=WEBSOCKET_REDIS_URL, @@ -124,7 +123,6 @@ else: USAGE_POOL = {} DOCUMENTS = {} # document_id -> Y.YDoc instance - DOCUMENT_USERS = {} # document_id -> set of user sids aquire_func = release_func = renew_func = lambda: True @@ -377,11 +375,9 @@ async def yjs_document_join(sid, data): "ydoc": Y.Doc(), # Create actual Yjs document "users": set(), } - DOCUMENT_USERS[document_id] = set() # Add user to document DOCUMENTS[document_id]["users"].add(sid) - DOCUMENT_USERS[document_id].add(sid) # Join Socket.IO room await sio.enter_room(sid, f"doc_{document_id}") @@ -506,9 +502,6 @@ async def yjs_document_leave(sid, data): if document_id in DOCUMENTS: DOCUMENTS[document_id]["users"].discard(sid) - if document_id in DOCUMENT_USERS: - DOCUMENT_USERS[document_id].discard(sid) - # Leave Socket.IO room await sio.leave_room(sid, f"doc_{document_id}") @@ -523,7 +516,6 @@ async def yjs_document_leave(sid, data): # If no users left, clean up the document log.info(f"Cleaning up document {document_id} as no users are left") del DOCUMENTS[document_id] - del DOCUMENT_USERS[document_id] except Exception as e: log.error(f"Error in yjs_document_leave: {e}")