OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-12 20:15:00 -07:00
parent 5844779384
commit fa61065c1e
1 changed files with 31 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -130,7 +130,7 @@
currentId: null currentId: null
}; };
let taskId = null; let taskIds = null;
// Chat Input // Chat Input
let prompt = ''; let prompt = '';
@ -818,8 +818,14 @@
await tick(); await tick();
if (history.currentId) { if (history.currentId) {
history.messages[history.currentId].done = true; for (const message of Object.values(history.messages)) {
if (message.role === 'assistant') {
message.done = true;
} }
}
}
taskIds = chat?.task_ids ?? null;
await tick(); await tick();
return true; return true;
@ -891,7 +897,7 @@
} }
} }
taskId = null; taskIds = null;
}; };
const chatActionHandler = async (chatId, actionId, modelId, responseMessageId, event = null) => { const chatActionHandler = async (chatId, actionId, modelId, responseMessageId, event = null) => {
@ -1649,7 +1655,11 @@
if (res.error) { if (res.error) {
await handleOpenAIError(res.error, responseMessage); await handleOpenAIError(res.error, responseMessage);
} else { } else {
taskId = res.task_id; if (taskIds) {
taskIds.push(res.task_id);
} else {
taskIds = [res.task_id];
}
} }
} }
@ -1700,17 +1710,22 @@
}; };
const stopResponse = async () => { const stopResponse = async () => {
if (taskId) { if (taskIds) {
for (const taskId of taskIds) {
const res = await stopTask(localStorage.token, taskId).catch((error) => { const res = await stopTask(localStorage.token, taskId).catch((error) => {
toast.error(`${error}`); toast.error(`${error}`);
return null; return null;
}); });
}
if (res) { taskIds = null;
taskId = null;
const responseMessage = history.messages[history.currentId]; const responseMessage = history.messages[history.currentId];
responseMessage.done = true;
// Set all response messages to done
for (const messageId of history.messages[responseMessage.parentId].childrenIds) {
history.messages[messageId].done = true;
}
history.messages[history.currentId] = responseMessage; history.messages[history.currentId] = responseMessage;
@ -1718,7 +1733,6 @@
scrollToBottom(); scrollToBottom();
} }
} }
}
}; };
const submitMessage = async (parentId, prompt) => { const submitMessage = async (parentId, prompt) => {