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

Use packaging over deprecated pkg_resources

Browse source 
Fix #2818
This commit is contained in:
Nicolas van Kempen 2024-04-04 08:30:00 -04:00
parent 053a06df97
commit 4ac3bfef62
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/integrations/lunary.py
View file

@ -6,6 +6,8 @@ import dotenv
import importlib
import sys
import packaging
dotenv.load_dotenv()
@ -56,13 +58,12 @@ class LunaryLogger:
def __init__(self):
try:
import lunary
from pkg_resources import parse_version
version = importlib.metadata.version("lunary")
# if version < 0.1.43 then raise ImportError
if parse_version(version) < parse_version("0.1.43"):
if packaging.version.Version(version) < packaging.version.Version("0.1.43"):
print(
"Lunary version outdated. Required: > 0.1.43. Upgrade via 'pip install lunary --upgrade'"
"Lunary version outdated. Required: >= 0.1.43. Upgrade via 'pip install lunary --upgrade'"
)
raise ImportError