tandem-server 0.6.0

HTTP server for Tandem engine APIs
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
129

use super::*;

use crate::runtime_event_log::{query_runtime_event_log, RuntimeEventLogQuery};

const DEFAULT_RUNTIME_EVENTS_LIMIT: usize = 250;
const MAX_RUNTIME_EVENTS_LIMIT: usize = 1_000;

#[derive(Debug, Deserialize, Default)]
pub(super) struct RuntimeEventsQuery {
    pub after_seq: Option<u64>,
    pub since_seq: Option<u64>,
    pub limit: Option<usize>,
}

pub(super) async fn get_run_events(
    State(state): State<AppState>,
    Extension(tenant_context): Extension<TenantContext>,
    Path(run_id): Path<String>,
    Query(query): Query<RuntimeEventsQuery>,
) -> Json<Value> {
    let limit = query
        .limit
        .unwrap_or(DEFAULT_RUNTIME_EVENTS_LIMIT)
        .clamp(1, MAX_RUNTIME_EVENTS_LIMIT);
    let rows = query_runtime_event_log(
        &state.runtime_events_path,
        &tenant_context,
        RuntimeEventLogQuery {
            run_id: &run_id,
            after_seq: query.after_seq.or(query.since_seq),
            limit: Some(limit),
        },
    );
    let last_seq = rows.last().map(|row| row.seq());
    let events = rows
        .iter()
        .map(|row| serde_json::to_value(row).unwrap_or(Value::Null))
        .collect::<Vec<_>>();

    Json(json!({
        "run_id": run_id,
        "events": events,
        "count": events.len(),
        "last_seq": last_seq,
        "limit": limit,
        "sequence_scope": "runtime_event_bus",
    }))
}

#[cfg(test)]
mod tests {
    use serde_json::json;
    use tandem_types::{EngineEvent, RuntimeEventEnvelope};
    use uuid::Uuid;

    use super::*;
    use crate::runtime_event_log::{append_runtime_event_log_row, RuntimeEventLogRow};

    fn tenant(org: &str, workspace: &str) -> TenantContext {
        TenantContext::explicit_user_workspace(org, workspace, None, "user-a")
    }

    fn event(seq: u64, run_id: &str, tenant_context: TenantContext) -> EngineEvent {
        EngineEvent::new(
            "session.run.started",
            json!({
                "runID": run_id,
                "sessionID": "session-a",
                "tenantContext": tenant_context,
            }),
        )
        .with_envelope(RuntimeEventEnvelope {
            event_id: format!("evt-{seq}"),
            seq,
            schema_version: 1,
            occurred_at_ms: 1_000 + seq,
            session_id: Some("session-a".to_string()),
            run_id: Some(run_id.to_string()),
            node_id: None,
            tenant_context: Some(tenant_context),
        })
    }

    #[tokio::test]
    async fn get_run_events_filters_by_tenant_and_sequence() {
        let mut state = crate::test_support::test_state().await;
        state.runtime_events_path =
            std::env::temp_dir().join(format!("runtime-events-api-{}.jsonl", Uuid::new_v4()));
        let tenant_a = tenant("org-a", "workspace-a");
        let tenant_b = tenant("org-b", "workspace-b");

        for event in [
            event(1, "run-a", tenant_a.clone()),
            event(2, "run-a", tenant_b),
            event(3, "run-a", tenant_a.clone()),
            event(4, "run-b", tenant_a.clone()),
        ] {
            let row = RuntimeEventLogRow::from_engine_event(&event).expect("runtime row");
            append_runtime_event_log_row(&state.runtime_events_path, &row)
                .await
                .expect("append row");
        }

        let Json(body) = get_run_events(
            State(state.clone()),
            Extension(tenant_a),
            Path("run-a".to_string()),
            Query(RuntimeEventsQuery {
                after_seq: Some(1),
                since_seq: None,
                limit: Some(10),
            }),
        )
        .await;

        assert_eq!(body.get("count").and_then(Value::as_u64), Some(1));
        assert_eq!(body.get("last_seq").and_then(Value::as_u64), Some(3));
        let events = body
            .get("events")
            .and_then(Value::as_array)
            .expect("events array");
        assert_eq!(
            events[0].get("event_type").and_then(Value::as_str),
            Some("session.run.started")
        );

        let _ = tokio::fs::remove_file(state.runtime_events_path).await;
    }
}