From c8b98681efbfae83293ed68d07e5a87f51304820 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Sat, 21 Sep 2024 00:25:17 +0800
Subject: [PATCH] fix: After stopping content generation, messages cannot be
 cleared #66

close #66
---
 src/renderer/src/providers/AnthropicProvider.ts | 4 ++--
 src/renderer/src/services/api.ts                | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/providers/AnthropicProvider.ts b/src/renderer/src/providers/AnthropicProvider.ts
index 838021bda8..0a6f29e13d 100644
--- a/src/renderer/src/providers/AnthropicProvider.ts
+++ b/src/renderer/src/providers/AnthropicProvider.ts
@@ -84,8 +84,8 @@ export default class AnthropicProvider extends BaseProvider {
         })
         .on('text', (text) => {
           if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) {
-            resolve()
-            return stream.controller.abort()
+            stream.controller.abort()
+            return resolve()
           }
           onChunk({ text })
         })
diff --git a/src/renderer/src/services/api.ts b/src/renderer/src/services/api.ts
index 6722d19d8f..9184d2dc4c 100644
--- a/src/renderer/src/services/api.ts
+++ b/src/renderer/src/services/api.ts
@@ -56,6 +56,9 @@ export async function fetchChatCompletion({
   const timer = setInterval(() => {
     if (window.keyv.get(EVENT_NAMES.CHAT_COMPLETION_PAUSED)) {
       paused = true
+      message.status = 'paused'
+      EventEmitter.emit(EVENT_NAMES.RECEIVE_MESSAGE, message)
+      store.dispatch(setGenerating(false))
       onResponse({ ...message, status: 'paused' })
       clearInterval(timer)
     }