Compare commits

...

1 Commits

Author SHA1 Message Date
Antonio Scandurra
9d3af4d2d2 WIP 2025-05-06 20:25:59 +02:00
2 changed files with 10 additions and 4 deletions

View File

@@ -536,6 +536,7 @@ impl EditAgent {
let request = LanguageModelRequest {
messages,
// temperature: Some(0.5),
..Default::default()
};
Ok(self.model.stream_completion_text(request, cx).await?.stream)

View File

@@ -1033,7 +1033,7 @@ impl EvalAssertion {
fn eval(iterations: usize, expected_pass_ratio: f32, mut eval: EvalInput) {
let mut evaluated_count = 0;
report_progress(evaluated_count, iterations);
report_progress(evaluated_count, 0, iterations);
let (tx, rx) = mpsc::channel();
@@ -1075,7 +1075,7 @@ fn eval(iterations: usize, expected_pass_ratio: f32, mut eval: EvalInput) {
}
evaluated_count += 1;
report_progress(evaluated_count, iterations);
report_progress(evaluated_count, failed_count, iterations);
}
let actual_pass_ratio = (iterations - failed_count) as f32 / iterations as f32;
@@ -1146,8 +1146,13 @@ impl Display for EvalOutput {
}
}
fn report_progress(evaluated_count: usize, iterations: usize) {
print!("\r\x1b[KEvaluated {}/{}", evaluated_count, iterations);
fn report_progress(evaluated_count: usize, failed_count: usize, iterations: usize) {
let failing_rate = failed_count as f32 / evaluated_count as f32;
let passing_rate = ((1. - failing_rate) * 100.).round() as usize;
print!(
"\r\x1b[KEvaluated {}/{} ({}% passing)",
evaluated_count, iterations, passing_rate
);
std::io::stdout().flush().unwrap();
}