highperfocused/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-10-10 21:03:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

refac: chat requests

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-12-19 01:00:32 -08:00
parent ea0d507e23
commit 2be9e55545
11 changed files with 752 additions and 424 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -217,15 +217,19 @@ async def disconnect(sid):
def get_event_emitter(request_info):
async def __event_emitter__(event_data):
await sio.emit(
"chat-events",
{
"chat_id": request_info["chat_id"],
"message_id": request_info["message_id"],
"data": event_data,
},
to=request_info["session_id"],
)
user_id = request_info["user_id"]
session_ids = USER_POOL.get(user_id, [])
for session_id in session_ids:
await sio.emit(
"chat-events",
{
"chat_id": request_info["chat_id"],
"message_id": request_info["message_id"],
"data": event_data,
},
to=session_id,
)
return __event_emitter__