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
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

use chrono::{DateTime, Utc};

use crate::runtime::goal::agent::GoalAgentTaskProposal;
use crate::runtime::goal::state::{GoalState, GOAL_LOCAL_VERIFY_TASK_ID};
use crate::runtime::goal::task_graph::GoalTaskGraph;
use crate::runtime::scheduler::task::Task;

pub(super) fn goal_agent_scheduler_tasks(
    state: &GoalState,
    task_graph: &GoalTaskGraph,
    generated_at: DateTime<Utc>,
    controller_task_id: &str,
    proposals: &[GoalAgentTaskProposal],
) -> Vec<Task> {
    proposals
        .iter()
        .map(|proposal| {
            let mut task = Task::new(proposal.id.clone(), proposal.title.clone())
                .with_description(goal_agent_task_prompt(
                    state,
                    task_graph,
                    generated_at,
                    proposal,
                ))
                .with_dependencies(proposal.dependencies.clone())
                .with_read_set(proposal.read_set.clone())
                .with_write_set(proposal.write_set.clone())
                .with_priority(proposal.priority)
                .with_max_retries(0);
            task.extra.insert(
                "acceptance".to_string(),
                serde_json::json!(proposal.acceptance),
            );
            task.extra.insert(
                "budget_secs".to_string(),
                serde_json::json!(proposal.budget_secs),
            );
            task.extra
                .insert("risk".to_string(), serde_json::json!(proposal.risk));
            task.extra.insert(
                "controller_task_id".to_string(),
                serde_json::json!(controller_task_id),
            );
            task
        })
        .collect()
}

pub(super) fn goal_agent_task_prompt(
    state: &GoalState,
    task_graph: &GoalTaskGraph,
    generated_at: DateTime<Utc>,
    proposal: &GoalAgentTaskProposal,
) -> String {
    let local_status = task_graph
        .tasks
        .iter()
        .find(|task| task.id == GOAL_LOCAL_VERIFY_TASK_ID)
        .map(|task| task.status.to_string())
        .unwrap_or_else(|| "unknown".to_string());

    let policy_text = if state.slice_execution {
        "Stay inside the current repository, keep the diff minimal, do not push, do not touch secrets, and summarize changed files plus verification still needed for production readiness. The controller will auto-commit your changes after the wave."
    } else {
        "Stay inside the current repository, keep the diff minimal, do not commit, do not publish, do not touch secrets, and summarize changed files plus verification still needed for production readiness."
    };

    let review_feedback = if state.slice_execution {
        proposal
            .description
            .lines()
            .find(|line| line.contains("[review-feedback]"))
            .map(|line| {
                line.trim_start_matches("[review-feedback]")
                    .trim()
                    .to_string()
            })
            .unwrap_or_default()
    } else {
        String::new()
    };

    let feedback_section = if review_feedback.is_empty() {
        String::new()
    } else {
        format!("\n\nPrevious review feedback — address these issues:\n{review_feedback}")
    };

    let write_scope_section = if proposal.write_set.is_empty() {
        String::new()
    } else {
        format!(
            "\n\nAllowed write paths:\n- {}",
            proposal.write_set.join("\n- ")
        )
    };

    format!(
        "Goal ID: {}\nGenerated: {generated_at}\n\nOriginal goal:\n{}\n\nNormalized goal:\n{}\n\nController task: {}\nTitle: {}\nBudget: {} seconds\nRisk: {}\n\nTask:\n{}\n\nAcceptance criteria:\n- {}\n\nPolicy:\n{policy_text}{feedback_section}{write_scope_section}\n\nLocal verification task status: {local_status}",
        state.goal_id,
        state.original_goal,
        state.normalized_goal,
        proposal.id,
        proposal.title,
        proposal.budget_secs,
        proposal.risk,
        proposal.description,
        proposal.acceptance.join("\n- ")
    )
}