scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-10-05 02:41:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Merge 054a831a21 into 4ef7e35b88

Browse source
This commit is contained in:
Classic298 2026-09-30 19:21:21 +02:00 • committed by GitHub
commit 0c635060f0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
backend/open_webui/socket/main.py
View file

@ -6,6 +6,7 @@ import logging
import random
import sys
import time
import weakref
from contextlib import suppress
from typing import Any
@ -204,6 +205,7 @@ REDIS_EVENT_CHANNEL = f'{REDIS_KEY_PREFIX}:direct_completion'
EVENT_QUEUES: dict[str, asyncio.Queue] = {}
EVENT_PUBLISH_LOCK = asyncio.Lock()
DOCUMENT_UPDATE_LOCKS: weakref.WeakValueDictionary[str, asyncio.Lock] = weakref.WeakValueDictionary()
def get_session_pool_batches():
@ -840,6 +842,18 @@ async def yjs_document_state(sid, data):
@sio.on('ydoc:document:update')
async def yjs_document_update(sid, data):
try:
document_id = normalize_document_id(data['document_id'])
except Exception as e:
log.error('Error in yjs_document_update: %s', e)
return
# Keep per-document updates in arrival order
async with DOCUMENT_UPDATE_LOCKS.setdefault(document_id, asyncio.Lock()):
await handle_yjs_document_update(sid, data)
async def handle_yjs_document_update(sid, data):
"""Handle Yjs document updates"""
try:
document_id = data['document_id']