agtrace-engine 0.7.1

Internal core engine for the agtrace CLI. Not intended for direct use.
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

use agtrace_types::{AgentEvent, EventPayload};

/// Filters for including/excluding event types.
///
/// Supports both allowlist (`only`) and blocklist (`hide`) patterns
/// for event type filtering.
pub struct EventFilters {
    /// Event types to exclude (blocklist).
    pub hide: Option<Vec<String>>,
    /// Event types to include exclusively (allowlist).
    pub only: Option<Vec<String>>,
}

/// Filter events by type using inclusion/exclusion patterns.
///
/// Applies `only` filter first (if present), then `hide` filter.
/// Supported patterns: "user", "assistant"/"message", "tool",
/// "reasoning", "token"/"tokenusage", "notification"/"info".
pub fn filter_events(events: &[AgentEvent], filters: EventFilters) -> Vec<AgentEvent> {
    let mut filtered = events.to_vec();

    if let Some(only_patterns) = filters.only {
        filtered.retain(|e| {
            only_patterns.iter().any(|pattern| {
                let pattern_lower = pattern.to_lowercase();
                match &e.payload {
                    EventPayload::User(_) => pattern_lower == "user",
                    EventPayload::Message(_) => {
                        pattern_lower == "assistant" || pattern_lower == "message"
                    }
                    EventPayload::ToolCall(_) | EventPayload::ToolResult(_) => {
                        pattern_lower == "tool"
                    }
                    EventPayload::Reasoning(_) => pattern_lower == "reasoning",
                    EventPayload::TokenUsage(_) => {
                        pattern_lower == "token" || pattern_lower == "tokenusage"
                    }
                    EventPayload::Notification(_) => {
                        pattern_lower == "notification" || pattern_lower == "info"
                    }
                    EventPayload::SlashCommand(_) => {
                        pattern_lower == "slashcommand" || pattern_lower == "command"
                    }
                    EventPayload::QueueOperation(_) => {
                        pattern_lower == "queueoperation" || pattern_lower == "queue"
                    }
                    EventPayload::Summary(_) => pattern_lower == "summary",
                }
            })
        });
    }

    if let Some(hide_patterns) = filters.hide {
        filtered.retain(|e| {
            !hide_patterns.iter().any(|pattern| {
                let pattern_lower = pattern.to_lowercase();
                match &e.payload {
                    EventPayload::User(_) => pattern_lower == "user",
                    EventPayload::Message(_) => {
                        pattern_lower == "assistant" || pattern_lower == "message"
                    }
                    EventPayload::ToolCall(_) | EventPayload::ToolResult(_) => {
                        pattern_lower == "tool"
                    }
                    EventPayload::Reasoning(_) => pattern_lower == "reasoning",
                    EventPayload::TokenUsage(_) => {
                        pattern_lower == "token" || pattern_lower == "tokenusage"
                    }
                    EventPayload::Notification(_) => {
                        pattern_lower == "notification" || pattern_lower == "info"
                    }
                    EventPayload::SlashCommand(_) => {
                        pattern_lower == "slashcommand" || pattern_lower == "command"
                    }
                    EventPayload::QueueOperation(_) => {
                        pattern_lower == "queueoperation" || pattern_lower == "queue"
                    }
                    EventPayload::Summary(_) => pattern_lower == "summary",
                }
            })
        });
    }

    filtered
}