phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix for importllib compatibility issue for python 3.8

Browse source 
was tested for python versions 3.8, 3.9, 3.10, 3.11, 3.12
This commit is contained in:
Alexandre Sorokine 2024-02-16 15:52:39 -05:00
parent b96330c5dd
commit 3c919cbbe7
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/utils.py
View file

@ -31,16 +31,17 @@ from dataclasses import (
field,
) # for storing API inputs, outputs, and metadata
# import pkg_resources
from importlib import resources
# filename = pkg_resources.resource_filename(__name__, "llms/tokenizers")
try:
filename = str(
resources.files().joinpath("llms/tokenizers") # type: ignore
) # for python 3.8 and 3.12
# this works in python 3.8
import pkg_resources
filename = pkg_resources.resource_filename(__name__, "llms/tokenizers")
# try:
# filename = str(
# resources.files().joinpath("llms/tokenizers") # type: ignore
# ) # for python 3.8 and 3.12
except:
# this works in python 3.9+
from importlib import resources
filename = str(
resources.files(litellm).joinpath("llms/tokenizers") # for python 3.10
) # for python 3.10+