chatty-rs 0.0.1-alpha1

A terminal-based chat client for OpenAI's GPT models.
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

use std::sync::Arc;

use tokio::sync::mpsc;
use tui_textarea::Input;

use super::Conversation;

#[derive(Debug)]
pub enum Event {
    Notice(crate::models::NoticeMessage),

    BackendAbort,
    BackendMessage(crate::models::Message),
    ChatCompletionResponse(crate::models::BackendResponse),

    SetConversation(Option<Conversation>),
    ConversationDeleted(String),
    ConversationUpdated(Conversation),

    KeyboardCharInput(Input),
    KeyboardEsc,
    KeyboardEnter,
    KeyboardAltEnter,
    KeyboardCtrlC,
    KeyboardCtrlR,
    KeyboardCtrlN,
    KeyboardCtrlE,
    KeyboardCtrlL,
    KeyboardCtrlH,
    KeyboardF1,
    KeyboardPaste(String),

    Quit,

    UiTick,
    UiScrollUp,
    UiScrollDown,
    UiScrollPageUp,
    UiScrollPageDown,
}

#[macro_export]
macro_rules! info_event {
    ($($arg:tt)*) => {
        Event::Notice($crate::info_notice!($($arg)*))
    }
}

#[macro_export]
macro_rules! warn_event {
    ($($arg:tt)*) => {
        Event::Notice($crate::warn_notice!($($arg)*))
    }
}

#[macro_export]
macro_rules! error_event {
    ($($arg:tt)*) => {
        Event::Notice($crate::error_notice!($($arg)*))
    }
}

#[async_trait::async_trait]
pub trait EventTx {
    async fn send(&self, event: Event) -> Result<(), mpsc::error::SendError<Event>>;
}

impl Event {
    pub fn is_keyboard_event(&self) -> bool {
        matches!(
            self,
            Event::KeyboardCharInput(_)
                | Event::KeyboardEsc
                | Event::KeyboardEnter
                | Event::KeyboardAltEnter
                | Event::KeyboardCtrlC
                | Event::KeyboardCtrlR
                | Event::KeyboardCtrlN
                | Event::KeyboardCtrlE
                | Event::KeyboardCtrlL
                | Event::KeyboardCtrlH
                | Event::KeyboardF1
                | Event::Quit
                | Event::UiScrollUp
                | Event::UiScrollDown
                | Event::UiScrollPageUp
                | Event::UiScrollPageDown
                | Event::KeyboardPaste(_)
        )
    }
}

#[async_trait::async_trait]
impl EventTx for mpsc::Sender<Event> {
    async fn send(&self, event: Event) -> Result<(), mpsc::error::SendError<Event>> {
        self.send(event).await
    }
}

#[async_trait::async_trait]
impl EventTx for mpsc::UnboundedSender<Event> {
    async fn send(&self, event: Event) -> Result<(), mpsc::error::SendError<Event>> {
        self.send(event)
    }
}

pub type ArcEventTx = Arc<dyn EventTx + Send + Sync>;