1use anyhow::Result;
 2
 3use crate::{PredictionProvider, example::Example};
 4
 5pub async fn run_distill(example: &mut Example) -> Result<()> {
 6    let has_repair = example
 7        .predictions
 8        .iter()
 9        .find(|p| p.provider == PredictionProvider::Repair);
10    let predictions = if let Some(has_repair) = has_repair {
11        vec![has_repair]
12    } else {
13        example.predictions.iter().collect()
14    };
15
16    let expected_patches = predictions
17        .into_iter()
18        .filter_map(|p| Some((p.actual_patch.clone()?, p.actual_cursor_offset)))
19        .collect();
20
21    example
22        .spec
23        .set_expected_patches_with_cursor_positions(expected_patches);
24    example.prompt = None;
25    example.predictions = Vec::new();
26    example.score = Vec::new();
27    example.qa = Vec::new();
28    Ok(())
29}