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

fix install error

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-09 14:15:19 -07:00
parent cf5c3c61db
commit 6ec91227cd
2 changed files with 30 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

31
litellm/utils.py
View file

@ -5,6 +5,8 @@ import litellm, openai
import random, uuid, requests
import datetime, time
import tiktoken
import pkg_resources
from pkg_resources import DistributionNotFound, VersionConflict
encoding = tiktoken.get_encoding("cl100k_base")
from .integrations.helicone import HeliconeLogger
from .integrations.aispend import AISpendLogger
@ -36,14 +38,33 @@ def print_verbose(print_statement):
####### Package Import Handler ###################
import importlib
import subprocess
def install_and_import(package):
def install_and_import(package: str):
try:
importlib.import_module(package)
except ImportError:
# Import the module
module = importlib.import_module(package)
# Get the package's information
dist = pkg_resources.get_distribution(package)
required = [req for req in pkg_resources.get_distribution(package).requires()]
# Check if there are dependencies
if required:
for req in required:
install_and_import(req.project_name)
else:
print(f"{package} has been successfully installed with no dependencies.")
except (DistributionNotFound, ImportError):
print(f"{package} is not installed. Installing...")
subprocess.call([sys.executable, '-m', 'pip', 'install', package])
finally:
subprocess.call([sys.executable, "-m", "pip", "install", package])
globals()[package] = importlib.import_module(package)
except VersionConflict as vc:
print(f"Detected version conflict for {package}. Upgrading...")
subprocess.call([sys.executable, "-m", "pip", "install", "--upgrade", package])
globals()[package] = importlib.import_module(package)
finally:
if package not in globals().keys():
globals()[package] = importlib.import_module(package)
##################################################
####### LOGGING ###################