klieo-runlog 2.2.0

Tier 2 observability — RunLog aggregate + replay engine for klieo agents.
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

//! `SqliteRunLogStore` CRUD smoke tests. Gated on the `sqlite` feature.

#![cfg(feature = "sqlite")]

use klieo_core::ids::RunId;
use klieo_runlog::projector::project;
use klieo_runlog::store::RunLogStore;
use klieo_runlog::store_sqlite::SqliteRunLogStore;
use klieo_runlog::types::Cost;
use tempfile::NamedTempFile;

fn tmp_db() -> NamedTempFile {
    NamedTempFile::new().unwrap()
}

#[tokio::test]
async fn put_then_get_round_trip() {
    let f = tmp_db();
    let store = SqliteRunLogStore::new(f.path()).await.unwrap();
    let log = project(RunId::new(), "writer", &[]);
    let rid = log.run_id;
    store.put(&log).await.unwrap();
    let back = store.get(rid).await.unwrap().unwrap();
    assert_eq!(back.agent, "writer");
}

#[tokio::test]
async fn list_orders_recent_first_and_respects_limit() {
    let f = tmp_db();
    let store = SqliteRunLogStore::new(f.path()).await.unwrap();
    for _ in 0..5 {
        store
            .put(&project(RunId::new(), "writer", &[]))
            .await
            .unwrap();
    }
    let rows = store.list(None, 3).await.unwrap();
    assert_eq!(rows.len(), 3);
    for w in rows.windows(2) {
        assert!(w[0].started_at >= w[1].started_at);
    }
}

#[tokio::test]
async fn list_filtered_by_agent() {
    let f = tmp_db();
    let store = SqliteRunLogStore::new(f.path()).await.unwrap();
    store
        .put(&project(RunId::new(), "writer", &[]))
        .await
        .unwrap();
    store
        .put(&project(RunId::new(), "editor", &[]))
        .await
        .unwrap();
    let rows = store.list(Some("writer"), 10).await.unwrap();
    assert_eq!(rows.len(), 1);
    assert_eq!(rows[0].agent, "writer");
}

#[tokio::test]
async fn delete_removes_log() {
    let f = tmp_db();
    let store = SqliteRunLogStore::new(f.path()).await.unwrap();
    let log = project(RunId::new(), "writer", &[]);
    let rid = log.run_id;
    store.put(&log).await.unwrap();
    store.delete(rid).await.unwrap();
    assert!(store.get(rid).await.unwrap().is_none());
}

#[tokio::test]
async fn cost_estimate_round_trips_prompt_and_completion_split() {
    let f = tmp_db();
    let store = SqliteRunLogStore::new(f.path()).await.unwrap();
    let mut log = project(RunId::new(), "writer", &[]);
    log.cost_estimate = Some(Cost {
        prompt_usd: 0.005,
        completion_usd: 0.015,
        total_usd: 0.020,
    });
    let rid = log.run_id;
    store.put(&log).await.unwrap();
    let back = store.get(rid).await.unwrap().expect("row exists");
    let cost = back.cost_estimate.expect("cost preserved");
    assert!((cost.prompt_usd - 0.005).abs() < 1e-12);
    assert!((cost.completion_usd - 0.015).abs() < 1e-12);
    assert!((cost.total_usd - 0.020).abs() < 1e-12);
}