fix: fix resend message not work as expected when followed by an erro…

…r message
lisiur · Apr 14, 2023 · 292dcb2 · 292dcb2
1 parent 49ab6f7
commit 292dcb2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/web/src/models/chat.ts b/web/src/models/chat.ts
@@ -216,8 +216,15 @@ export class Chat {
 
     const userMessage = this.messages[index] as UserMessage;
 
-    if (index === this.messages.length - 1) {
+    if (
+      index === this.messages.length - 1 ||
+      (index === this.messages.length - 2 &&
+        this.messages[index + 1] instanceof ErrorMessage)
+    ) {
       // We can reuse the message if it is the last message
+      // or is the second last message and the last message is an error message
+
+      this.messages.length = index + 1;
 
       userMessage.delivered = false;
       userMessage.finished = null;