oaapi 0.2.0

An unofficial Rust client for the OpenAI API.
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

use std::fmt::Display;
use crate::macros::impl_enum_string_serialization;

/// The role of the messages author.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum Role {
    /// system
    System,
    /// user
    User,
    /// assistant
    Assistant,
    /// tool
    Tool,
}

impl Default for Role {
    fn default() -> Self {
        Role::User
    }
}

impl Display for Role {
    fn fmt(
        &self,
        f: &mut std::fmt::Formatter<'_>,
    ) -> std::fmt::Result {
        match self {
            | Role::System => {
                write!(f, "system")
            },
            | Role::User => {
                write!(f, "user")
            },
            | Role::Assistant => {
                write!(f, "assistant")
            },
            | Role::Tool => {
                write!(f, "tool")
            },
        }
    }
}

impl_enum_string_serialization!(
    Role,
    System => "system",
    User => "user",
    Assistant => "assistant",
    Tool => "tool"
);

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn role_serialization() {
        assert_eq!(
            serde_json::to_string(&Role::System).unwrap(),
            "\"system\""
        );

        assert_eq!(
            serde_json::to_string(&Role::User).unwrap(),
            "\"user\""
        );

        assert_eq!(
            serde_json::to_string(&Role::Assistant).unwrap(),
            "\"assistant\""
        );

        assert_eq!(
            serde_json::to_string(&Role::Tool).unwrap(),
            "\"tool\""
        );
    }

    #[test]
    fn role_deserialization() {
        assert_eq!(
            serde_json::from_str::<Role>("\"user\"").unwrap(),
            Role::User
        );

        assert_eq!(
            serde_json::from_str::<Role>("\"system\"").unwrap(),
            Role::System
        );

        assert_eq!(
            serde_json::from_str::<Role>("\"assistant\"").unwrap(),
            Role::Assistant
        );

        assert_eq!(
            serde_json::from_str::<Role>("\"tool\"").unwrap(),
            Role::Tool
        );
    }
}