From a0bd769dbb31d60f049dacf87517dd5053f76e1f Mon Sep 17 00:00:00 2001
From: Dongri Jin <dongrium@gmail.com>
Date: Fri, 19 Jul 2024 05:48:23 +0900
Subject: [PATCH] Add gpt-4o mini model

---
 examples/chat_completion.rs | 4 ++--
 src/v1/common.rs            | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/examples/chat_completion.rs b/examples/chat_completion.rs
index 7a31241..95a6e7d 100644
--- a/examples/chat_completion.rs
+++ b/examples/chat_completion.rs
@@ -1,6 +1,6 @@
 use openai_api_rs::v1::api::OpenAIClient;
 use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
-use openai_api_rs::v1::common::GPT4_O;
+use openai_api_rs::v1::common::GPT4_O_MINI;
 use std::env;
 
 #[tokio::main]
@@ -8,7 +8,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let client = OpenAIClient::new(env::var("OPENAI_API_KEY").unwrap().to_string());
 
     let req = ChatCompletionRequest::new(
-        GPT4_O.to_string(),
+        GPT4_O_MINI.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
index f0e3c5b..ab3fbc0 100644
--- a/src/v1/common.rs
+++ b/src/v1/common.rs
@@ -20,6 +20,10 @@ macro_rules! impl_builder_methods {
         }
     };
 }
+
+// https://platform.openai.com/docs/models/gpt-4o-mini
+pub const GPT4_O_MINI: &str = "gpt-4o-mini";
+
 // 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";