From ce1f9929f23b1983a679033ddfe93811d6519b64 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 10 Apr 2024 09:27:39 +0800 Subject: refactor: update openai models list (#396) --- src/client/openai.rs | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'src/client/openai.rs') diff --git a/src/client/openai.rs b/src/client/openai.rs index 83dd65d..f7378dd 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -12,14 +12,12 @@ use serde_json::{json, Value}; const API_BASE: &str = "https://api.openai.com/v1"; -const MODELS: [(&str, usize, &str); 5] = [ - // https://platform.openai.com/docs/models/gpt-3-5-turbo +const MODELS: [(&str, usize, &str); 4] = [ + // https://platform.openai.com/docs/models ("gpt-3.5-turbo", 16385, "text"), - ("gpt-3.5-turbo-1106", 16385, "text"), - // https://platform.openai.com/docs/models/gpt-4-and-gpt-4-turbo - ("gpt-4-turbo-preview", 128000, "text"), - ("gpt-4-1106-preview", 128000, "text"), - ("gpt-4-vision-preview", 128000, "text,vision"), + ("gpt-4-turbo", 128000, "text,vision"), + ("gpt-4", 8192, "text"), + ("gpt-4-32k", 32768, "text"), ]; pub const OPENAI_TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2); -- cgit v1.2.3