fix: Fix rare occasions where streaming continues infinitely

tjtanjin · May 18, 2024 · a8ba8a3 · a8ba8a3
1 parent a1057e8
commit a8ba8a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/components/ChatBotContainer.tsx b/src/components/ChatBotContainer.tsx
@@ -445,7 +445,7 @@ const ChatBotContainer = ({ flow }: { flow: Flow }) => {
 				});
 
 				// when streaming is done, remove task, unlock text area, and resolve the promise
-				if (streamIndex === endStreamIndex) {
+				if (streamIndex >= endStreamIndex) {
 					clearInterval(intervalId);
 					resolve();
 				}