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-19 11:07:02 -08:00
parent c3e8cd03b2
commit 4c989808d6
1 changed files with 62 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -601,7 +601,11 @@ async def process_chat_response(request, response, user, events, metadata, tasks
if message: if message:
messages = get_message_list(message_map, message.get("id")) messages = get_message_list(message_map, message.get("id"))
if TASKS.TITLE_GENERATION in tasks: if tasks:
if (
TASKS.TITLE_GENERATION in tasks
and tasks[TASKS.TITLE_GENERATION]
):
res = await generate_title( res = await generate_title(
request, request,
{ {
@ -619,7 +623,9 @@ async def process_chat_response(request, response, user, events, metadata, tasks
.get("content", message.get("content", "New Chat")) .get("content", message.get("content", "New Chat"))
) )
Chats.update_chat_title_by_id(metadata["chat_id"], title) Chats.update_chat_title_by_id(
metadata["chat_id"], title
)
await event_emitter( await event_emitter(
{ {
@ -628,7 +634,10 @@ async def process_chat_response(request, response, user, events, metadata, tasks
} }
) )
if TASKS.TAGS_GENERATION in tasks: if (
TASKS.TAGS_GENERATION in tasks
and tasks[TASKS.TAGS_GENERATION]
):
res = await generate_chat_tags( res = await generate_chat_tags(
request, request,
{ {