kwaak 0.4.0

Run a team of autonomous agents on your code, right from your terminal
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

use std::collections::HashSet;

use ratatui::widgets::ScrollbarState;

use crate::chat_message::ChatMessage;

#[derive(Debug, Clone)]
pub struct Chat {
    /// Display name of the chat
    pub name: String,
    /// Identifier used to match responses
    pub uuid: uuid::Uuid,
    pub messages: Vec<ChatMessage>,
    pub state: ChatState,
    pub new_message_count: usize,
    pub completed_tool_call_ids: HashSet<String>,

    // Scrolling is per chat
    // but handled in the ui
    pub vertical_scroll_state: ScrollbarState,
    pub vertical_scroll: usize,
    pub num_lines: usize,
}

impl Chat {
    pub fn add_message(&mut self, message: ChatMessage) {
        if !message.role().is_user() {
            self.new_message_count += 1;
        }

        // If it's a completed tool call, just register it is done and do not add the message
        // The state is updated when rendering on the initial tool call
        if message.role().is_tool() {
            let Some(tool_call) = message.maybe_completed_tool_call() else {
                tracing::error!(
                    "Received a tool message without a tool call ID: {:?}",
                    message
                );
                return;
            };

            self.completed_tool_call_ids
                .insert(tool_call.id().to_string());

            return;
        }
        self.messages.push(message);
    }

    pub fn transition(&mut self, state: ChatState) {
        self.state = state;
    }

    #[must_use]
    pub fn is_loading(&self) -> bool {
        matches!(
            self.state,
            ChatState::Loading | ChatState::LoadingWithMessage(_)
        )
    }

    #[must_use]
    pub fn is_tool_call_completed(&self, tool_call_id: &str) -> bool {
        self.completed_tool_call_ids.contains(tool_call_id)
    }
}

#[derive(Debug, Clone, Default, strum::EnumIs, PartialEq)]
pub enum ChatState {
    Loading,
    LoadingWithMessage(String),
    #[default]
    Ready,
}

impl Default for Chat {
    fn default() -> Self {
        Self {
            name: "Chat".to_string(),
            uuid: uuid::Uuid::new_v4(),
            messages: Vec::new(),
            state: ChatState::default(),
            new_message_count: 0,
            completed_tool_call_ids: HashSet::new(),
            vertical_scroll_state: ScrollbarState::default(),
            vertical_scroll: 0,
            num_lines: 0,
        }
    }
}

#[cfg(test)]
mod tests {
    use swiftide::chat_completion;

    use super::*;
    use crate::chat_message::ChatMessage;

    #[test]
    fn test_add_message_increases_new_message_count() {
        let mut chat = Chat::default();
        let message = ChatMessage::new_system("Test message");

        chat.add_message(message);

        assert_eq!(chat.new_message_count, 1);
        assert_eq!(chat.messages.len(), 1);
    }

    #[test]
    fn test_add_message_does_not_increase_new_message_count_for_user() {
        let mut chat = Chat::default();
        let message = ChatMessage::new_user("Test message");

        chat.add_message(message);

        assert_eq!(chat.new_message_count, 0);
        assert_eq!(chat.messages.len(), 1);
    }

    #[test]
    fn test_add_message_tool_call() {
        let mut chat = Chat::default();
        let tool_call = chat_completion::ToolCall::builder()
            .id("tool_call_id")
            .name("some_tool")
            .build()
            .unwrap();
        let message =
            chat_completion::ChatMessage::new_tool_output(tool_call, String::new()).into();

        chat.add_message(message);

        assert!(chat.is_tool_call_completed("tool_call_id"));
        assert_eq!(chat.messages.len(), 0);
    }

    #[test]
    fn test_transition() {
        let mut chat = Chat::default();
        chat.transition(ChatState::Loading);

        assert!(chat.is_loading());
    }

    #[test]
    fn test_is_loading() {
        let chat = Chat {
            state: ChatState::Loading,
            ..Default::default()
        };

        assert!(chat.is_loading());
    }

    #[test]
    fn test_is_tool_call_completed() {
        let mut chat = Chat::default();
        chat.completed_tool_call_ids
            .insert("tool_call_id".to_string());

        assert!(chat.is_tool_call_completed("tool_call_id"));
    }
}