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

use async_trait::async_trait;
use synaptic_core::{CallbackHandler, RunEvent, SynapticError};

pub struct TracingCallback;

impl TracingCallback {
    pub fn new() -> Self {
        Self
    }
}

impl Default for TracingCallback {
    fn default() -> Self {
        Self::new()
    }
}

#[async_trait]
impl CallbackHandler for TracingCallback {
    async fn on_event(&self, event: RunEvent) -> Result<(), SynapticError> {
        match event {
            RunEvent::RunStarted { run_id, session_id } => {
                tracing::info!(run_id = %run_id, session_id = %session_id, "run started");
            }
            RunEvent::RunStep { run_id, step } => {
                tracing::info!(run_id = %run_id, step = step, "run step");
            }
            RunEvent::LlmCalled {
                run_id,
                message_count,
            } => {
                tracing::info!(run_id = %run_id, message_count = message_count, "LLM called");
            }
            RunEvent::ToolCalled { run_id, tool_name } => {
                tracing::info!(run_id = %run_id, tool_name = %tool_name, "tool called");
            }
            RunEvent::RunFinished { run_id, output } => {
                tracing::info!(run_id = %run_id, output_len = output.len(), "run finished");
            }
            RunEvent::RunFailed { run_id, error } => {
                tracing::error!(run_id = %run_id, error = %error, "run failed");
            }
            RunEvent::BeforeToolCall {
                run_id,
                tool_name,
                arguments,
            } => {
                tracing::info!(run_id = %run_id, tool_name = %tool_name, args_len = arguments.len(), "before tool call");
            }
            RunEvent::AfterToolCall {
                run_id,
                tool_name,
                result,
            } => {
                tracing::info!(run_id = %run_id, tool_name = %tool_name, result_len = result.len(), "after tool call");
            }
            RunEvent::BeforeMessage {
                run_id,
                message_count,
            } => {
                tracing::info!(run_id = %run_id, message_count = message_count, "before message");
            }
            RunEvent::AfterMessage {
                run_id,
                response_length,
            } => {
                tracing::info!(run_id = %run_id, response_length = response_length, "after message");
            }
        }
        Ok(())
    }
}