distri-types 0.3.8

Shared message, tool, and config types for Distri
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
122
123
124
125
126
127
128

use crate::events::{AgentEventType, CompactionTier};
use crate::execution::ContextBudget;

#[test]
fn context_budget_update_event_serializes() {
    let event = AgentEventType::ContextBudgetUpdate {
        budget: ContextBudget {
            system_prompt_static_tokens: 3000,
            context_window_size: 200_000,
            ..Default::default()
        },
        is_warning: false,
        is_critical: false,
    };
    let json = serde_json::to_string(&event).unwrap();
    let decoded: AgentEventType = serde_json::from_str(&json).unwrap();
    match decoded {
        AgentEventType::ContextBudgetUpdate { budget, .. } => {
            assert_eq!(budget.system_prompt_static_tokens, 3000)
        }
        _ => panic!("expected ContextBudgetUpdate"),
    }
}

#[test]
fn step_completed_carries_budget() {
    let event = AgentEventType::StepCompleted {
        step_id: "s1".into(),
        success: true,
        usage: None,
        context_budget: Some(ContextBudget {
            conversation_tokens: 50000,
            context_window_size: 200_000,
            ..Default::default()
        }),
    };
    let json = serde_json::to_string(&event).unwrap();
    assert!(json.contains("context_budget"));
}

#[test]
fn run_finished_carries_budget() {
    let event = AgentEventType::RunFinished {
        success: true,
        total_steps: 3,
        failed_steps: 0,
        usage: None,
        context_budget: Some(ContextBudget {
            conversation_tokens: 80000,
            context_window_size: 200_000,
            ..Default::default()
        }),
    };
    let json = serde_json::to_string(&event).unwrap();
    assert!(json.contains("context_budget"));
}

#[test]
fn compaction_event_carries_budget() {
    let event = AgentEventType::ContextCompaction {
        tier: CompactionTier::Trim,
        tokens_before: 8000,
        tokens_after: 3200,
        entries_affected: 5,
        context_limit: 8000,
        usage_ratio: 0.65,
        summary: None,
        reinjected_skills: vec![],
        context_budget: Some(ContextBudget {
            conversation_tokens: 3200,
            context_window_size: 200_000,
            ..Default::default()
        }),
    };
    let json = serde_json::to_string(&event).unwrap();
    let decoded: AgentEventType = serde_json::from_str(&json).unwrap();
    match decoded {
        AgentEventType::ContextCompaction { context_budget, .. } => {
            assert!(context_budget.is_some());
            assert_eq!(context_budget.unwrap().conversation_tokens, 3200);
        }
        _ => panic!("expected ContextCompaction"),
    }
}

#[test]
fn event_types_exhaustive_match() {
    let events: Vec<AgentEventType> = vec![
        AgentEventType::RunStarted {},
        AgentEventType::RunFinished {
            success: true,
            total_steps: 1,
            failed_steps: 0,
            usage: None,
            context_budget: None,
        },
        AgentEventType::RunError {
            message: "err".into(),
            code: None,
            usage: None,
        },
        AgentEventType::PlanStarted { initial_plan: true },
        AgentEventType::PlanFinished { total_steps: 1 },
        AgentEventType::StepCompleted {
            step_id: "s1".into(),
            success: true,
            usage: None,
            context_budget: None,
        },
        AgentEventType::ContextBudgetUpdate {
            budget: ContextBudget::default(),
            is_warning: false,
            is_critical: false,
        },
        AgentEventType::ContextCompaction {
            tier: CompactionTier::Trim,
            tokens_before: 0,
            tokens_after: 0,
            entries_affected: 0,
            context_limit: 0,
            usage_ratio: 0.0,
            summary: None,
            reinjected_skills: vec![],
            context_budget: None,
        },
    ];
    assert!(events.len() >= 8);
}