Merge pull request #246 from Guocork/fixed-issue-206

fix-issue-206
moxin-org · Sep 9, 2024 · 32bed45 · 32bed45
2 parents 6714258 + c47bab2
commit 32bed45
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/data/chats/mod.rs b/src/data/chats/mod.rs
@@ -107,6 +107,11 @@ impl Chats {
     pub fn cancel_chat_streaming(&mut self) {
         if let Some(chat) = self.get_current_chat() {
             chat.borrow_mut().cancel_streaming(self.backend.as_ref());
+            let mut chat = self.get_current_chat().unwrap().borrow_mut();
+            let message = chat.messages.last_mut().unwrap();
+            if message.content.trim().is_empty() {
+                chat.messages.pop();
+            }
         }
     }