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

fix import path

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-23 14:48:13 -07:00
parent ce58c53ff1
commit 101c350f98
7 changed files with 7 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm_proxy/proxy_server.py
View file

@ -372,15 +372,7 @@ from fastapi.security.api_key import APIKeyHeader
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
# import enterprise folder # import enterprise folder
try: import enterprise
# when using litellm cli
import litellm_proxy.enterprise as enterprise
except Exception:
# when using litellm docker image
try:
import enterprise # type: ignore
except Exception:
pass
server_root_path = os.getenv("SERVER_ROOT_PATH", "") server_root_path = os.getenv("SERVER_ROOT_PATH", "")
_license_check = LicenseCheck() _license_check = LicenseCheck()

2
tests/local_testing/test_blocked_user_list.py
View file

@ -29,7 +29,7 @@ from litellm import Router, mock_completion
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
from litellm.caching.caching import DualCache from litellm.caching.caching import DualCache
from litellm_proxy._types import UserAPIKeyAuth from litellm_proxy._types import UserAPIKeyAuth
from litellm_proxy.enterprise.enterprise_hooks.blocked_user_list import ( from enterprise.enterprise_hooks.blocked_user_list import (
_ENTERPRISE_BlockedUserList, _ENTERPRISE_BlockedUserList,
) )
from litellm_proxy.management_endpoints.internal_user_endpoints import ( from litellm_proxy.management_endpoints.internal_user_endpoints import (

2
tests/local_testing/test_llm_guard.py
View file

@ -16,7 +16,7 @@ sys.path.insert(
) # Adds the parent directory to the system path ) # Adds the parent directory to the system path
import pytest import pytest
import litellm import litellm
from litellm_proxy.enterprise.enterprise_hooks.llm_guard import _ENTERPRISE_LLMGuard from enterprise.enterprise_hooks.llm_guard import _ENTERPRISE_LLMGuard
from litellm import Router, mock_completion from litellm import Router, mock_completion
from litellm_proxy.utils import ProxyLogging, hash_token from litellm_proxy.utils import ProxyLogging, hash_token
from litellm_proxy._types import UserAPIKeyAuth from litellm_proxy._types import UserAPIKeyAuth

2
tests/local_testing/test_openai_moderations_hook.py
View file

@ -16,7 +16,7 @@ sys.path.insert(
) # Adds the parent directory to the system path ) # Adds the parent directory to the system path
import pytest import pytest
import litellm import litellm
from litellm_proxy.enterprise.enterprise_hooks.openai_moderation import ( from enterprise.enterprise_hooks.openai_moderation import (
_ENTERPRISE_OpenAI_Moderation, _ENTERPRISE_OpenAI_Moderation,
) )
from litellm import Router, mock_completion from litellm import Router, mock_completion

2
tests/local_testing/test_secret_detect_hook.py
View file

@ -29,7 +29,7 @@ from litellm import Router, mock_completion
from litellm.caching.caching import DualCache from litellm.caching.caching import DualCache
from litellm.integrations.custom_logger import CustomLogger from litellm.integrations.custom_logger import CustomLogger
from litellm_proxy._types import UserAPIKeyAuth from litellm_proxy._types import UserAPIKeyAuth
from litellm_proxy.enterprise.enterprise_hooks.secret_detection import ( from enterprise.enterprise_hooks.secret_detection import (
_ENTERPRISE_SecretDetection, _ENTERPRISE_SecretDetection,
) )
from litellm_proxy.proxy_server import chat_completion from litellm_proxy.proxy_server import chat_completion

2
tests/proxy_unit_tests/test_banned_keyword_list.py
View file

@ -15,7 +15,7 @@ sys.path.insert(
) # Adds the parent directory to the system path ) # Adds the parent directory to the system path
import pytest import pytest
import litellm import litellm
from litellm_proxy.enterprise.enterprise_hooks.banned_keywords import ( from enterprise.enterprise_hooks.banned_keywords import (
_ENTERPRISE_BannedKeywords, _ENTERPRISE_BannedKeywords,
) )
from litellm import Router, mock_completion from litellm import Router, mock_completion

2
tests/proxy_unit_tests/test_proxy_reject_logging.py
View file

@ -34,7 +34,7 @@ from litellm import Router, mock_completion
from litellm.caching.caching import DualCache from litellm.caching.caching import DualCache
from litellm.integrations.custom_logger import CustomLogger from litellm.integrations.custom_logger import CustomLogger
from litellm_proxy._types import UserAPIKeyAuth from litellm_proxy._types import UserAPIKeyAuth
from litellm_proxy.enterprise.enterprise_hooks.secret_detection import ( from enterprise.enterprise_hooks.secret_detection import (
_ENTERPRISE_SecretDetection, _ENTERPRISE_SecretDetection,
) )
from litellm_proxy.proxy_server import ( from litellm_proxy.proxy_server import (