bcode_session 0.0.1-alpha.0

Session lifecycle and attachment management for bcode
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

//! Stable current runtime-work projection value codecs.

use bcode_session_models::{RuntimeWorkKind, RuntimeWorkStatus};

#[must_use]
pub const fn runtime_work_kind_name(kind: RuntimeWorkKind) -> &'static str {
    match kind {
        RuntimeWorkKind::Tool => "tool",
        RuntimeWorkKind::PluginInvocation => "plugin_invocation",
        RuntimeWorkKind::ModelTurn => "model_turn",
        RuntimeWorkKind::EventDelivery => "event_delivery",
        RuntimeWorkKind::Workflow => "workflow",
        RuntimeWorkKind::WorkflowNode => "workflow_node",
    }
}

pub fn parse_runtime_work_kind(value: &str) -> Option<RuntimeWorkKind> {
    Some(match value {
        "tool" => RuntimeWorkKind::Tool,
        "plugin_invocation" => RuntimeWorkKind::PluginInvocation,
        "model_turn" => RuntimeWorkKind::ModelTurn,
        "event_delivery" => RuntimeWorkKind::EventDelivery,
        "workflow" => RuntimeWorkKind::Workflow,
        "workflow_node" => RuntimeWorkKind::WorkflowNode,
        _ => return None,
    })
}

#[must_use]
pub const fn runtime_work_status_name(status: RuntimeWorkStatus) -> &'static str {
    match status {
        RuntimeWorkStatus::Queued => "queued",
        RuntimeWorkStatus::Running => "running",
        RuntimeWorkStatus::Cancelling => "cancelling",
        RuntimeWorkStatus::Completed => "completed",
        RuntimeWorkStatus::Failed => "failed",
        RuntimeWorkStatus::TimedOut => "timed_out",
        RuntimeWorkStatus::Cancelled => "cancelled",
        RuntimeWorkStatus::Suspended => "suspended",
    }
}

pub fn parse_runtime_work_status(value: &str) -> Option<RuntimeWorkStatus> {
    Some(match value {
        "running" => RuntimeWorkStatus::Running,
        "queued" => RuntimeWorkStatus::Queued,
        "cancelling" => RuntimeWorkStatus::Cancelling,
        "completed" => RuntimeWorkStatus::Completed,
        "failed" => RuntimeWorkStatus::Failed,
        "timed_out" => RuntimeWorkStatus::TimedOut,
        "cancelled" => RuntimeWorkStatus::Cancelled,
        "suspended" => RuntimeWorkStatus::Suspended,
        _ => return None,
    })
}

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

    #[test]
    fn runtime_work_values_round_trip() {
        for status in [
            RuntimeWorkStatus::Queued,
            RuntimeWorkStatus::Running,
            RuntimeWorkStatus::Cancelling,
            RuntimeWorkStatus::Completed,
            RuntimeWorkStatus::Failed,
            RuntimeWorkStatus::TimedOut,
            RuntimeWorkStatus::Cancelled,
            RuntimeWorkStatus::Suspended,
        ] {
            assert_eq!(
                parse_runtime_work_status(runtime_work_status_name(status)),
                Some(status)
            );
        }
        for kind in [
            RuntimeWorkKind::Tool,
            RuntimeWorkKind::PluginInvocation,
            RuntimeWorkKind::ModelTurn,
            RuntimeWorkKind::EventDelivery,
            RuntimeWorkKind::Workflow,
            RuntimeWorkKind::WorkflowNode,
        ] {
            assert_eq!(
                parse_runtime_work_kind(runtime_work_kind_name(kind)),
                Some(kind)
            );
        }
        for value in ["", "future_variant", "RUNNING"] {
            assert_eq!(parse_runtime_work_kind(value), None);
            assert_eq!(parse_runtime_work_status(value), None);
        }
    }
}