aether_core/events/
user_message.rs1use llm::{ChatMessage, ContentBlock, MessageId, ReasoningEffort, StreamingModelProvider};
2
3#[derive(Debug)]
8pub enum Command {
9 UserCommand(UserCommand),
10 AgentCommand(AgentCommand),
11}
12
13impl Command {
14 pub fn text(text: impl Into<String>) -> Self {
15 Self::with_content(vec![ContentBlock::text(text)])
16 }
17
18 pub fn with_content(content: Vec<ContentBlock>) -> Self {
19 Self::with_message_id(MessageId::new(), content)
20 }
21
22 pub fn with_message_id(message_id: MessageId, content: Vec<ContentBlock>) -> Self {
23 Self::UserCommand(UserCommand::Text { message_id, content })
24 }
25
26 pub fn cancel() -> Self {
27 Self::UserCommand(UserCommand::Cancel)
28 }
29
30 pub fn clear_context() -> Self {
31 Self::UserCommand(UserCommand::ClearContext)
32 }
33
34 pub fn agent(command: AgentCommand) -> Self {
35 Self::AgentCommand(command)
36 }
37}
38
39pub enum UserCommand {
41 Text { message_id: MessageId, content: Vec<ContentBlock> },
42 Cancel,
43 ClearContext,
44}
45
46impl std::fmt::Debug for UserCommand {
47 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
48 match self {
49 UserCommand::Text { content, .. } => {
50 f.debug_struct("Text").field("content_blocks", &content.len()).finish()
51 }
52 UserCommand::Cancel => write!(f, "Cancel"),
53 UserCommand::ClearContext => write!(f, "ClearContext"),
54 }
55 }
56}
57
58pub enum AgentCommand {
60 SwitchModel(Box<dyn StreamingModelProvider>),
61 SetReasoningEffort(Option<ReasoningEffort>),
62 ReplaceConversation(Vec<ChatMessage>),
63}
64
65impl std::fmt::Debug for AgentCommand {
66 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
67 match self {
68 AgentCommand::SwitchModel(provider) => {
69 f.debug_tuple("SwitchModel").field(&provider.display_name()).finish()
70 }
71 AgentCommand::SetReasoningEffort(effort) => f.debug_tuple("SetReasoningEffort").field(effort).finish(),
72 AgentCommand::ReplaceConversation(messages) => {
73 f.debug_tuple("ReplaceConversation").field(&messages.len()).finish()
74 }
75 }
76 }
77}