Add GPT4o models

dongri · May 14, 2024 · 1a8b5e3 · 1a8b5e3
1 parent df82373
commit 1a8b5e3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/chat_completion.rs b/examples/chat_completion.rs
@@ -1,13 +1,13 @@
 use openai_api_rs::v1::api::Client;
 use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
-use openai_api_rs::v1::common::GPT4;
+use openai_api_rs::v1::common::GPT4_O;
 use std::env;
 
 fn main() -> Result<(), Box<dyn std::error::Error>> {
     let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
 
     let req = ChatCompletionRequest::new(
-        GPT4.to_string(),
+        GPT4_O.to_string(),
         vec![chat_completion::ChatCompletionMessage {
             role: chat_completion::MessageRole::user,
             content: chat_completion::Content::Text(String::from("What is bitcoin?")),

diff --git a/src/v1/common.rs b/src/v1/common.rs
@@ -20,6 +20,9 @@ macro_rules! impl_builder_methods {
         }
     };
 }
+// https://platform.openai.com/docs/models/gpt-4o
+pub const GPT4_O: &str = "gpt-4o";
+pub const GPT4_O_2024_05_13: &str = "gpt-4o-2024-05-13";
 
 // https://platform.openai.com/docs/models/gpt-3-5
 pub const GPT3_5_TURBO_1106: &str = "gpt-3.5-turbo-1106";