busy prompt changes text for each goose streamed response

block · Nov 25, 2024 · 21a8cee · 21a8cee
1 parent 66a4deb
commit 21a8cee
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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