known-types-openai 0.1.1

Well-known types for OpenAI APIs.
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

// This is free and unencumbered software released into the public domain.

impl ChatCompletionRequestMessage {
    pub fn role(&self) -> &str {
        use ChatCompletionRequestMessage::*;
        match self {
            Developer(ChatCompletionRequestDeveloperMessage { .. }) => "developer",
            System(ChatCompletionRequestSystemMessage { .. }) => "system",
            User(ChatCompletionRequestUserMessage { .. }) => "user",
            Assistant(ChatCompletionRequestAssistantMessage { .. }) => "assistant",
            Tool(ChatCompletionRequestToolMessage { .. }) => "tool",
            Function(ChatCompletionRequestFunctionMessage { .. }) => "function",
        }
    }

    pub fn name(&self) -> Option<&str> {
        use ChatCompletionRequestMessage::*;
        match self {
            Developer(ChatCompletionRequestDeveloperMessage { name, .. }) => {
                name.as_ref().map(|s| s.as_str())
            }
            System(ChatCompletionRequestSystemMessage { name, .. }) => {
                name.as_ref().map(|s| s.as_str())
            }
            User(ChatCompletionRequestUserMessage { name, .. }) => {
                name.as_ref().map(|s| s.as_str())
            }
            Assistant(ChatCompletionRequestAssistantMessage { name, .. }) => {
                name.as_ref().map(|s| s.as_str())
            }
            Tool(..) => return None,
            Function(ChatCompletionRequestFunctionMessage { name, .. }) => {
                if name.as_str().is_empty() {
                    None
                } else {
                    Some(name.as_str())
                }
            }
        }
    }

    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestMessage::*;
        match self {
            Developer(ChatCompletionRequestDeveloperMessage { content, .. }) => {
                content.text_content()
            }
            System(ChatCompletionRequestSystemMessage { content, .. }) => content.text_content(),
            User(ChatCompletionRequestUserMessage { content, .. }) => content.text_content(),
            Assistant(ChatCompletionRequestAssistantMessage { content, .. }) => {
                content.as_ref().and_then(|c| c.text_content())
            }
            Tool(ChatCompletionRequestToolMessage { content, .. }) => content.text_content(),
            Function(ChatCompletionRequestFunctionMessage { content, .. }) => {
                content.as_ref().map(|s| s.as_str())
            }
        }
    }
}

impl ChatCompletionRequestDeveloperMessage_Content {
    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestDeveloperMessage_Content::*;
        match self {
            Text(s) => Some(s.as_str()),
            Array(..) => None,
        }
    }
}

impl ChatCompletionRequestSystemMessage_Content {
    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestSystemMessage_Content::*;
        match self {
            Text(s) => Some(s.as_str()),
            Array(..) => None,
        }
    }
}

impl ChatCompletionRequestUserMessage_Content {
    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestUserMessage_Content::*;
        match self {
            Text(s) => Some(s.as_str()),
            Array(..) => None,
        }
    }
}

impl ChatCompletionRequestAssistantMessage_Content {
    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestAssistantMessage_Content::*;
        match self {
            Text(s) => Some(s.as_str()),
            Array(..) => None,
        }
    }
}

impl ChatCompletionRequestToolMessage_Content {
    pub fn text_content(&self) -> Option<&str> {
        use ChatCompletionRequestToolMessage_Content::*;
        match self {
            Text(s) => Some(s.as_str()),
            Array(..) => None,
        }
    }
}