beam-core 0.3.3

Shared core types and workflow/session logic for beam
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189

use serde::{Deserialize, Serialize};

use crate::{config::ScreenAnalyzerConfig, session::AdoptedFrom};

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum DisplayMode {
    Hidden,
    Screenshot,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum ScreenStatus {
    Starting,
    Working,
    Idle,
    Analyzing,
    Limited,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum CliUsageLimitKind {
    Usage,
    Rate,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct CliUsageLimitState {
    pub limited: bool,
    pub kind: CliUsageLimitKind,
    pub retry_at_ms: u64,
    pub retry_label: String,
    pub retry_ready: bool,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum TermActionKey {
    Esc,
    CtrlC,
    Tab,
    Enter,
    Space,
    Up,
    Down,
    Left,
    Right,
    HalfPageUp,
    HalfPageDown,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct TuiPromptOption {
    #[serde(default)]
    pub label: Option<String>,
    pub text: String,
    pub selected: bool,
    #[serde(rename = "type", default)]
    pub option_type: Option<String>,
    #[serde(default)]
    pub keys: Vec<String>,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum FinalOutputKind {
    Bridge,
    LocalTurn,
    LocalTurnHeadless,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct InitConfig {
    pub session_id: String,
    pub title: String,
    pub chat_id: String,
    pub root_message_id: String,
    pub working_dir: String,
    pub cli_id: String,
    pub cli_bin: String,
    #[serde(default)]
    pub cli_args: Vec<String>,
    pub prompt: String,
    #[serde(default)]
    pub resume: bool,
    #[serde(default)]
    pub cli_session_id: Option<String>,
    pub lark_app_id: String,
    pub lark_app_secret: String,
    #[serde(default)]
    pub prompt_turn_id: Option<String>,
    #[serde(default)]
    pub owner_open_id: Option<String>,
    #[serde(default)]
    pub adopted_from: Option<AdoptedFrom>,
    #[serde(default)]
    pub adopt_restored_from_metadata: bool,
    #[serde(default)]
    pub screen_analyzer: ScreenAnalyzerConfig,
    #[serde(default)]
    pub initial_prompt: Option<String>,
    #[serde(default)]
    pub model: Option<String>,
    #[serde(default)]
    pub locale: Option<String>,
    #[serde(default)]
    pub bot_name: Option<String>,
    #[serde(default)]
    pub bot_open_id: Option<String>,
    #[serde(default)]
    pub resume_session_id: Option<String>,
    #[serde(default)]
    pub disable_cli_bypass: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum DaemonToWorker {
    Init(InitConfig),
    Message { content: String, turn_id: String },
    RawInput { content: String, turn_id: String },
    Close,
    Restart,
    SetDisplayMode { mode: DisplayMode },
    TermAction { key: TermActionKey },
    SpecialKeys { keys: Vec<String> },
    TuiKeys { keys: Vec<String>, is_final: bool },
    TuiTextInput { keys: Vec<String>, text: String },
    RefreshScreen,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum WorkerToDaemon {
    Ready {
        zellij_session: String,
    },
    PromptReady,
    ScreenUpdate {
        content: String,
        status: ScreenStatus,
        #[serde(default)]
        usage_limit: Option<CliUsageLimitState>,
    },
    ScreenshotUploaded {
        image_key: String,
        status: ScreenStatus,
        #[serde(default)]
        usage_limit: Option<CliUsageLimitState>,
    },
    CliSessionId {
        cli_session_id: String,
    },
    CliExit {
        code: Option<i32>,
        signal: Option<String>,
    },
    TuiPrompt {
        description: String,
        options: Vec<TuiPromptOption>,
        #[serde(default)]
        multi_select: bool,
    },
    TuiPromptResolved {
        #[serde(default)]
        selected_text: Option<String>,
    },
    FinalOutput {
        content: String,
        turn_id: String,
        #[serde(default)]
        kind: Option<FinalOutputKind>,
        #[serde(default)]
        user_text: Option<String>,
    },
    AdoptPreamble {
        user_text: String,
        assistant_text: String,
    },
    UserNotify {
        message: String,
    },
    Error {
        message: String,
    },
}