phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): add try-except around the tiktoken local import

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-15 17:57:23 -08:00
parent 4be6d21c65
commit aa78f55dff
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -34,7 +34,14 @@ from dataclasses import (
from importlib import resources
# filename = pkg_resources.resource_filename(__name__, "llms/tokenizers")
filename = str(resources.files(litellm).joinpath("llms/tokenizers"))
try:
filename = str(
resources.files().joinpath("llms/tokenizers") # type: ignore
) # for python 3.8 and 3.12
except:
filename = str(
resources.files(litellm).joinpath("llms/tokenizers") # for python 3.10
) # for python 3.10+
os.environ[
"TIKTOKEN_CACHE_DIR"
] = filename # use local copy of tiktoken b/c of - https://github.com/BerriAI/litellm/issues/1071