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

fix import loc

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-23 14:53:16 -07:00
parent 101c350f98
commit 444acbccf4
5 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_secret_detect_hook.py
View file

@ -262,7 +262,7 @@ async def test_chat_completion_request_with_redaction():
Ensures that the secret is redacted EVEN on the callback Ensures that the secret is redacted EVEN on the callback
""" """
from litellm.proxy import proxy_server from litellm_proxy import proxy_server
setattr(proxy_server, "llm_router", router) setattr(proxy_server, "llm_router", router)
_test_logger = testLogger() _test_logger = testLogger()

2
tests/proxy_unit_tests/test_key_generate_prisma.py
View file

@ -3295,7 +3295,7 @@ async def test_aadmin_only_routes(prisma_client):
"allowed_routes": ["/embeddings", "/key/generate"], "allowed_routes": ["/embeddings", "/key/generate"],
"admin_only_routes": ["/key/generate"], "admin_only_routes": ["/key/generate"],
} }
from litellm.proxy import proxy_server from litellm_proxy import proxy_server
initial_general_settings = getattr(proxy_server, "general_settings") initial_general_settings = getattr(proxy_server, "general_settings")

2
tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py
View file

@ -12,7 +12,7 @@ sys.path.insert(
0, os.path.abspath("../..") 0, os.path.abspath("../..")
) # Adds-the parent directory to the system path ) # Adds-the parent directory to the system path
from litellm.proxy import proxy_server from litellm_proxy import proxy_server
from litellm_proxy.common_utils.encrypt_decrypt_utils import ( from litellm_proxy.common_utils.encrypt_decrypt_utils import (
decrypt_value_helper, decrypt_value_helper,
encrypt_value_helper, encrypt_value_helper,

2
tests/proxy_unit_tests/test_proxy_reject_logging.py
View file

@ -129,7 +129,7 @@ async def test_chat_completion_request_with_redaction(route, body):
Ensures that the secret is redacted EVEN on the callback Ensures that the secret is redacted EVEN on the callback
""" """
from litellm.proxy import proxy_server from litellm_proxy import proxy_server
setattr(proxy_server, "llm_router", router) setattr(proxy_server, "llm_router", router)
_test_logger = testLogger() _test_logger = testLogger()

2
tests/proxy_unit_tests/test_user_api_key_auth.py
View file

@ -287,7 +287,7 @@ async def test_auth_with_allowed_routes(route, should_raise_error):
general_settings = {"allowed_routes": ["/embeddings"]} general_settings = {"allowed_routes": ["/embeddings"]}
from fastapi import Request from fastapi import Request
from litellm.proxy import proxy_server from litellm_proxy import proxy_server
initial_general_settings = getattr(proxy_server, "general_settings") initial_general_settings = getattr(proxy_server, "general_settings")