From 4d0f3f528f3a0c10202ce8c9ceba77d4511d4c8d Mon Sep 17 00:00:00 2001 From: 0xMochan Date: Fri, 13 Sep 2024 14:30:06 -0400 Subject: [PATCH] feat: add o1-preview and o1-mini --- rig-core/src/providers/openai.rs | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/rig-core/src/providers/openai.rs b/rig-core/src/providers/openai.rs index 586dc9ce..8c12a745 100644 --- 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