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| p.actual_patch.clone())
19        .collect();
20
21    example.spec.expected_patches = expected_patches;
22    example.prompt = None;
23    example.predictions = Vec::new();
24    example.score = Vec::new();
25    example.qa = Vec::new();
26    Ok(())
27}