Skip to content

Commit 8d055b2

Browse files
committed
fix: guard chat:tasks:cancel handler with message_id check
1 parent 05afed4 commit 8d055b2

1 file changed

Lines changed: 8 additions & 5 deletions

File tree

src/lib/components/chat/Chat.svelte

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -438,11 +438,14 @@
438438
} else if (type === 'chat:completion') {
439439
chatCompletionEventHandler(data, message, event.chat_id);
440440
} else if (type === 'chat:tasks:cancel') {
441-
taskIds = null;
442-
const responseMessage = history.messages[history.currentId];
443-
// Set all response messages to done
444-
for (const messageId of history.messages[responseMessage.parentId].childrenIds) {
445-
history.messages[messageId].done = true;
441+
if (event.message_id === history.currentId) {
442+
taskIds = null;
443+
// Set all response messages to done
444+
for (const messageId of history.messages[message.parentId].childrenIds) {
445+
history.messages[messageId].done = true;
446+
}
447+
} else {
448+
message.done = true;
446449
}
447450
} else if (type === 'chat:message:delta' || type === 'message') {
448451
message.content += data.content;

0 commit comments

Comments
 (0)