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

refactor: rename enterprise specific hooks and callbacks to be more precise

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-17 17:52:32 -08:00
parent ad81e19282
commit 73acdf3736
6 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/utils.py
View file

@ -34,6 +34,7 @@ from dataclasses import (
try:
# this works in python 3.8
import pkg_resources
filename = pkg_resources.resource_filename(__name__, "llms/tokenizers")
# try:
# filename = str(
@ -42,6 +43,7 @@ try:
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+
@ -87,16 +89,10 @@ from .exceptions import (
UnprocessableEntityError,
)
# Import Enterprise features
project_path = abspath(join(dirname(__file__), "..", ".."))
# Add the "enterprise" directory to sys.path
verbose_logger.debug(f"current project_path: {project_path}")
enterprise_path = abspath(join(project_path, "enterprise"))
sys.path.append(enterprise_path)
verbose_logger.debug(f"sys.path: {sys.path}")
try:
from enterprise.callbacks.generic_api_callback import GenericAPILogger
from .proxy.enterprise.enterprise_callbacks.generic_api_callback import (
GenericAPILogger,
)
except Exception as e:
verbose_logger.debug(f"Exception import enterprise features {str(e)}")