engram-memory 0.2.0

Core binary and shared infrastructure for Engram memory system
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use serde_json::Value;

#[derive(Clone, clap::ValueEnum)]
pub enum OutputFormat {
    Json,
    Text,
    Jsonl,
}

pub fn format_output(value: &Value, format: &OutputFormat) -> String {
    match format {
        OutputFormat::Json => format_as_json(value),
        OutputFormat::Text => format_as_text(value),
        OutputFormat::Jsonl => format_as_jsonl(value),
    }
}

fn format_as_json(value: &Value) -> String {
    serde_json::to_string_pretty(value).unwrap_or_default()
}

fn format_as_jsonl(value: &Value) -> String {
    match value {
        Value::Array(items) => items
            .iter()
            .map(|item| serde_json::to_string(item).unwrap_or_default())
            .collect::<Vec<_>>()
            .join("\n"),
        _ => serde_json::to_string(value).unwrap_or_default(),
    }
}

fn format_as_text(value: &Value) -> String {
    match value {
        Value::Object(map) => format_object_as_text(map),
        Value::Array(items) => format_array_as_text(items),
        Value::String(string) => string.clone(),
        Value::Null => String::new(),
        other => other.to_string(),
    }
}

fn format_object_as_text(map: &serde_json::Map<String, Value>) -> String {
    map.iter()
        .map(|(key, value)| format!("{key}: {}", scalar_to_text(value)))
        .collect::<Vec<_>>()
        .join("\n")
}

fn format_array_as_text(items: &[Value]) -> String {
    items
        .iter()
        .map(format_as_text)
        .collect::<Vec<_>>()
        .join("\n---\n")
}

fn scalar_to_text(value: &Value) -> String {
    match value {
        Value::String(string) => string.clone(),
        Value::Null => String::new(),
        Value::Array(_) | Value::Object(_) => serde_json::to_string(value).unwrap_or_default(),
        other => other.to_string(),
    }
}