1use anyhow::Result;
 2use serde_json::Value;
 3
 4pub fn move_edit_prediction_provider_to_edit_predictions(value: &mut Value) -> Result<()> {
 5    let Some(obj) = value.as_object_mut() else {
 6        return Ok(());
 7    };
 8
 9    let Some(features) = obj.get_mut("features") else {
10        return Ok(());
11    };
12
13    let Some(features_obj) = features.as_object_mut() else {
14        return Ok(());
15    };
16
17    let Some(provider) = features_obj.remove("edit_prediction_provider") else {
18        return Ok(());
19    };
20
21    let features_is_empty = features_obj.is_empty();
22
23    if features_is_empty {
24        obj.remove("features");
25    }
26
27    let edit_predictions = obj
28        .entry("edit_predictions")
29        .or_insert_with(|| Value::Object(Default::default()));
30
31    let Some(edit_predictions_obj) = edit_predictions.as_object_mut() else {
32        anyhow::bail!("Expected edit_predictions to be an object");
33    };
34
35    if !edit_predictions_obj.contains_key("provider") {
36        edit_predictions_obj.insert("provider".to_string(), provider);
37    }
38
39    Ok(())
40}