openai-struct 0.0.4

利用openai的openapi生成的rust结构体
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

/*
 * OpenAI API
 *
 * The OpenAI REST API. Please see pub https://platform.openai.com/docs/api-reference for more details.
 *
 * OpenAPI spec pub version: 2.3.0
 *
 * Generated pub by: https://github.com/swagger-api/swagger-codegen.git
 */

use crate::{
    ChatCompletionRequestAssistantMessage, ChatCompletionRequestDeveloperMessage,
    ChatCompletionRequestFunctionMessage, ChatCompletionRequestSystemMessage,
    ChatCompletionRequestToolMessage, ChatCompletionRequestUserMessage,
};
#[allow(unused_imports)]
use serde_json::Value;

#[derive(Debug, PartialEq, Serialize, Deserialize)]
#[serde(tag = "role")]
pub enum ChatCompletionRequestMessage {
    #[serde(rename = "developer")]
    Developer(ChatCompletionRequestDeveloperMessage),
    #[serde(rename = "system")]
    System(ChatCompletionRequestSystemMessage),
    #[serde(rename = "user")]
    User(ChatCompletionRequestUserMessage),
    #[serde(rename = "assistant")]
    Assistant(ChatCompletionRequestAssistantMessage),
    #[serde(rename = "tool")]
    Tool(ChatCompletionRequestToolMessage),
    #[serde(rename = "function")]
    Function(ChatCompletionRequestFunctionMessage),
}

#[test]
fn test_serde_chat_completion_request_message() {
    let tmp = ChatCompletionRequestMessage::User(ChatCompletionRequestUserMessage {
        content: crate::ChatCompletionRequestUserMessageContent::Text("once upon a time...".into()),
        name: None,
    });

    let serialized = serde_json::to_string(&tmp).unwrap();
    assert_eq!(
        serialized,
        r#"{"role":"user","content":"once upon a time...","name":null}"#
    );

    let deserialized: ChatCompletionRequestMessage = serde_json::from_str(&serialized).unwrap();
    assert_eq!(tmp, deserialized);

    let emm: ChatCompletionRequestMessage =
        serde_json::from_str(r#"{"content":"once upon a time...","name":null,"role":"user"}"#)
            .unwrap();
    assert_eq!(tmp, emm);
}