Fix: avoid forever loop

1 files changed, 3 insertions, 2 deletions

diff --git a/bot.go b/bot.go
index fe1e56e..d09fa63 100644
--- a/bot.go
+++ b/bot.go
@@ -684,8 +684,9 @@ out:
 			toolResp.WriteString(toolChunk)
 			tv.ScrollToEnd()
 		case <-streamDone:
-			// rrain any remaining chunks from chunkChan before exiting
-			for chunk := range chunkChan {
+			// drain any remaining chunks from chunkChan before exiting
+			for len(chunkChan) > 0 {
+				chunk := <-chunkChan
 				fmt.Fprint(tv, chunk)
 				respText.WriteString(chunk)
 				tv.ScrollToEnd()