Skip to content

Commit 3b04874

Browse files
committed
Fix for gpt-4o-mini in token count test
Signed-off-by: Mark Sze <[email protected]>
1 parent a898cc9 commit 3b04874

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_token_count.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def test_model_aliases():
142142
assert get_max_token_limit("gpt4") == get_max_token_limit("gpt-4")
143143
assert get_max_token_limit("gpt4-32k") == get_max_token_limit("gpt-4-32k")
144144
assert get_max_token_limit("gpt4o") == get_max_token_limit("gpt-4o")
145-
assert get_max_token_limit("gpt4omini") == get_max_token_limit("gpt-4o-mini")
145+
assert get_max_token_limit("gpt4o-mini") == get_max_token_limit("gpt-4o-mini")
146146

147147

148148
if __name__ == "__main__":

0 commit comments

Comments
 (0)