From afea99e891ff7e1c6cb45fddf3466eb18c78ae2b Mon Sep 17 00:00:00 2001
From: Salman Mohammed <smohammed@squareup.com>
Date: Wed, 5 Feb 2025 10:06:44 -0500
Subject: [PATCH] log recorded session at the very end

---
 crates/goose-cli/src/session.rs | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/crates/goose-cli/src/session.rs b/crates/goose-cli/src/session.rs
index 7b7d9644e3..df6e6f4382 100644
--- a/crates/goose-cli/src/session.rs
+++ b/crates/goose-cli/src/session.rs
@@ -309,16 +309,15 @@ We've removed the conversation up to the most recent user message
     }
 
     async fn close_session(&mut self) {
+        self.prompt.render(raw_message("Closing session..."));
+
+        let usage = self.agent.usage().await;
+        log_usage(self.session_file.to_string_lossy().to_string(), usage);
+
         self.prompt.render(raw_message(
-            format!(
-                "Closing session. Recorded to {}\n",
-                self.session_file.display()
-            )
-            .as_str(),
+            format!("Recorded to {}\n", self.session_file.display()).as_str(),
         ));
         self.prompt.close();
-        let usage = self.agent.usage().await;
-        log_usage(self.session_file.to_string_lossy().to_string(), usage);
     }
 
     pub fn session_file(&self) -> PathBuf {