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);