log recorded session at the very end

block · Feb 5, 2025 · 152ddfa · 152ddfa
1 parent 27ce15a
commit 152ddfa
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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 {