UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-01 08:42:14 +00:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-20 22:56:37 -08:00
parent 50db2514dc
commit 2fd7bbc259
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -252,6 +252,4 @@ def get_event_call(request_info):
def get_user_id_from_session_pool(sid): def get_user_id_from_session_pool(sid):
print("get_user_id_from_session_pool", sid)
print(SESSION_POOL.get(sid))
return SESSION_POOL.get(sid) return SESSION_POOL.get(sid)

2
backend/open_webui/utils/middleware.py
View File

@ -602,12 +602,12 @@ async def process_chat_response(request, response, user, events, metadata, tasks
if done: if done:
data = {"done": True, "content": content, "title": title} data = {"done": True, "content": content, "title": title}
# Send a webhook notification if the user is not active
if ( if (
get_user_id_from_session_pool(metadata["session_id"]) get_user_id_from_session_pool(metadata["session_id"])
is None is None
): ):
webhook_url = Users.get_user_webhook_url_by_id(user.id) webhook_url = Users.get_user_webhook_url_by_id(user.id)
print(f"webhook_url: {webhook_url}")
if webhook_url: if webhook_url:
post_webhook( post_webhook(
webhook_url, webhook_url,