From f0d301fd68459a9e960b139947e89caf3ea820ee Mon Sep 17 00:00:00 2001 From: Jelson Stoelben Rodrigues Date: Wed, 24 Sep 2025 08:12:38 -0300 Subject: [PATCH] chore: add dates used in calculation to response time file fix: response time file header not being added --- src/main.rs | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/src/main.rs b/src/main.rs index 4046bf2..5202559 100644 --- a/src/main.rs +++ b/src/main.rs @@ -173,11 +173,8 @@ fn main() -> anyhow::Result<()> { // Create the response time folder if !std::fs::exists(format!("./evaluations/{formatted_day_before}/response_time.csv")).unwrap() { let mut response_time_file = std::fs::File::create_new(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to response_time.csv"); - let header = "NOME;ID_TALK;TEMPO DE RESPOSTA;DATA"; - response_time_file.write_all(header.as_bytes()).expect("Failed to write header to file"); - drop(response_time_file) } - + // Read system prompt let prompt = std::fs::read_to_string("PROMPT.txt").unwrap(); @@ -356,16 +353,18 @@ fn main() -> anyhow::Result<()> { let response_time = (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32(); let name = agent_first_message["user"]["name"].as_str().unwrap().to_owned(); let id = json["tracking_number"].as_str().unwrap_or("").to_owned(); - let date = date_user_message_sent.to_owned(); + let bot_transfer_date = date_transfer_message_sent_parsed.to_owned(); + let user_response_date = date_user_message_sent.to_owned(); println!("response_time: {}s", (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32()); - format!("{};{};{};{}", name, id, response_time, date) + format!("{};{};{};{};{}", name, id, response_time, bot_transfer_date, user_response_date) }).reduce(|acc, e|{format!("{}\n{}",acc,e)}) .unwrap_or("".to_string()); - // Open file and append to it - let mut response_time_file = std::fs::OpenOptions::new().write(true).open(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to open response time file for write"); - response_time_file.write_all(response_time.as_bytes()).expect("Failed to write to file!"); + // Open file and write to it + let header = "NOME;ID_TALK;TEMPO DE RESPOSTA;TRANFERENCIA PELO BOT;PRIMEIRA RESPOSTA DO AGENTE"; + let mut response_time_file = std::fs::OpenOptions::new().write(true).open(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to open response time file for write"); + response_time_file.write_all(format!("{header}\n{response_time}").as_bytes()).expect("Failed to write header to file"); filtered_chats .clone()