Adição de filtro de type=out no .txt
This commit is contained in:
10
src/main.rs
10
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(),
|
||||
|
||||
Reference in New Issue
Block a user