clear context when spark return error

ai-shifu · Nov 25, 2023 · 49976e7 · 49976e7
1 parent 3ed50a8
commit 49976e7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/bots/SparkBot.js b/src/bots/SparkBot.js
@@ -86,11 +86,14 @@ export default class SparkBot extends Bot {
           );
 
           let text = "";
+          let error = false;
           source.addEventListener("message", (event) => {
             if (event.data === "<end>") {
               onUpdateResponse(callbackParam, { done: true });
               source.close();
               resolve();
+            } else if (event.data === "[error]") {
+              error = true;
             } else if (event.data.slice(-5) === "<sid>") {
               // ignore <sid> message
               return;
@@ -111,6 +114,11 @@ export default class SparkBot extends Bot {
               }
               text += partialText;
               onUpdateResponse(callbackParam, { content: text, done: false });
+
+              // clear context if error
+              if (error) {
+                this.setChatContext(null);
+              }
             }
           });