fix: After stopping content generation, messages cannot be cleared #66

close #66
This commit is contained in:
kangfenmao 2024-09-21 00:25:17 +08:00
parent ae4542ce68
commit c8b98681ef
2 changed files with 5 additions and 2 deletions

View File

@ -84,8 +84,8 @@ export default class AnthropicProvider extends BaseProvider {
}) })
.on('text', (text) => { .on('text', (text) => {
if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) { if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) {
resolve() stream.controller.abort()
return stream.controller.abort() return resolve()
} }
onChunk({ text }) onChunk({ text })
}) })

View File

@ -56,6 +56,9 @@ export async function fetchChatCompletion({
const timer = setInterval(() => { const timer = setInterval(() => {
if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) { if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) {
paused = true paused = true
message.status = 'paused'
EventEmitter.emit(EVENT_NAMES.RECEIVE_MESSAGE, message)
store.dispatch(setGenerating(false))
onResponse({ ...message, status: 'paused' }) onResponse({ ...message, status: 'paused' })
clearInterval(timer) clearInterval(timer)
} }