brainos-grpcadapter 0.5.0

gRPC adapter for Brain OS
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

//! `BrainEvent` filter predicate and timestamp accessor used by the
//! `AgentService::brain_events` server-streaming RPC.

use crate::agent_proto::BrainEventsRequest;

/// Predicate matching a `BrainEvent` against the gRPC filter fields.
pub(crate) fn brain_event_matches(ev: &observe::BrainEvent, filter: &BrainEventsRequest) -> bool {
    if !filter.kind.is_empty() && ev.kind() != filter.kind {
        return false;
    }
    if !filter.tool_id.is_empty() && ev.tool_id() != Some(filter.tool_id.as_str()) {
        return false;
    }
    if !filter.principal.is_empty() {
        // Bus events do not yet carry a principal; the filter rejects
        // everything when set.
        return false;
    }
    if !filter.since.is_empty() {
        let Ok(since) = chrono::DateTime::parse_from_rfc3339(&filter.since) else {
            return false;
        };
        let since = since.with_timezone(&chrono::Utc);
        if brain_event_ts(ev) < since {
            return false;
        }
    }
    true
}

pub(crate) fn brain_event_ts(ev: &observe::BrainEvent) -> chrono::DateTime<chrono::Utc> {
    use observe::BrainEvent::*;
    match ev {
        SignalReceived { ts, .. }
        | IntentClassified { ts, .. }
        | ToolRouteResolved { ts, .. }
        | ConfirmationRequested { ts, .. }
        | ConfirmationResolved { ts, .. }
        | ToolCallStarted { ts, .. }
        | ToolCallFinished { ts, .. }
        | ReflexFired { ts, .. }
        | AuditAppended { ts, .. }
        | BudgetCrossed { ts, .. }
        | ResourcePressure { ts, .. }
        | BreakerStateChange { ts, .. }
        | Error { ts, .. }
        | TerminalSessionOpened { ts, .. }
        | TerminalSessionClosed { ts, .. }
        | TaskStateChange { ts, .. }
        | ServiceHealthChanged { ts, .. } => *ts,
    }
}