diff options
Diffstat (limited to 'src/client/openai.rs')
| -rw-r--r-- | src/client/openai.rs | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs index 06a5316..dbe2661 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -19,11 +19,13 @@ use std::env; const API_BASE: &str = "https://api.openai.com/v1"; -const MODELS: [(&str, usize); 4] = [ +const MODELS: [(&str, usize); 6] = [ ("gpt-3.5-turbo", 4096), - ("gpt-3.5-turbo-16k", 16384), + ("gpt-3.5-turbo-16k", 16385), + ("gpt-3.5-turbo-1106", 16385), ("gpt-4", 8192), ("gpt-4-32k", 32768), + ("gpt-4-1106-preview", 128000), ]; pub const OPENAI_TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2); |