1use serde::{Deserialize, Serialize};
12use utoipa::OpenApi;
13
14
15use crate::models;
16
17#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, OpenApi)]
18pub struct OpenAiModel {
19 #[serde(rename = "messages", skip_serializing_if = "Option::is_none")]
21 pub messages: Option<Vec<models::OpenAiMessage>>,
22 #[serde(rename = "tools", skip_serializing_if = "Option::is_none")]
24 pub tools: Option<Vec<models::AnyscaleModelToolsInner>>,
25 #[serde(rename = "toolIds", skip_serializing_if = "Option::is_none")]
27 pub tool_ids: Option<Vec<String>>,
28 #[serde(rename = "knowledgeBase", skip_serializing_if = "Option::is_none")]
29 pub knowledge_base: Option<models::AnyscaleModelKnowledgeBase>,
30 #[serde(rename = "knowledgeBaseId", skip_serializing_if = "Option::is_none")]
32 pub knowledge_base_id: Option<String>,
33 #[serde(rename = "provider")]
35 pub provider: Provider,
36 #[serde(rename = "model")]
38 pub model: Model,
39 #[serde(rename = "fallbackModels", skip_serializing_if = "Option::is_none")]
41 pub fallback_models: Option<Vec<FallbackModels>>,
42 #[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
44 pub temperature: Option<f64>,
45 #[serde(rename = "maxTokens", skip_serializing_if = "Option::is_none")]
47 pub max_tokens: Option<f64>,
48 #[serde(
50 rename = "emotionRecognitionEnabled",
51 skip_serializing_if = "Option::is_none"
52 )]
53 pub emotion_recognition_enabled: Option<bool>,
54 #[serde(rename = "numFastTurns", skip_serializing_if = "Option::is_none")]
56 pub num_fast_turns: Option<f64>,
57}
58
59impl OpenAiModel {
60 pub fn new(provider: Provider, model: Model) -> OpenAiModel {
61 OpenAiModel {
62 messages: None,
63 tools: None,
64 tool_ids: None,
65 knowledge_base: None,
66 knowledge_base_id: None,
67 provider,
68 model,
69 fallback_models: None,
70 temperature: None,
71 max_tokens: None,
72 emotion_recognition_enabled: None,
73 num_fast_turns: None,
74 }
75 }
76}
77#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, OpenApi)]
79pub enum Provider {
80 #[serde(rename = "openai")]
81 Openai,
82}
83
84impl Default for Provider {
85 fn default() -> Provider {
86 Self::Openai
87 }
88}
89#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, OpenApi)]
91pub enum Model {
92 #[serde(rename = "gpt-4.5-preview")]
93 Gpt4Period5Preview,
94 #[serde(rename = "chatgpt-4o-latest")]
95 Chatgpt4oLatest,
96 #[serde(rename = "o3-mini")]
97 O3Mini,
98 #[serde(rename = "o1-preview")]
99 O1Preview,
100 #[serde(rename = "o1-preview-2024-09-12")]
101 O1Preview20240912,
102 #[serde(rename = "o1-mini")]
103 O1Mini,
104 #[serde(rename = "o1-mini-2024-09-12")]
105 O1Mini20240912,
106 #[serde(rename = "gpt-4o-realtime-preview-2024-10-01")]
107 Gpt4oRealtimePreview20241001,
108 #[serde(rename = "gpt-4o-realtime-preview-2024-12-17")]
109 Gpt4oRealtimePreview20241217,
110 #[serde(rename = "gpt-4o-mini-realtime-preview-2024-12-17")]
111 Gpt4oMiniRealtimePreview20241217,
112 #[serde(rename = "gpt-4o-mini")]
113 Gpt4oMini,
114 #[serde(rename = "gpt-4o-mini-2024-07-18")]
115 Gpt4oMini20240718,
116 #[serde(rename = "gpt-4o")]
117 Gpt4o,
118 #[serde(rename = "gpt-4o-2024-05-13")]
119 Gpt4o20240513,
120 #[serde(rename = "gpt-4o-2024-08-06")]
121 Gpt4o20240806,
122 #[serde(rename = "gpt-4o-2024-11-20")]
123 Gpt4o20241120,
124 #[serde(rename = "gpt-4-turbo")]
125 Gpt4Turbo,
126 #[serde(rename = "gpt-4-turbo-2024-04-09")]
127 Gpt4Turbo20240409,
128 #[serde(rename = "gpt-4-turbo-preview")]
129 Gpt4TurboPreview,
130 #[serde(rename = "gpt-4-0125-preview")]
131 Gpt40125Preview,
132 #[serde(rename = "gpt-4-1106-preview")]
133 Gpt41106Preview,
134 #[serde(rename = "gpt-4")]
135 Gpt4,
136 #[serde(rename = "gpt-4-0613")]
137 Gpt40613,
138 #[serde(rename = "gpt-3.5-turbo")]
139 Gpt3Period5Turbo,
140 #[serde(rename = "gpt-3.5-turbo-0125")]
141 Gpt3Period5Turbo0125,
142 #[serde(rename = "gpt-3.5-turbo-1106")]
143 Gpt3Period5Turbo1106,
144 #[serde(rename = "gpt-3.5-turbo-16k")]
145 Gpt3Period5Turbo16k,
146 #[serde(rename = "gpt-3.5-turbo-0613")]
147 Gpt3Period5Turbo0613,
148}
149
150impl Default for Model {
151 fn default() -> Model {
152 Self::Gpt4Period5Preview
153 }
154}
155#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, OpenApi)]
157pub enum FallbackModels {
158 #[serde(rename = "gpt-4.5-preview")]
159 Gpt4Period5Preview,
160 #[serde(rename = "chatgpt-4o-latest")]
161 Chatgpt4oLatest,
162 #[serde(rename = "o3-mini")]
163 O3Mini,
164 #[serde(rename = "o1-preview")]
165 O1Preview,
166 #[serde(rename = "o1-preview-2024-09-12")]
167 O1Preview20240912,
168 #[serde(rename = "o1-mini")]
169 O1Mini,
170 #[serde(rename = "o1-mini-2024-09-12")]
171 O1Mini20240912,
172 #[serde(rename = "gpt-4o-realtime-preview-2024-10-01")]
173 Gpt4oRealtimePreview20241001,
174 #[serde(rename = "gpt-4o-realtime-preview-2024-12-17")]
175 Gpt4oRealtimePreview20241217,
176 #[serde(rename = "gpt-4o-mini-realtime-preview-2024-12-17")]
177 Gpt4oMiniRealtimePreview20241217,
178 #[serde(rename = "gpt-4o-mini")]
179 Gpt4oMini,
180 #[serde(rename = "gpt-4o-mini-2024-07-18")]
181 Gpt4oMini20240718,
182 #[serde(rename = "gpt-4o")]
183 Gpt4o,
184 #[serde(rename = "gpt-4o-2024-05-13")]
185 Gpt4o20240513,
186 #[serde(rename = "gpt-4o-2024-08-06")]
187 Gpt4o20240806,
188 #[serde(rename = "gpt-4o-2024-11-20")]
189 Gpt4o20241120,
190 #[serde(rename = "gpt-4-turbo")]
191 Gpt4Turbo,
192 #[serde(rename = "gpt-4-turbo-2024-04-09")]
193 Gpt4Turbo20240409,
194 #[serde(rename = "gpt-4-turbo-preview")]
195 Gpt4TurboPreview,
196 #[serde(rename = "gpt-4-0125-preview")]
197 Gpt40125Preview,
198 #[serde(rename = "gpt-4-1106-preview")]
199 Gpt41106Preview,
200 #[serde(rename = "gpt-4")]
201 Gpt4,
202 #[serde(rename = "gpt-4-0613")]
203 Gpt40613,
204 #[serde(rename = "gpt-3.5-turbo")]
205 Gpt3Period5Turbo,
206 #[serde(rename = "gpt-3.5-turbo-0125")]
207 Gpt3Period5Turbo0125,
208 #[serde(rename = "gpt-3.5-turbo-1106")]
209 Gpt3Period5Turbo1106,
210 #[serde(rename = "gpt-3.5-turbo-16k")]
211 Gpt3Period5Turbo16k,
212 #[serde(rename = "gpt-3.5-turbo-0613")]
213 Gpt3Period5Turbo0613,
214}
215
216impl Default for FallbackModels {
217 fn default() -> FallbackModels {
218 Self::Gpt4Period5Preview
219 }
220}