feat: extract reponse time
This commit is contained in:
Jelson Stoelben Rodrigues
167
src/main.rs
167
src/main.rs
@@ -7,7 +7,7 @@ use chrono::{self, Timelike};
|
||||
use dotenv;
|
||||
use ipaddress;
|
||||
use itertools::{self, Itertools};
|
||||
use lettre;
|
||||
use lettre::{self, message};
|
||||
use reqwest;
|
||||
use serde_json::{self, json};
|
||||
use zip;
|
||||
@@ -170,6 +170,14 @@ fn main() -> anyhow::Result<()> {
|
||||
std::fs::create_dir(format!("./evaluations/{formatted_day_before}")).expect("Failed to create directory")
|
||||
}
|
||||
|
||||
// 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();
|
||||
|
||||
@@ -190,6 +198,163 @@ fn main() -> anyhow::Result<()> {
|
||||
|
||||
println!("IDS {:?}", talk_ids);
|
||||
|
||||
// Calculate the response time in seconds
|
||||
let response_time = talk_ids
|
||||
.iter()
|
||||
.cloned()
|
||||
.map(|talk_id| {
|
||||
let talk_id_get_request = client
|
||||
.get(format!("https://{}/api/talk_histories", PIPERUN_API_URL))
|
||||
.bearer_auth(&access_token)
|
||||
.header("Content-Type", "application/json")
|
||||
.header("Accept", "application/json")
|
||||
.query(&[
|
||||
("talk_id", talk_id),
|
||||
("type", "a".to_string()),
|
||||
("only_view", "1".to_string()),
|
||||
]);
|
||||
|
||||
let talk_id_get_result = talk_id_get_request.send();
|
||||
|
||||
return talk_id_get_result;
|
||||
}).filter_map_ok(|result| {
|
||||
let json = result.json::<serde_json::Value>().expect("Failed to deserialize response to JSON").to_owned();
|
||||
let talk_histories = &json["talk_histories"];
|
||||
let data = &talk_histories["data"];
|
||||
|
||||
// Filter chats that have very few messages
|
||||
let talk_lenght = talk_histories.as_array().expect("Wrong message type received from talk histories").len();
|
||||
if talk_lenght < MINIMUM_NUMBER_OF_MESSAGES_TO_EVALUATE {return None;}
|
||||
|
||||
// Filter chats that have less that specified ammount of talks with support agent form the last queue transfer
|
||||
let found = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]");
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
match found {
|
||||
None => {return None;},
|
||||
Some(pos) => {
|
||||
let pos_found = pos.0;
|
||||
if pos_found < MINIMUM_NUMBER_OF_MESSAGES_WITH_AGENT_TO_EVALUATE {return None;}
|
||||
}
|
||||
};
|
||||
|
||||
// Filter chats that pass by multiple agents
|
||||
let number_of_agents_except_pipebot = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter()
|
||||
.map(|message| &message["user"]["name"])
|
||||
.filter(|name| name.as_str().unwrap() != "PipeBot".to_string())
|
||||
.unique()
|
||||
.count();
|
||||
if number_of_agents_except_pipebot > 1 {return None;}
|
||||
|
||||
// Filter stop notification active
|
||||
let found_stop_notification = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
let found = message.find("ATENÇÃO CLIENTE COM NOTIFICAÇÃO DE PARADA ATIVA");
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
if found_stop_notification.is_some() {return None;}
|
||||
|
||||
// Filter lack of dialogue
|
||||
let lack_of_dialogue = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
let found = message.to_uppercase().find(&"Esse atendimento está sendo encerrado devido a inexistência de diálogo".to_uppercase());
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
if lack_of_dialogue.is_some() {return None;}
|
||||
|
||||
// Filter is portal access
|
||||
let portal_link_offer = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
let found = message.to_uppercase().find(&"https://novanet.portal.7az.com.br/login".to_uppercase());
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
if portal_link_offer.is_some() {return None;}
|
||||
|
||||
// Filter no comercial agent available
|
||||
let comercial_agents_unavailable = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
let found = message.to_uppercase().find(&"Transbordo automático para a fila [NovaNet -> Atendimento -> Suporte], pois não havia agentes disponíveis na fila [NovaNet -> Atendimento -> Comercial]".to_uppercase());
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
if comercial_agents_unavailable.is_some() {return None;}
|
||||
|
||||
// Filter Bot finished chats
|
||||
if json["agent"]["user"]["name"].as_str().unwrap_or("unknown_user") == "PipeBot" {return None;}
|
||||
|
||||
return Some(json);
|
||||
})
|
||||
// .clone()
|
||||
.map(|messages| {
|
||||
let json = messages.unwrap();
|
||||
let talk_histories = &json["talk_histories"];
|
||||
|
||||
// dbg!(&talk_histories);
|
||||
|
||||
// talk_histories.as_array().unwrap().into_iter().enumerate().for_each(|(pos, message_obj)|{println!("{}: {}", pos, message_obj["message"])});
|
||||
|
||||
// find the bot transfer message
|
||||
let bot_transfer_message = talk_histories
|
||||
.as_array().expect("Wrong message type received from talk histories").into_iter()
|
||||
.enumerate()
|
||||
.filter(|(pos, message_object)| {
|
||||
let user_name = message_object["user"]["name"].as_str().expect("Failed to decode message as string");
|
||||
user_name == "PipeBot".to_string()
|
||||
}).find(|(pos, message_object)|{
|
||||
let message = message_object["message"].as_str().expect("Failed to decode message as string");
|
||||
// let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]");
|
||||
let found = message.find("Atendimento entregue da fila de espera para o agente");
|
||||
found.is_some()
|
||||
});
|
||||
|
||||
// Find first agent message sent after the last bot message
|
||||
let (pos, transfer_message) = bot_transfer_message.expect("Failed to get the transfer bot message position");
|
||||
|
||||
let msg = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter()
|
||||
.take(pos)
|
||||
.rev()
|
||||
.filter(|message| {
|
||||
message["type"] == "out".to_string() && message["user"]["name"] != "PipeBot".to_string()
|
||||
})
|
||||
.take(1).collect_vec();
|
||||
|
||||
let agent_first_message = msg[0];
|
||||
|
||||
// Calculate time difference between bot message and agent message
|
||||
let date_user_message_sent = agent_first_message["sent_at"].as_str().unwrap();
|
||||
|
||||
let format = "%Y-%m-%d %H:%M:%S";
|
||||
|
||||
let date_user_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(date_user_message_sent, format) {
|
||||
Ok(dt) => dt,
|
||||
Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")},
|
||||
};
|
||||
|
||||
let date_transfer_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(transfer_message["sent_at"].as_str().unwrap(), format) {
|
||||
Ok(dt) => dt,
|
||||
Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")},
|
||||
};
|
||||
|
||||
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();
|
||||
println!("response_time: {}s", (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32());
|
||||
|
||||
format!("{};{};{};{}", name, id, response_time, 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!");
|
||||
|
||||
talk_ids
|
||||
.iter()
|
||||
.cloned()
|
||||
|
||||
Reference in New Issue
Block a user