This commit is contained in:
0xiaorun 2025-04-24 01:01:58 -07:00 committed by GitHub
commit fd83817fb8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1627,10 +1627,7 @@ def openai_token_counter( # noqa: PLR0915
"""
print_verbose(f"LiteLLM: Utils - Counting tokens for OpenAI model={model}")
try:
if "gpt-4o" in model:
encoding = tiktoken.get_encoding("o200k_base")
else:
encoding = tiktoken.encoding_for_model(model)
encoding = tiktoken.encoding_for_model(model)
except KeyError:
print_verbose("Warning: model not found. Using cl100k_base encoding.")
encoding = tiktoken.get_encoding("cl100k_base")