a3s-code-core 5.2.4

A3S Code Core - Embeddable AI agent library with tool execution
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
66
67
68
69
70
71
72
73

use super::tests::test_config;
use super::*;
use crate::store::MemorySessionStore;

fn replacement_options(store: Arc<MemorySessionStore>, model: &str) -> SessionOptions {
    SessionOptions::new()
        .with_session_store(store)
        .with_memory(Arc::new(a3s_memory::InMemoryStore::new()))
        .with_model(model)
}

#[tokio::test]
async fn replacement_is_atomic_and_keeps_the_session_id() {
    let agent = Agent::from_config(test_config()).await.unwrap();
    let store = Arc::new(MemorySessionStore::new());
    let session_id = "atomic-session-replacement";
    let current = agent
        .session_async(
            "/tmp/atomic-session-replacement",
            Some(
                replacement_options(Arc::clone(&store), "anthropic/claude-sonnet-4-20250514")
                    .with_session_id(session_id),
            ),
        )
        .await
        .unwrap();

    let replacement = agent
        .replace_session_async(
            &current,
            replacement_options(Arc::clone(&store), "openai/gpt-4o"),
        )
        .await
        .unwrap();

    assert!(current.is_closed());
    assert!(!replacement.is_closed());
    assert_eq!(replacement.session_id(), session_id);
    assert_eq!(replacement.model_name, "openai/gpt-4o");
    assert_eq!(agent.list_sessions().await, vec![session_id.to_string()]);
}

#[tokio::test]
async fn failed_replacement_leaves_the_current_session_live() {
    let agent = Agent::from_config(test_config()).await.unwrap();
    let store = Arc::new(MemorySessionStore::new());
    let session_id = "failed-session-replacement";
    let current = agent
        .session_async(
            "/tmp/failed-session-replacement",
            Some(
                replacement_options(Arc::clone(&store), "anthropic/claude-sonnet-4-20250514")
                    .with_session_id(session_id),
            ),
        )
        .await
        .unwrap();

    let error = agent
        .replace_session_async(
            &current,
            SessionOptions::new()
                .with_memory(Arc::new(a3s_memory::InMemoryStore::new()))
                .with_model("openai/gpt-4o"),
        )
        .await
        .expect_err("replacement without a session store must fail");

    assert!(error.to_string().contains("session_store"));
    assert!(!current.is_closed());
    current.save().await.unwrap();
    assert_eq!(agent.list_sessions().await, vec![session_id.to_string()]);
}