fix: update flaky log test (#1295)

This commit is contained in:
Yingjie He
2025-02-19 16:49:32 -08:00
committed by GitHub
parent 26ce0e6706
commit 4b94906bb6
2 changed files with 83 additions and 77 deletions
+13 -6
View File
@@ -8,6 +8,7 @@ mod thinking;
pub use builder::build_session;
use anyhow::Result;
use etcetera::choose_app_strategy;
use goose::agents::extension::{Envs, ExtensionConfig};
use goose::agents::Agent;
use goose::message::{Message, MessageContent};
@@ -168,12 +169,18 @@ impl Session {
}
// Log usage and cleanup
let usage = self.agent.usage().await;
log_usage(self.session_file.to_string_lossy().to_string(), usage);
println!(
"\nClosing session. Recorded to {}",
self.session_file.display()
);
if let Ok(home_dir) = choose_app_strategy(crate::APP_STRATEGY.clone()) {
let usage = self.agent.usage().await;
log_usage(
home_dir,
self.session_file.to_string_lossy().to_string(),
usage,
);
println!(
"\nClosing session. Recorded to {}",
self.session_file.display()
);
}
Ok(())
}