mii/openai-api-rs
1
0
Fork 0
mirror of https://github.com/mii443/openai-api-rs.git synced 2025-08-22 23:25:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add gpt-4o mini model

Browse Source
This commit is contained in:
Dongri Jin
2024-07-19 05:48:23 +09:00
parent 4fcdd1d582
commit a0bd769dbb
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/chat_completion.rs
View File

@ -1,6 +1,6 @@
use openai_api_rs::v1::api::OpenAIClient; use openai_api_rs::v1::api::OpenAIClient;
use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest}; 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; use std::env;
#[tokio::main] #[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 client = OpenAIClient::new(env::var("OPENAI_API_KEY").unwrap().to_string());
let req = ChatCompletionRequest::new( let req = ChatCompletionRequest::new(
GPT4_O.to_string(), GPT4_O_MINI.to_string(),
vec![chat_completion::ChatCompletionMessage { vec![chat_completion::ChatCompletionMessage {
role: chat_completion::MessageRole::user, role: chat_completion::MessageRole::user,
content: chat_completion::Content::Text(String::from("What is bitcoin?")), content: chat_completion::Content::Text(String::from("What is bitcoin?")),

4
src/v1/common.rs
View File

@ -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 // https://platform.openai.com/docs/models/gpt-4o
pub const GPT4_O: &str = "gpt-4o"; pub const GPT4_O: &str = "gpt-4o";
pub const GPT4_O_2024_05_13: &str = "gpt-4o-2024-05-13"; pub const GPT4_O_2024_05_13: &str = "gpt-4o-2024-05-13";