1use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
15pub struct OpenAiModel {
16 #[serde(rename = "messages", skip_serializing_if = "Option::is_none")]
18 pub messages: Option<Vec<models::OpenAiMessage>>,
19 #[serde(rename = "tools", skip_serializing_if = "Option::is_none")]
21 pub tools: Option<Vec<models::AnyscaleModelToolsInner>>,
22 #[serde(rename = "toolIds", skip_serializing_if = "Option::is_none")]
24 pub tool_ids: Option<Vec<String>>,
25 #[serde(rename = "knowledgeBase", skip_serializing_if = "Option::is_none")]
26 pub knowledge_base: Option<models::AnyscaleModelKnowledgeBase>,
27 #[serde(rename = "knowledgeBaseId", skip_serializing_if = "Option::is_none")]
29 pub knowledge_base_id: Option<String>,
30 #[serde(rename = "provider")]
32 pub provider: Provider,
33 #[serde(rename = "model")]
35 pub model: Model,
36 #[serde(rename = "fallbackModels", skip_serializing_if = "Option::is_none")]
38 pub fallback_models: Option<Vec<FallbackModels>>,
39 #[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
41 pub temperature: Option<f64>,
42 #[serde(rename = "maxTokens", skip_serializing_if = "Option::is_none")]
44 pub max_tokens: Option<f64>,
45 #[serde(rename = "emotionRecognitionEnabled", skip_serializing_if = "Option::is_none")]
47 pub emotion_recognition_enabled: Option<bool>,
48 #[serde(rename = "numFastTurns", skip_serializing_if = "Option::is_none")]
50 pub num_fast_turns: Option<f64>,
51}
52
53impl OpenAiModel {
54 pub fn new(provider: Provider, model: Model) -> OpenAiModel {
55 OpenAiModel {
56 messages: None,
57 tools: None,
58 tool_ids: None,
59 knowledge_base: None,
60 knowledge_base_id: None,
61 provider,
62 model,
63 fallback_models: None,
64 temperature: None,
65 max_tokens: None,
66 emotion_recognition_enabled: None,
67 num_fast_turns: None,
68 }
69 }
70}
71#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
73pub enum Provider {
74 #[serde(rename = "openai")]
75 Openai,
76}
77
78impl Default for Provider {
79 fn default() -> Provider {
80 Self::Openai
81 }
82}
83#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
85pub enum Model {
86 #[serde(rename = "gpt-4.1")]
87 Gpt4Period1,
88 #[serde(rename = "gpt-4.1-mini")]
89 Gpt4Period1Mini,
90 #[serde(rename = "gpt-4.1-nano")]
91 Gpt4Period1Nano,
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-2024-07-18")]
113 Gpt4oMini20240718,
114 #[serde(rename = "gpt-4o-mini")]
115 Gpt4oMini,
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::Gpt4Period1
153 }
154}
155#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
157pub enum FallbackModels {
158 #[serde(rename = "gpt-4.1")]
159 Gpt4Period1,
160 #[serde(rename = "gpt-4.1-mini")]
161 Gpt4Period1Mini,
162 #[serde(rename = "gpt-4.1-nano")]
163 Gpt4Period1Nano,
164 #[serde(rename = "gpt-4.5-preview")]
165 Gpt4Period5Preview,
166 #[serde(rename = "chatgpt-4o-latest")]
167 Chatgpt4oLatest,
168 #[serde(rename = "o3-mini")]
169 O3Mini,
170 #[serde(rename = "o1-preview")]
171 O1Preview,
172 #[serde(rename = "o1-preview-2024-09-12")]
173 O1Preview20240912,
174 #[serde(rename = "o1-mini")]
175 O1Mini,
176 #[serde(rename = "o1-mini-2024-09-12")]
177 O1Mini20240912,
178 #[serde(rename = "gpt-4o-realtime-preview-2024-10-01")]
179 Gpt4oRealtimePreview20241001,
180 #[serde(rename = "gpt-4o-realtime-preview-2024-12-17")]
181 Gpt4oRealtimePreview20241217,
182 #[serde(rename = "gpt-4o-mini-realtime-preview-2024-12-17")]
183 Gpt4oMiniRealtimePreview20241217,
184 #[serde(rename = "gpt-4o-mini-2024-07-18")]
185 Gpt4oMini20240718,
186 #[serde(rename = "gpt-4o-mini")]
187 Gpt4oMini,
188 #[serde(rename = "gpt-4o")]
189 Gpt4o,
190 #[serde(rename = "gpt-4o-2024-05-13")]
191 Gpt4o20240513,
192 #[serde(rename = "gpt-4o-2024-08-06")]
193 Gpt4o20240806,
194 #[serde(rename = "gpt-4o-2024-11-20")]
195 Gpt4o20241120,
196 #[serde(rename = "gpt-4-turbo")]
197 Gpt4Turbo,
198 #[serde(rename = "gpt-4-turbo-2024-04-09")]
199 Gpt4Turbo20240409,
200 #[serde(rename = "gpt-4-turbo-preview")]
201 Gpt4TurboPreview,
202 #[serde(rename = "gpt-4-0125-preview")]
203 Gpt40125Preview,
204 #[serde(rename = "gpt-4-1106-preview")]
205 Gpt41106Preview,
206 #[serde(rename = "gpt-4")]
207 Gpt4,
208 #[serde(rename = "gpt-4-0613")]
209 Gpt40613,
210 #[serde(rename = "gpt-3.5-turbo")]
211 Gpt3Period5Turbo,
212 #[serde(rename = "gpt-3.5-turbo-0125")]
213 Gpt3Period5Turbo0125,
214 #[serde(rename = "gpt-3.5-turbo-1106")]
215 Gpt3Period5Turbo1106,
216 #[serde(rename = "gpt-3.5-turbo-16k")]
217 Gpt3Period5Turbo16k,
218 #[serde(rename = "gpt-3.5-turbo-0613")]
219 Gpt3Period5Turbo0613,
220}
221
222impl Default for FallbackModels {
223 fn default() -> FallbackModels {
224 Self::Gpt4Period1
225 }
226}
227