ratkit 0.2.14

A comprehensive collection of reusable TUI components for ratatui including resizable splits, tree views, markdown rendering, toast notifications, dialogs, and terminal embedding
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

/// Role of a message sender.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum MessageRole {
    /// Message from user
    User,
    /// Message from AI
    Assistant,
}

/// A single chat message.
#[derive(Debug, Clone)]
pub struct Message {
    /// Role of sender
    pub role: MessageRole,
    /// Message content
    pub content: String,
    /// Optional file attachments
    pub attachments: Vec<String>,
    /// Timestamp
    pub timestamp: std::time::SystemTime,
}

impl Message {
    /// Create a new user message.
    pub fn user(content: String) -> Self {
        Self {
            role: MessageRole::User,
            content,
            attachments: Vec::new(),
            timestamp: std::time::SystemTime::now(),
        }
    }

    /// Create a new AI message.
    pub fn assistant(content: String) -> Self {
        Self {
            role: MessageRole::Assistant,
            content,
            attachments: Vec::new(),
            timestamp: std::time::SystemTime::now(),
        }
    }

    /// Add file attachment.
    pub fn with_attachment(mut self, file: String) -> Self {
        self.attachments.push(file);
        self
    }
}

/// Store for chat message history.
#[derive(Debug, Default, Clone)]
pub struct MessageStore {
    /// All messages in conversation
    messages: Vec<Message>,
}

impl MessageStore {
    /// Create a new empty message store.
    pub fn new() -> Self {
        Self::default()
    }

    /// Add a message to store.
    pub fn add(&mut self, message: Message) {
        self.messages.push(message);
    }

    /// Get all messages.
    pub fn messages(&self) -> &[Message] {
        &self.messages
    }

    /// Clear all messages.
    pub fn clear(&mut self) {
        self.messages.clear();
    }

    /// Get number of messages.
    pub fn len(&self) -> usize {
        self.messages.len()
    }

    /// Check if store is empty.
    pub fn is_empty(&self) -> bool {
        self.messages.is_empty()
    }

    /// Get last message.
    pub fn last(&self) -> Option<&Message> {
        self.messages.last()
    }
}