llm-agent-runtime 1.74.0

Unified Tokio agent runtime -- orchestration, memory, knowledge graph, and ReAct loop in one crate
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

//! Integration test: persistence round-trip with many steps.

#[cfg(feature = "persistence")]
mod persistence_roundtrip {
    use llm_agent_runtime::persistence::{FilePersistenceBackend, PersistenceBackend};

    #[tokio::test]
    async fn test_session_checkpoint_round_trip_many_steps() {
        // Create a temporary directory.
        let dir = std::env::temp_dir()
            .join(format!("art_roundtrip_{}", uuid::Uuid::new_v4()));
        tokio::fs::create_dir_all(&dir).await.unwrap();

        struct Guard(std::path::PathBuf);
        impl Drop for Guard {
            fn drop(&mut self) {
                let _ = std::fs::remove_dir_all(&self.0);
            }
        }
        let _guard = Guard(dir.clone());

        let backend = FilePersistenceBackend::new(&dir);

        // Simulate 50 agent step checkpoints.
        for step in 0u32..50 {
            let key = format!("session:agent-1:step:{step}");
            let value = format!(
                r#"{{"step":{step},"thought":"thinking about step {step}","action":"tool_{step}"}}"#
            );
            backend.save(&key, value.as_bytes()).await.unwrap();
        }

        // Verify all 50 checkpoints can be loaded back.
        for step in 0u32..50 {
            let key = format!("session:agent-1:step:{step}");
            let loaded = backend.load(&key).await.unwrap();
            assert!(loaded.is_some(), "step {step} should exist");
            let bytes = loaded.unwrap();
            let text = std::str::from_utf8(&bytes).unwrap();
            assert!(
                text.contains(&format!("\"step\":{step}")),
                "round-trip data mismatch at step {step}"
            );
        }

        // Verify list_keys returns 50 entries.
        let keys = backend.list_keys().await.unwrap();
        assert_eq!(keys.len(), 50, "expected 50 stored checkpoints, got {}", keys.len());

        // Delete all and confirm they're gone.
        for step in 0u32..50 {
            let key = format!("session:agent-1:step:{step}");
            backend.delete(&key).await.unwrap();
            let loaded = backend.load(&key).await.unwrap();
            assert!(loaded.is_none(), "step {step} should be deleted");
        }
    }
}