bamboo-engine 2026.8.24

Execution engine and orchestration for the Bamboo agent framework
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use chrono::{DateTime, Utc};

use bamboo_metrics::{
    MetricsCollector, RoundStatus as MetricsRoundStatus, SessionStatus as MetricsSessionStatus,
    TokenUsage as MetricsTokenUsage,
};

pub(in crate::runtime::runner) fn record_round_started(
    metrics_collector: Option<&MetricsCollector>,
    round_id: &str,
    session_id: &str,
    model_name: &str,
) {
    record_round_started_at(
        metrics_collector,
        round_id,
        session_id,
        model_name,
        Utc::now(),
    );
}

pub(in crate::runtime::runner) fn record_round_started_at(
    metrics_collector: Option<&MetricsCollector>,
    round_id: &str,
    session_id: &str,
    model_name: &str,
    started_at: DateTime<Utc>,
) {
    let Some(metrics) = metrics_collector else {
        return;
    };
    metrics.round_started(
        round_id.to_string(),
        session_id.to_string(),
        model_name.to_string(),
        started_at,
    );
}

#[allow(clippy::too_many_arguments)]
pub(in crate::runtime::runner) fn record_round_completed(
    metrics_collector: Option<&MetricsCollector>,
    round_id: &str,
    session_id: &str,
    message_count: u32,
    round_status: MetricsRoundStatus,
    round_usage: MetricsTokenUsage,
    prompt_cached_tool_outputs: u32,
    prompt_cached_tool_tokens_saved: u32,
    round_error: Option<String>,
) {
    record_round_completed_at(
        metrics_collector,
        round_id,
        session_id,
        message_count,
        round_status,
        round_usage,
        prompt_cached_tool_outputs,
        prompt_cached_tool_tokens_saved,
        round_error,
        Utc::now(),
    );
}

#[allow(clippy::too_many_arguments)]
pub(in crate::runtime::runner) fn record_round_completed_at(
    metrics_collector: Option<&MetricsCollector>,
    round_id: &str,
    session_id: &str,
    message_count: u32,
    round_status: MetricsRoundStatus,
    round_usage: MetricsTokenUsage,
    prompt_cached_tool_outputs: u32,
    prompt_cached_tool_tokens_saved: u32,
    round_error: Option<String>,
    completed_at: DateTime<Utc>,
) {
    let Some(metrics) = metrics_collector else {
        return;
    };
    metrics.round_completed(
        round_id.to_string(),
        completed_at,
        round_status,
        round_usage,
        prompt_cached_tool_outputs,
        prompt_cached_tool_tokens_saved,
        round_error,
    );
    metrics.session_message_count(session_id.to_string(), message_count, completed_at);
}

#[allow(clippy::too_many_arguments)]
pub(in crate::runtime::runner) fn record_round_and_session_error(
    metrics_collector: Option<&MetricsCollector>,
    round_id: &str,
    session_id: &str,
    message_count: u32,
    round_status: MetricsRoundStatus,
    round_usage: MetricsTokenUsage,
    round_error: Option<String>,
    session_status: MetricsSessionStatus,
) {
    let Some(metrics) = metrics_collector else {
        return;
    };

    metrics.round_completed(
        round_id.to_string(),
        Utc::now(),
        round_status,
        round_usage,
        0,
        0,
        round_error,
    );
    metrics.session_message_count(session_id.to_string(), message_count, Utc::now());
    metrics.session_completed(session_id.to_string(), session_status, Utc::now());
}