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 { let request = LanguageModelRequest {
messages, messages,
// temperature: Some(0.5),
..Default::default() ..Default::default()
}; };
Ok(self.model.stream_completion_text(request, cx).await?.stream) 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) { fn eval(iterations: usize, expected_pass_ratio: f32, mut eval: EvalInput) {
let mut evaluated_count = 0; let mut evaluated_count = 0;
report_progress(evaluated_count, iterations); report_progress(evaluated_count, 0, iterations);
let (tx, rx) = mpsc::channel(); let (tx, rx) = mpsc::channel();
@@ -1075,7 +1075,7 @@ fn eval(iterations: usize, expected_pass_ratio: f32, mut eval: EvalInput) {
} }
evaluated_count += 1; 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; 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) { fn report_progress(evaluated_count: usize, failed_count: usize, iterations: usize) {
print!("\r\x1b[KEvaluated {}/{}", evaluated_count, iterations); 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(); std::io::stdout().flush().unwrap();
} }