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:
await event_emitter(
{
"type": "chat-completion",
"type": "chat:completion",
"data": event,
}
)
@ -590,7 +590,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter(
{
"type": "chat-completion",
"type": "chat:completion",
"data": data,
}
)
@ -623,7 +623,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter(
{
"type": "chat-title",
"type": "chat:title",
"data": title,
}
)
@ -658,7 +658,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
await event_emitter(
{
"type": "chat-tags",
"type": "chat:tags",
"data": tags,
}
)

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

@ -244,13 +244,13 @@
message.sources = [data];
}
}
} else if (type === 'chat-completion') {
} else if (type === 'chat:completion') {
chatCompletionEventHandler(data, message, event.chat_id);
} else if (type === 'chat-title') {
} else if (type === 'chat:title') {
chatTitle.set(data);
currentChatPage.set(1);
await chats.set(await getChatList(localStorage.token, $currentChatPage));
} else if (type === 'chat-tags') {
} else if (type === 'chat:tags') {
chat = await getChatById(localStorage.token, $chatId);
allTags.set(await getAllTags(localStorage.token));
} else if (type === 'message') {
@ -303,6 +303,17 @@
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));
}
}
};