diff --git a/crates/language_models/src/provider/open_ai.rs b/crates/language_models/src/provider/open_ai.rs index 36742b0035f511b8552826a34ce26c4a1432300f..f5529757b5f064b1c208beb3a23a743c53a504b8 100644 --- a/crates/language_models/src/provider/open_ai.rs +++ b/crates/language_models/src/provider/open_ai.rs @@ -361,7 +361,10 @@ pub fn count_open_ai_tokens( .collect::>(); match model { - open_ai::Model::Custom { .. } | open_ai::Model::O1Mini | open_ai::Model::O1 => { + open_ai::Model::Custom { .. } + | open_ai::Model::O1Mini + | open_ai::Model::O1 + | open_ai::Model::O3Mini => { tiktoken_rs::num_tokens_from_messages("gpt-4", &messages) } _ => tiktoken_rs::num_tokens_from_messages(model.id(), &messages),