From 21a8ceed458c7b8e0b4209f636dfd3d305482db8 Mon Sep 17 00:00:00 2001
From: Jarrod Sibbison <jsibbison@squareup.com>
Date: Tue, 26 Nov 2024 09:48:58 +1100
Subject: [PATCH] busy prompt changes text for each goose streamed response

---
 crates/goose-cli/src/session.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/crates/goose-cli/src/session.rs b/crates/goose-cli/src/session.rs
index 3a053c8532..6bdc96f52c 100644
--- a/crates/goose-cli/src/session.rs
+++ b/crates/goose-cli/src/session.rs
@@ -152,7 +152,9 @@ impl<'a> Session<'a> {
                         Some(Ok(message)) => {
                             self.messages.push(message.clone());
                             persist_messages(&self.session_file, &self.messages).unwrap_or_else(|e| eprintln!("Failed to persist messages: {}", e));
+                            self.prompt.hide_busy();
                             self.prompt.render(Box::new(message.clone()));
+                            self.prompt.show_busy();
                         }
                         Some(Err(e)) => {
                             // TODO: Handle error display through prompt