tenant.openagents/omega

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