Browse Source

Merge pull request #1475 from PaRaD1SE98/main

remove error messages in toBeSummarizedMsgs
Yifei Zhang 1 year ago
parent
commit
75488e4bd5
1 changed files with 7 additions and 4 deletions
  1. 7 4
      app/store/chat.ts

+ 7 - 4
app/store/chat.ts

@@ -402,14 +402,17 @@ export const useChatStore = create<ChatStore>()(
 
       summarizeSession() {
         const session = get().currentSession();
+        
+        // remove error messages if any
+        const cleanMessages = session.messages.filter((msg) => !msg.isError);
 
         // should summarize topic after chating more than 50 words
         const SUMMARIZE_MIN_LEN = 50;
         if (
           session.topic === DEFAULT_TOPIC &&
-          countMessages(session.messages) >= SUMMARIZE_MIN_LEN
+          countMessages(cleanMessages) >= SUMMARIZE_MIN_LEN
         ) {
-          requestWithPrompt(session.messages, Locale.Store.Prompt.Topic, {
+          requestWithPrompt(cleanMessages, Locale.Store.Prompt.Topic, {
             model: "gpt-3.5-turbo",
           }).then((res) => {
             get().updateCurrentSession(
@@ -420,10 +423,10 @@ export const useChatStore = create<ChatStore>()(
         }
 
         const modelConfig = session.mask.modelConfig;
-        let toBeSummarizedMsgs = session.messages.slice(
+        let toBeSummarizedMsgs = cleanMessages.slice(
           session.lastSummarizeIndex,
         );
-
+  
         const historyMsgLength = countMessages(toBeSummarizedMsgs);
 
         if (historyMsgLength > modelConfig?.max_tokens ?? 4000) {