feat: add o1-preview and o1-mini

0xPlaygrounds · Sep 13, 2024 · 4d0f3f5 · 4d0f3f5
1 parent 0526b2f
commit 4d0f3f5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/rig-core/src/providers/openai.rs b/rig-core/src/providers/openai.rs
@@ -305,6 +305,14 @@ impl EmbeddingModel {
 // ================================================================
 // OpenAI Completion API
 // ================================================================
+/// `o1-preview` completion model
+pub const O1_PREVIEW: &str = "o1-preview";
+/// `o1-preview-2024-09-12` completion model
+pub const O1_PREVIEW_2024_09_12: &str = "o1-preview-2024-09-12";
+/// `o1-mini completion model
+pub const O1_MINI: &str = "o1-mini";
+/// `o1-mini-2024-09-12` completion model
+pub const O1_MINI_2024_09_12: &str = "o1-mini-2024-09-12";
 /// `gpt-4o` completion model
 pub const GPT_4O: &str = "gpt-4o";
 /// `gpt-4o-2024-05-13` completion model