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
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

/*
 * 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
 */
/// pub ChatCompletionToolChoiceOption : Controls which (if any) tool is called by the model. `none` means the model will not call any tool and instead generates a message. `auto` means the model can pick between generating a message or calling one or more tools. `required` means the model must call one or more tools. Specifying a particular tool via `{\"type\": \"function\", \"function\": {\"name\": \"my_function\"}}` forces the model to call that tool.  `none` is the default when no tools are present. `auto` is the default if tools are present.

#[allow(unused_imports)]
use serde_json::Value;

use crate::ChatCompletionNamedToolChoice;

/// # on openapi.yaml
///
/// ```yaml
/// ChatCompletionToolChoiceOption:
///   description: >
///     Controls which (if any) tool is called by the model.
///
///     `none` means the model will not call any tool and instead generates a
///     message.
///
///     `auto` means the model can pick between generating a message or calling
///     one or more tools.
///
///     `required` means the model must call one or more tools.
///
///     Specifying a particular tool via `{"type": "function", "function":
///     {"name": "my_function"}}` forces the model to call that tool.
///
///
///     `none` is the default when no tools are present. `auto` is the default
///     if tools are present.
///   oneOf:
///     - type: string
///       description: >
///         `none` means the model will not call any tool and instead generates
///         a message. `auto` means the model can pick between generating a
///         message or calling one or more tools. `required` means the model
///         must call one or more tools.
///       enum:
///         - none
///         - auto
///         - required
///     - $ref: "#/components/schemas/ChatCompletionNamedToolChoice"
/// ```
#[derive(Debug, Serialize, Deserialize)]
#[serde(untagged)]
pub enum ChatCompletionToolChoiceOption {
    /// Str
    String(ChatCompletionToolChoiceStrEnum),
    /// Named tool choice variant for specific tool selection
    VariantNamedToolChoice(ChatCompletionNamedToolChoice),
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum ChatCompletionToolChoiceStrEnum {
    /// Model will not call any tool and instead generates a message
    #[serde(rename = "none")]
    None,
    /// Model can pick between generating a message or calling tools
    #[serde(rename = "auto")]
    Auto,
    /// Model must call one or more tools
    #[serde(rename = "required")]
    Required,
}

#[test]
fn test_cct_co() {
    assert_eq!(
        serde_json::to_string(&ChatCompletionToolChoiceOption::String(
            ChatCompletionToolChoiceStrEnum::None
        ))
        .unwrap(),
        r#""none""#
    );
    assert_eq!(
        serde_json::to_string(&ChatCompletionToolChoiceOption::VariantNamedToolChoice(
            ChatCompletionNamedToolChoice {
                function: crate::models::AssistantsNamedToolChoiceFunction { name: "Emm".into() },
                _type: "function".into()
            }
        ))
        .unwrap()
        .to_string(),
        r#"{"function":{"name":"Emm"},"type":"function"}"#
    );
}