scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-10-18 04:43:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2026-10-16 01:43:50 +04:00
parent 1895f5ffb9
commit 8191ac30dd
2 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -235,9 +235,11 @@ def get_session_pool_batches():
async def publish_chat_presence(chat_id: str):
# Room maps only contain this worker's sockets; the session pool spans workers.
lock = (
REDIS.lock(f'{REDIS_KEY_PREFIX}:chat:{chat_id}:presence:lock',
REDIS.lock(
f'{REDIS_KEY_PREFIX}:chat:{chat_id}:presence:lock',
timeout=WEBSOCKET_REDIS_LOCK_TIMEOUT,
blocking_timeout=WEBSOCKET_REDIS_LOCK_TIMEOUT)
blocking_timeout=WEBSOCKET_REDIS_LOCK_TIMEOUT,
)
if REDIS is not None
else SOCKET_EVENT_LOCKS.setdefault(('chat', chat_id), asyncio.Lock())
)
@ -298,7 +300,10 @@ async def periodic_session_pool_cleanup():
if expired:
log.warning('Reaping %d orphaned session(s) from the session pool', len(expired))
chat_ids = {
chat_id for sid, entry in batch if sid in expired for chat_id in entry.get('chat_ids', [])
chat_id
for sid, entry in batch
if sid in expired
for chat_id in entry.get('chat_ids', [])
}
if WEBSOCKET_MANAGER == 'redis':
SESSION_POOL.delete_many(*expired)
@ -634,7 +639,8 @@ async def refresh_chat_access(chat_id=None):
current = SESSION_POOL.get(sid)
if current:
SESSION_POOL[sid] = {
**current, 'chat_ids': [id for id in current.get('chat_ids', []) if id != cid]
**current,
'chat_ids': [id for id in current.get('chat_ids', []) if id != cid],
}
await sio.emit(
'events', {'chat_id': cid, 'shared': True, 'data': {'type': 'chat:access', 'data': {}}}, to=sid

2
src/lib/components/common/ChatParticipants.svelte
View file

@ -48,7 +48,7 @@
{/snippet}
<div class="flex shrink-0 items-center gap-1" data-chat-participants draggable="false">
<div class="flex items-center -space-x-1 rtl:space-x-reverse">
<div class="flex items-center {mode === 'presence' ? '-space-x-2' : '-space-x-1'} rtl:space-x-reverse">
{#each people.slice(0, 3) as person (person.id)}
<div class="relative rounded-full hover:z-10 focus-within:z-10">
<ProfilePreview