From 4ac3bfef62b23b61cbf8b155d38c89a4313f25a3 Mon Sep 17 00:00:00 2001 From: Nicolas van Kempen Date: Thu, 4 Apr 2024 08:30:00 -0400 Subject: [PATCH] Use packaging over deprecated pkg_resources Fix #2818 --- litellm/integrations/lunary.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/integrations/lunary.py b/litellm/integrations/lunary.py index 908fa74fce..6ddf2ca599 100644 --- a/litellm/integrations/lunary.py +++ b/litellm/integrations/lunary.py @@ -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