phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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 f88545261c
commit 4270337092
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 importlib
import sys import sys
import packaging
dotenv.load_dotenv() dotenv.load_dotenv()
@ -56,13 +58,12 @@ class LunaryLogger:
def __init__(self): def __init__(self):
try: try:
import lunary import lunary
from pkg_resources import parse_version
version = importlib.metadata.version("lunary") version = importlib.metadata.version("lunary")
# if version < 0.1.43 then raise ImportError # 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( 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 raise ImportError