rig-memory 0.45.0

Conversations over time for the Rig agent framework: recent messages, compaction, and saving and resuming sessions.
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

use std::collections::BTreeMap;

use serde_json::json;

use crate::{MODEL, SavedAgent, Session, TITLE};

fn agent(id: &str, parent: Option<&str>, retired: bool) -> SavedAgent {
    SavedAgent {
        id: id.to_owned(),
        parent: parent.map(str::to_owned),
        retired,
        ..SavedAgent::default()
    }
}

/// The main agent is the first nothing spawned, by id, of those a restore
/// brings back; its model and title read by their keys, and a value of
/// another shape is an error, not a silent `None`.
#[test]
fn the_main_agent_and_its_saved_values_read_without_the_host() {
    let mut main = agent("b", None, false);
    main.saved = BTreeMap::from([
        (MODEL.to_owned(), json!("openai/gpt-5")),
        (TITLE.to_owned(), json!(["not", "a", "string"])),
    ]);
    let session = Session {
        agents: vec![
            agent("0-child", Some("b"), false),
            agent("a-gone", None, true),
            agent("a-orphan", Some("a-gone"), false),
            main,
            agent("c", None, false),
        ],
        ..Session::default()
    };
    let found = session.main_agent();
    assert_eq!(found.map(|agent| agent.id.as_str()), Some("b"));
    assert_eq!(
        found.and_then(SavedAgent::model).as_deref(),
        Some("openai/gpt-5")
    );
    assert_eq!(found.and_then(SavedAgent::title), None);
    let title = found.and_then(|agent| agent.value::<String>(TITLE));
    assert!(title.is_some_and(|title| title.is_err()));
    assert!(Session::default().main_agent().is_none());
}