ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 13:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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:
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(
request,
{
@ -619,7 +623,9 @@ async def process_chat_response(request, response, user, events, metadata, tasks
.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(
{
@ -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(
request,
{