Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: chat completion notification

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-12-19 15:14:09 -08:00
parent f133353734
commit d9573befff
5 changed files with 96 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -582,9 +582,10 @@ async def process_chat_response(request, response, user, events, metadata, tasks
except Exception as e:
done = "data: [DONE]" in line
title = Chats.get_chat_title_by_id(metadata["chat_id"])
if done:
data = {"done": True}
data = {"done": True, "content": content, "title": title}
else:
continue
@@ -602,27 +603,41 @@ async def process_chat_response(request, response, user, events, metadata, tasks
messages = get_message_list(message_map, message.get("id"))
if tasks:
if (
TASKS.TITLE_GENERATION in tasks
and tasks[TASKS.TITLE_GENERATION]
):
res = await generate_title(
request,
{
"model": message["model"],
"messages": messages,
"chat_id": metadata["chat_id"],
},
user,
)
if res and isinstance(res, dict):
title = (
res.get("choices", [])[0]
.get("message", {})
.get("content", message.get("content", "New Chat"))
if TASKS.TITLE_GENERATION in tasks:
if tasks[TASKS.TITLE_GENERATION]:
res = await generate_title(
request,
{
"model": message["model"],
"messages": messages,
"chat_id": metadata["chat_id"],
},
user,
)
if res and isinstance(res, dict):
title = (
res.get("choices", [])[0]
.get("message", {})
.get(
"content",
message.get("content", "New Chat"),
)
)
Chats.update_chat_title_by_id(
metadata["chat_id"], title
)
await event_emitter(
{
"type": "chat:title",
"data": title,
}
)
elif len(messages) == 2:
title = messages[0].get("content", "New Chat")
Chats.update_chat_title_by_id(
metadata["chat_id"], title
)
@@ -630,7 +645,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter(
{
"type": "chat:title",
"data": title,
"data": message.get("content", "New Chat"),
}
)