fix: Add check for "[DONE]" message in QianWenBot

ai-shifu · Mar 17, 2024 · 996a900 · 996a900
1 parent fc19733
commit 996a900
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/bots/QianWenBot.js b/src/bots/QianWenBot.js
@@ -102,6 +102,8 @@ export default class QianWenBot extends Bot {
         });
 
         source.addEventListener("message", (event) => {
+          if (event.data === "[DONE]") return;
+
           if (event.data === "") {
             // Empty message usually means error
             const resp = JSON.parse(source.chunk);