UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-29 07:12:07 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-19 01:05:47 -08:00
parent 64fe2de962
commit 0d5ce23885
2 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -541,7 +541,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
for event in events: for event in events:
await event_emitter( await event_emitter(
{ {
"type": "chat-completion", "type": "chat:completion",
"data": event, "data": event,
} }
) )
@ -590,7 +590,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter( await event_emitter(
{ {
"type": "chat-completion", "type": "chat:completion",
"data": data, "data": data,
} }
) )
@ -623,7 +623,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter( await event_emitter(
{ {
"type": "chat-title", "type": "chat:title",
"data": title, "data": title,
} }
) )
@ -658,7 +658,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter( await event_emitter(
{ {
"type": "chat-tags", "type": "chat:tags",
"data": tags, "data": tags,
} }
) )

17
src/lib/components/chat/Chat.svelte
View File

@ -244,13 +244,13 @@
message.sources = [data]; message.sources = [data];
} }
} }
} else if (type === 'chat-completion') { } else if (type === 'chat:completion') {
chatCompletionEventHandler(data, message, event.chat_id); chatCompletionEventHandler(data, message, event.chat_id);
} else if (type === 'chat-title') { } else if (type === 'chat:title') {
chatTitle.set(data); chatTitle.set(data);
currentChatPage.set(1); currentChatPage.set(1);
await chats.set(await getChatList(localStorage.token, $currentChatPage)); await chats.set(await getChatList(localStorage.token, $currentChatPage));
} else if (type === 'chat-tags') { } else if (type === 'chat:tags') {
chat = await getChatById(localStorage.token, $chatId); chat = await getChatById(localStorage.token, $chatId);
allTags.set(await getAllTags(localStorage.token)); allTags.set(await getAllTags(localStorage.token));
} else if (type === 'message') { } else if (type === 'message') {
@ -303,6 +303,17 @@
history.messages[event.message_id] = message; history.messages[event.message_id] = message;
} }
} else {
await tick();
const type = event?.data?.type ?? null;
const data = event?.data?.data ?? null;
if (type === 'chat:title') {
currentChatPage.set(1);
await chats.set(await getChatList(localStorage.token, $currentChatPage));
} else if (type === 'chat:tags') {
allTags.set(await getAllTags(localStorage.token));
}
} }
}; };