synaptic-callbacks 0.4.0

Callback handlers: RecordingCallback, TracingCallback, CompositeCallback
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use synaptic_callbacks::RecordingCallback;
use synaptic_core::{CallbackHandler, RunEvent};

#[tokio::test]
async fn records_events_in_order() {
    let callback = RecordingCallback::new();

    callback
        .on_event(RunEvent::RunStarted {
            run_id: "run-1".to_string(),
            session_id: "session-1".to_string(),
        })
        .await
        .expect("start event");

    callback
        .on_event(RunEvent::RunFinished {
            run_id: "run-1".to_string(),
            output: "done".to_string(),
        })
        .await
        .expect("finish event");

    let events = callback.events().await;
    assert_eq!(events.len(), 2);
    assert!(matches!(events[0], RunEvent::RunStarted { .. }));
    assert!(matches!(events[1], RunEvent::RunFinished { .. }));
}

#[tokio::test]
async fn recording_starts_empty() {
    let callback = RecordingCallback::new();
    let events = callback.events().await;
    assert!(events.is_empty());
}

#[tokio::test]
async fn recording_captures_all_event_types() {
    let callback = RecordingCallback::new();

    callback
        .on_event(RunEvent::RunStarted {
            run_id: "r1".to_string(),
            session_id: "s1".to_string(),
        })
        .await
        .unwrap();
    callback
        .on_event(RunEvent::LlmCalled {
            run_id: "r1".to_string(),
            message_count: 2,
        })
        .await
        .unwrap();
    callback
        .on_event(RunEvent::ToolCalled {
            run_id: "r1".to_string(),
            tool_name: "echo".to_string(),
        })
        .await
        .unwrap();
    callback
        .on_event(RunEvent::RunStep {
            run_id: "r1".to_string(),
            step: 1,
        })
        .await
        .unwrap();
    callback
        .on_event(RunEvent::RunFinished {
            run_id: "r1".to_string(),
            output: "done".to_string(),
        })
        .await
        .unwrap();
    callback
        .on_event(RunEvent::RunFailed {
            run_id: "r1".to_string(),
            error: "oops".to_string(),
        })
        .await
        .unwrap();

    let events = callback.events().await;
    assert_eq!(events.len(), 6);
}

#[tokio::test]
async fn recording_clone_independence() {
    let callback = RecordingCallback::new();
    let clone = callback.clone();

    callback
        .on_event(RunEvent::RunStarted {
            run_id: "r1".to_string(),
            session_id: "s1".to_string(),
        })
        .await
        .unwrap();

    // Clone shares the same inner Arc, so both see the event
    let events_original = callback.events().await;
    let events_clone = clone.events().await;
    assert_eq!(events_original.len(), 1);
    assert_eq!(events_clone.len(), 1);
}