diff --git a/tiktoken-rs/src/model.rs b/tiktoken-rs/src/model.rs index f0c98ab..a946eb3 100644 --- a/tiktoken-rs/src/model.rs +++ b/tiktoken-rs/src/model.rs @@ -44,7 +44,7 @@ pub fn get_context_size(model: &str) -> usize { if starts_with_any!(model, "gpt-4") { return 8192; } - if starts_with_any!(model, "gpt-3.5-turbo-0125"){ + if starts_with_any!(model, "gpt-3.5-turbo-0125") { return 16_385; } if starts_with_any!(model, "gpt-3.5-turbo-1106") {