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

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

diff --git a/crates/goose-cli/src/session.rs b/crates/goose-cli/src/session.rs
index 7b7d9644e3..dfd0b08106 100644
--- a/crates/goose-cli/src/session.rs
+++ b/crates/goose-cli/src/session.rs
@@ -309,6 +309,9 @@ We've removed the conversation up to the most recent user message
     }
 
     async fn close_session(&mut self) {
+        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",
@@ -317,8 +320,6 @@ We've removed the conversation up to the most recent user message
             .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 {