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<Box<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.5-preview")]
87 Gpt4Period5Preview,
88 #[serde(rename = "chatgpt-4o-latest")]
89 Chatgpt4oLatest,
90 #[serde(rename = "o3-mini")]
91 O3Mini,
92 #[serde(rename = "o1-preview")]
93 O1Preview,
94 #[serde(rename = "o1-preview-2024-09-12")]
95 O1Preview20240912,
96 #[serde(rename = "o1-mini")]
97 O1Mini,
98 #[serde(rename = "o1-mini-2024-09-12")]
99 O1Mini20240912,
100 #[serde(rename = "gpt-4o-realtime-preview-2024-10-01")]
101 Gpt4oRealtimePreview20241001,
102 #[serde(rename = "gpt-4o-realtime-preview-2024-12-17")]
103 Gpt4oRealtimePreview20241217,
104 #[serde(rename = "gpt-4o-mini-realtime-preview-2024-12-17")]
105 Gpt4oMiniRealtimePreview20241217,
106 #[serde(rename = "gpt-4o-mini")]
107 Gpt4oMini,
108 #[serde(rename = "gpt-4o-mini-2024-07-18")]
109 Gpt4oMini20240718,
110 #[serde(rename = "gpt-4o")]
111 Gpt4o,
112 #[serde(rename = "gpt-4o-2024-05-13")]
113 Gpt4o20240513,
114 #[serde(rename = "gpt-4o-2024-08-06")]
115 Gpt4o20240806,
116 #[serde(rename = "gpt-4o-2024-11-20")]
117 Gpt4o20241120,
118 #[serde(rename = "gpt-4-turbo")]
119 Gpt4Turbo,
120 #[serde(rename = "gpt-4-turbo-2024-04-09")]
121 Gpt4Turbo20240409,
122 #[serde(rename = "gpt-4-turbo-preview")]
123 Gpt4TurboPreview,
124 #[serde(rename = "gpt-4-0125-preview")]
125 Gpt40125Preview,
126 #[serde(rename = "gpt-4-1106-preview")]
127 Gpt41106Preview,
128 #[serde(rename = "gpt-4")]
129 Gpt4,
130 #[serde(rename = "gpt-4-0613")]
131 Gpt40613,
132 #[serde(rename = "gpt-3.5-turbo")]
133 Gpt3Period5Turbo,
134 #[serde(rename = "gpt-3.5-turbo-0125")]
135 Gpt3Period5Turbo0125,
136 #[serde(rename = "gpt-3.5-turbo-1106")]
137 Gpt3Period5Turbo1106,
138 #[serde(rename = "gpt-3.5-turbo-16k")]
139 Gpt3Period5Turbo16k,
140 #[serde(rename = "gpt-3.5-turbo-0613")]
141 Gpt3Period5Turbo0613,
142}
143
144impl Default for Model {
145 fn default() -> Model {
146 Self::Gpt4Period5Preview
147 }
148}
149#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
151pub enum FallbackModels {
152 #[serde(rename = "gpt-4.5-preview")]
153 Gpt4Period5Preview,
154 #[serde(rename = "chatgpt-4o-latest")]
155 Chatgpt4oLatest,
156 #[serde(rename = "o3-mini")]
157 O3Mini,
158 #[serde(rename = "o1-preview")]
159 O1Preview,
160 #[serde(rename = "o1-preview-2024-09-12")]
161 O1Preview20240912,
162 #[serde(rename = "o1-mini")]
163 O1Mini,
164 #[serde(rename = "o1-mini-2024-09-12")]
165 O1Mini20240912,
166 #[serde(rename = "gpt-4o-realtime-preview-2024-10-01")]
167 Gpt4oRealtimePreview20241001,
168 #[serde(rename = "gpt-4o-realtime-preview-2024-12-17")]
169 Gpt4oRealtimePreview20241217,
170 #[serde(rename = "gpt-4o-mini-realtime-preview-2024-12-17")]
171 Gpt4oMiniRealtimePreview20241217,
172 #[serde(rename = "gpt-4o-mini")]
173 Gpt4oMini,
174 #[serde(rename = "gpt-4o-mini-2024-07-18")]
175 Gpt4oMini20240718,
176 #[serde(rename = "gpt-4o")]
177 Gpt4o,
178 #[serde(rename = "gpt-4o-2024-05-13")]
179 Gpt4o20240513,
180 #[serde(rename = "gpt-4o-2024-08-06")]
181 Gpt4o20240806,
182 #[serde(rename = "gpt-4o-2024-11-20")]
183 Gpt4o20241120,
184 #[serde(rename = "gpt-4-turbo")]
185 Gpt4Turbo,
186 #[serde(rename = "gpt-4-turbo-2024-04-09")]
187 Gpt4Turbo20240409,
188 #[serde(rename = "gpt-4-turbo-preview")]
189 Gpt4TurboPreview,
190 #[serde(rename = "gpt-4-0125-preview")]
191 Gpt40125Preview,
192 #[serde(rename = "gpt-4-1106-preview")]
193 Gpt41106Preview,
194 #[serde(rename = "gpt-4")]
195 Gpt4,
196 #[serde(rename = "gpt-4-0613")]
197 Gpt40613,
198 #[serde(rename = "gpt-3.5-turbo")]
199 Gpt3Period5Turbo,
200 #[serde(rename = "gpt-3.5-turbo-0125")]
201 Gpt3Period5Turbo0125,
202 #[serde(rename = "gpt-3.5-turbo-1106")]
203 Gpt3Period5Turbo1106,
204 #[serde(rename = "gpt-3.5-turbo-16k")]
205 Gpt3Period5Turbo16k,
206 #[serde(rename = "gpt-3.5-turbo-0613")]
207 Gpt3Period5Turbo0613,
208}
209
210impl Default for FallbackModels {
211 fn default() -> FallbackModels {
212 Self::Gpt4Period5Preview
213 }
214}
215