newt-core 0.8.0

Newt-Agent core types, errors, and the NeMoCode-style tier router
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

use super::*;

#[tokio::test]
async fn rolling_window_can_retain_originals_without_an_integer_overflow() {
    let mut rw = RollingWindow::new(usize::MAX);
    rw.sync_turn("question", "answer", &dummy_metrics()).await;
    let messages = rw.build_messages("system", "next");
    assert_eq!(messages.len(), 4);
    assert_eq!(messages[1].content, "question");
    assert_eq!(messages[2].content, "answer");
}

#[tokio::test]
async fn rolling_window_empty_produces_two_messages() {
    let rw = RollingWindow::new(5);
    let msgs = rw.build_messages("sys", "hello");
    assert_eq!(msgs.len(), 2);
    assert_eq!(msgs[0].role, Role::System);
    assert_eq!(msgs[1].role, Role::User);
    assert_eq!(msgs[1].content, "hello");
}
#[tokio::test]
async fn rolling_window_includes_history() {
    let mut rw = RollingWindow::new(5);
    rw.sync_turn("q1", "a1", &dummy_metrics()).await;
    rw.sync_turn("q2", "a2", &dummy_metrics()).await;
    let msgs = rw.build_messages("sys", "q3");
    // system + (q1,a1) + (q2,a2) + q3 = 6
    assert_eq!(msgs.len(), 6);
    assert_eq!(msgs[1].content, "q1");
    assert_eq!(msgs[2].content, "a1");
    assert_eq!(msgs[5].content, "q3");
}
#[tokio::test]
async fn rolling_window_caps_at_max_turns() {
    let mut rw = RollingWindow::new(2);
    for i in 0..5u32 {
        rw.sync_turn(&format!("q{i}"), &format!("a{i}"), &dummy_metrics())
            .await;
    }
    let msgs = rw.build_messages("sys", "q5");
    // system + 2 turns * 2 messages + current = 6
    assert_eq!(msgs.len(), 6);
    // The last 2 turns should be q3/a3 and q4/a4
    assert_eq!(msgs[1].content, "q3");
    assert_eq!(msgs[3].content, "q4");
    assert_eq!(msgs[5].content, "q5");
}
#[tokio::test]
async fn rolling_window_usage_reports_correctly() {
    let mut rw = RollingWindow::new(10);
    rw.sync_turn("q", "a", &dummy_metrics()).await;
    rw.sync_turn("q", "a", &dummy_metrics()).await;
    let (label, cur, max) = rw.usage().unwrap();
    assert_eq!(label, "turns");
    assert_eq!(cur, 2);
    assert_eq!(max, 10);
}
#[tokio::test]
async fn rolling_window_on_session_end_noop() {
    let mut rw = RollingWindow::new(5);
    rw.on_session_end(&[]).await; // must not panic
}