From 00963f4bf77682f50bf7ec58938114a10a8293b7 Mon Sep 17 00:00:00 2001
From: Tianzhou Chen <t@bytebase.com>
Date: Tue, 2 May 2023 12:14:45 +0800
Subject: [PATCH] fix: should only add message with DONE status

---
 src/components/ConversationView/index.tsx | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/components/ConversationView/index.tsx b/src/components/ConversationView/index.tsx
index 76da933..f286b83 100644
--- a/src/components/ConversationView/index.tsx
+++ b/src/components/ConversationView/index.tsx
@@ -176,17 +176,20 @@ const ConversationView = () => {
       prompt = promptGenerator(schema);
     }
 
+    // Sliding window to add messages with DONE status all the way back up to MAX_TOKENS
     let usageMessageList: Message[] = [];
     let formatedMessageList = [];
     for (let i = messageList.length - 1; i >= 0; i--) {
       const message = messageList[i];
-      if (tokens < MAX_TOKENS) {
-        tokens += countTextTokens(message.content);
-        usageMessageList.unshift(message);
-        formatedMessageList.unshift({
-          role: message.creatorRole,
-          content: message.content,
-        });
+      if (message.status === "DONE") {
+        if (tokens < MAX_TOKENS) {
+          tokens += countTextTokens(message.content);
+          usageMessageList.unshift(message);
+          formatedMessageList.unshift({
+            role: message.creatorRole,
+            content: message.content,
+          });
+        }
       }
     }
     usageMessageList.unshift({