Skip to content

Commit 759340e

Browse files
Congyuwangzurawiki
andauthored
Add support for chatgpt-4o-latest (#85)
Co-authored-by: Roger Zurawicki <[email protected]>
1 parent b4ce7a6 commit 759340e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tiktoken-rs/src/tokenizer.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ const MODEL_PREFIX_TO_TOKENIZER: &[(&str, Tokenizer)] = &[
4949
// https://github.com/openai/tiktoken/blob/63527649963def8c759b0f91f2eb69a40934e468/tiktoken/model.py#L22
5050
const MODEL_TO_TOKENIZER: &[(&str, Tokenizer)] = &[
5151
// chat
52+
("chatgpt-4o-latest", Tokenizer::O200kBase),
5253
("gpt-4o", Tokenizer::O200kBase),
5354
("gpt-4", Tokenizer::Cl100kBase),
5455
("gpt-3.5-turbo", Tokenizer::Cl100kBase),
@@ -152,6 +153,10 @@ mod tests {
152153

153154
#[test]
154155
fn test_get_tokenizer() {
156+
assert_eq!(
157+
get_tokenizer("chatgpt-4o-latest"),
158+
Some(Tokenizer::O200kBase)
159+
);
155160
assert_eq!(
156161
get_tokenizer("gpt-4o-2024-05-13"),
157162
Some(Tokenizer::O200kBase)

0 commit comments

Comments
 (0)