aidens-budget-kit 0.1.0

Budget, retry, fanout, and stop-rule guards for agent 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

//! Budget, retry, fanout, and stop-rule guards.

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct BudgetV1 {
    pub max_tool_calls: u32,
    pub max_retries: u32,
    pub max_turn_millis: u64,
}

impl Default for BudgetV1 {
    fn default() -> Self {
        Self {
            max_tool_calls: 8,
            max_retries: 2,
            max_turn_millis: 30_000,
        }
    }
}

impl BudgetV1 {
    pub fn allows_tool_call(&self, tool_calls_so_far: u32, requested_tool_calls: u32) -> bool {
        tool_calls_so_far.saturating_add(requested_tool_calls) <= self.max_tool_calls
    }

    pub fn allows_retry(&self, retries_so_far: u32) -> bool {
        retries_so_far < self.max_retries
    }

    pub fn deadline_exceeded(&self, elapsed_millis: u64) -> bool {
        elapsed_millis > self.max_turn_millis
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn tool_call_budget_is_cumulative() {
        let budget = BudgetV1 {
            max_tool_calls: 2,
            max_retries: 1,
            max_turn_millis: 1000,
        };

        assert!(budget.allows_tool_call(1, 1));
        assert!(!budget.allows_tool_call(2, 1));
    }

    #[test]
    fn deadline_check_is_explicit() {
        let budget = BudgetV1 {
            max_tool_calls: 1,
            max_retries: 1,
            max_turn_millis: 10,
        };

        assert!(!budget.deadline_exceeded(10));
        assert!(budget.deadline_exceeded(11));
    }
}