diff --git a/src/main.rs b/src/main.rs index 2f51a48..ae9eda4 100644 --- a/src/main.rs +++ b/src/main.rs @@ -411,8 +411,6 @@ fn main() -> anyhow::Result<()> { }); // 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 (pos, transfer_message) = match bot_transfer_message { Some((pos, msg)) => (pos, msg), @@ -512,20 +510,27 @@ fn main() -> anyhow::Result<()> { .expect("Wrong message type received from talk histories") .iter() .rev() +//filtro para escrever somente as mensagens out no .txt + .filter(|(message_object)| { + let message_type = message_object["type"] + .as_str() + .expect("Failed to decode message as string"); + message_type == "out".to_string() + }) +//fim do filtro .map(|message_object| { let new_json_filtered = format!( "{{ - message: {}, - sent_at: {}, - type: {}, - user_name: {} -}}", + message: {}, + sent_at: {}, + type: {}, + user_name: {} + }}", message_object["message"], message_object["sent_at"], message_object["type"], message_object["user"]["name"] ); - // println!("{}", new_json_filtered); new_json_filtered }) .reduce(|acc, e| format!("{acc}\n{e}")) @@ -541,7 +546,6 @@ fn main() -> anyhow::Result<()> { serde_json::json!({ "model": OLLAMA_AI_MODEL, "prompt": format!("{prompt} \n{talk}"), - // "options": serde_json::json!({"temperature": 0.1}), "stream": false, }) .to_string(),