omk 0.5.0

A Rust runtime for Kimi CLI. Turns prompts into proof-backed engineering runs with gates, worktrees, and replay.
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

use tempfile::TempDir;

use crate::runtime::config::EVENTS_FILE;
use crate::runtime::events::{EventKind, EventReader};
use tokio_util::sync::CancellationToken;

use super::*;

#[tokio::test]
async fn test_run_with_cancel_reason_records_custom_task_failure_error() {
    let tmp = TempDir::new().unwrap();
    let mut runner = make_runner(&tmp).await;
    runner.seed_task("Pause this work");

    let worker = make_spec(&tmp, "worker-cancelled").await;
    let cancel = CancellationToken::new();
    cancel.cancel();

    let summary = runner
        .run_with_cancel_reason(std::slice::from_ref(&worker), &cancel, "cancelled by user")
        .await
        .unwrap();

    assert_eq!(summary.completed, 0);
    assert_eq!(summary.failed, 0);
    assert_eq!(summary.cancelled, 1);
    assert_eq!(summary.total, 1);

    let events = EventReader::read_all(&tmp.path().join("state").join(EVENTS_FILE))
        .await
        .unwrap();
    assert!(events.iter().any(|event| {
        event.kind == EventKind::TaskFailed
            && event
                .payload
                .as_ref()
                .and_then(|payload| payload.get("task_id"))
                .and_then(|value| value.as_str())
                == Some("task-1")
            && event
                .payload
                .as_ref()
                .and_then(|payload| payload.get("error"))
                .and_then(|value| value.as_str())
                == Some("cancelled by user")
    }));
}