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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use std::path::Path;
use std::time::Duration;
use tokio::task::JoinHandle;

use crate::runtime::config::{HEARTBEAT_FILE, INBOX_FILE, OUTBOX_FILE, WORKERS_DIR};
use crate::runtime::worker::WorkerSpec;

use crate::runtime::goal::state::{GOAL_AGENT_WORKER_ID, GOAL_AGENT_WORKER_ROLE};

pub(crate) fn goal_agent_wire_runtime_available() -> bool {
    std::env::var_os("MOCK_KIMI").is_some() || which::which("kimi").is_ok()
}

pub(crate) async fn stop_wire_worker(handle: &mut JoinHandle<()>) {
    if tokio::time::timeout(Duration::from_secs(2), &mut *handle)
        .await
        .is_err()
    {
        handle.abort();
        let _ = handle.await;
    }
}

pub(crate) fn goal_agent_worker_name(index: usize) -> String {
    if index == 0 {
        GOAL_AGENT_WORKER_ID.to_string()
    } else {
        format!("goal-agent-worker-{index}")
    }
}

pub(crate) fn goal_agent_worker_count(max_agents: usize, task_count: usize) -> usize {
    max_agents.max(1).min(task_count.max(1))
}

pub(crate) fn goal_agent_lease_seconds_override() -> Option<i64> {
    std::env::var("OMK_GOAL_AGENT_LEASE_SECS")
        .ok()
        .and_then(|raw| raw.trim().parse::<i64>().ok())
        .filter(|secs| *secs > 0)
}

pub(crate) async fn prepare_goal_agent_workers(
    run_dir: &Path,
    project_dir: &Path,
    worker_count: usize,
) -> anyhow::Result<Vec<WorkerSpec>> {
    let mut specs = Vec::with_capacity(worker_count);
    for index in 0..worker_count {
        let name = goal_agent_worker_name(index);
        let worker_dir = run_dir.join(WORKERS_DIR).join(&name);
        crate::runtime::config::ensure_private_dir(&worker_dir).await?;
        let spec = WorkerSpec {
            name,
            role: GOAL_AGENT_WORKER_ROLE.to_string(),
            inbox: worker_dir.join(INBOX_FILE),
            outbox: worker_dir.join(OUTBOX_FILE),
            heartbeat: worker_dir.join(HEARTBEAT_FILE),
            project_dir: Some(project_dir.to_path_buf()),
            external_tools: None,
            approval_policy: crate::runtime::wire_worker::ApprovalPolicy::default(),
            approval_timeout_secs: crate::runtime::worker::default_approval_timeout_secs(),
        };
        spec.save().await?;
        tokio::fs::write(&spec.inbox, b"").await?;
        tokio::fs::write(&spec.outbox, b"").await?;
        tokio::fs::write(worker_dir.join("wire-events.jsonl"), b"").await?;
        specs.push(spec);
    }
    Ok(specs)
}