kaynine-core 0.1.0

Core agent loop, messages, events, policies, and provider abstractions for Kaynine
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

use crate::error::RunFailureReason;
use crate::ids::{BranchId, RunId, SessionId, ToolCallId};
use crate::message::{FinishReason, Usage};
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct EventEnvelope<E> {
    pub session_id: SessionId,
    pub branch_id: BranchId,
    pub run_id: Option<RunId>,
    pub revision: u64,
    pub run_seq: Option<u64>,
    pub payload: E,
}

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[serde(tag = "kind")]
pub enum RealtimeEvent {
    RunStarted,
    TurnStarted {
        turn: u32,
    },
    TextDelta {
        block: u32,
        text: String,
    },
    ReasoningDelta {
        block: u32,
        text: String,
    },
    ToolCallStarted {
        block: u32,
        id: ToolCallId,
        name: String,
    },
    UsageUpdated {
        usage: Usage,
    },
    ToolProgress {
        call_id: ToolCallId,
        message: String,
    },
    TurnCompleted {
        finish_reason: FinishReason,
    },
    RunCompleted,
    RunFailed {
        reason: RunFailureReason,
    },
    RunCancelled,
    ApprovalPending {
        call_id: ToolCallId,
        deadline_unix: i64,
    },
    ApprovalResolved {
        call_id: ToolCallId,
        approved: bool,
    },
    SteerInjected {
        count: u32,
    },
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn envelope_roundtrips() {
        let env = EventEnvelope {
            session_id: SessionId::from("s1"),
            branch_id: BranchId::from("b1"),
            run_id: Some(RunId::from("r1")),
            revision: 7,
            run_seq: Some(3),
            payload: RealtimeEvent::TextDelta {
                block: 0,
                text: "你".into(),
            },
        };
        let json = serde_json::to_string(&env).unwrap();
        let back: EventEnvelope<RealtimeEvent> = serde_json::from_str(&json).unwrap();
        assert_eq!(back, env);
    }

    #[test]
    fn run_seq_is_optional_for_non_run_events() {
        let env = EventEnvelope {
            session_id: SessionId::from("s1"),
            branch_id: BranchId::from("b1"),
            run_id: None,
            revision: 1,
            run_seq: None,
            payload: RealtimeEvent::RunCompleted,
        };
        let json = serde_json::to_string(&env).unwrap();
        assert!(json.contains("\"run_seq\":null"));
    }
}