AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-16 22:25:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

refac: socket

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-09 14:23:16 +04:00
parent d84a3c0bf4
commit ebf5cb22ab
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
backend/open_webui/socket/main.py
View File

@ -159,18 +159,19 @@ def get_models_in_use():
@sio.on("usage") @sio.on("usage")
async def usage(sid, data): async def usage(sid, data):
model_id = data["model"] if sid in SESSION_POOL:
# Record the timestamp for the last update model_id = data["model"]
current_time = int(time.time()) # Record the timestamp for the last update
current_time = int(time.time())
# Store the new usage data and task # Store the new usage data and task
USAGE_POOL[model_id] = { USAGE_POOL[model_id] = {
**(USAGE_POOL[model_id] if model_id in USAGE_POOL else {}), **(USAGE_POOL[model_id] if model_id in USAGE_POOL else {}),
sid: {"updated_at": current_time}, sid: {"updated_at": current_time},
} }
# Broadcast the usage data to all clients # Broadcast the usage data to all clients
await sio.emit("usage", {"models": get_models_in_use()}) await sio.emit("usage", {"models": get_models_in_use()})
@sio.event @sio.event
@ -278,7 +279,8 @@ async def channel_events(sid, data):
@sio.on("user-list") @sio.on("user-list")
async def user_list(sid): async def user_list(sid):
await sio.emit("user-list", {"user_ids": list(USER_POOL.keys())}) if sid in SESSION_POOL:
await sio.emit("user-list", {"user_ids": list(USER_POOL.keys())})
@sio.event @sio.event