chidori 0.1.28

A Rust library for executing prompt-graph models
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use serde_json::Value as JsonValue;
use std::collections::VecDeque;
use prompt_graph_core::proto::SerializedValue;
use prompt_graph_core::templates::json_value_to_serialized_value;

pub fn json_value_to_paths(
    d: &JsonValue,
) -> Vec<(Vec<String>, SerializedValue)> {
    let mut paths = Vec::new();
    let mut queue: VecDeque<(Vec<String>, &JsonValue)> = VecDeque::new();
    queue.push_back((Vec::new(), d));

    while let Some((mut path, dict)) = queue.pop_front() {
        match dict {
            JsonValue::Object(map) => {
                for (key, val) in map {
                    let key_str = key.clone();
                    path.push(key_str.clone());
                    match val {
                        JsonValue::Object(_) => {
                            queue.push_back((path.clone(), val));
                        },
                        _ => {
                            paths.push((path.clone(), json_value_to_serialized_value(&val)));
                        }
                    }
                    path.pop();
                }
            },
            JsonValue::Array(arr) => {
                for (i, val) in arr.iter().enumerate() {
                    path.push(i.to_string());
                    match val {
                        JsonValue::Object(_) => {
                            queue.push_back((path.clone(), val));
                        },
                        _ => {
                            paths.push((path.clone(), json_value_to_serialized_value(&val)));
                        }
                    }
                    path.pop();
                }
            },
            _ => panic!("Root should be a JSON object but was {:?}", d),
        }
    }

    paths
}