diff --git a/src/main.rs b/src/main.rs index f6851fd..85e8edf 100644 --- a/src/main.rs +++ b/src/main.rs @@ -232,7 +232,6 @@ fn main() -> anyhow::Result<()> { )) .expect("Failed to create response_time.csv"); } - */ // Read system prompt @@ -507,6 +506,14 @@ 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!( "{{ @@ -535,7 +542,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(),