From 444acbccf4729c777312a7ffe2279e1ed1c06cf5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 23 Apr 2025 14:53:16 -0700 Subject: [PATCH] fix import loc --- tests/local_testing/test_secret_detect_hook.py | 2 +- tests/proxy_unit_tests/test_key_generate_prisma.py | 2 +- tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py | 2 +- tests/proxy_unit_tests/test_proxy_reject_logging.py | 2 +- tests/proxy_unit_tests/test_user_api_key_auth.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/local_testing/test_secret_detect_hook.py b/tests/local_testing/test_secret_detect_hook.py index 958861a0f0..b6d768bb1a 100644 --- a/tests/local_testing/test_secret_detect_hook.py +++ b/tests/local_testing/test_secret_detect_hook.py @@ -262,7 +262,7 @@ async def test_chat_completion_request_with_redaction(): 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) _test_logger = testLogger() diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index 81cdc031d4..2af0ce4469 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -3295,7 +3295,7 @@ async def test_aadmin_only_routes(prisma_client): "allowed_routes": ["/embeddings", "/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") diff --git a/tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py b/tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py index b24f4d7863..3d17d27801 100644 --- a/tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py +++ b/tests/proxy_unit_tests/test_proxy_encrypt_decrypt.py @@ -12,7 +12,7 @@ sys.path.insert( 0, os.path.abspath("../..") ) # 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 ( decrypt_value_helper, encrypt_value_helper, diff --git a/tests/proxy_unit_tests/test_proxy_reject_logging.py b/tests/proxy_unit_tests/test_proxy_reject_logging.py index 38d9ac3d22..4efad2d086 100644 --- a/tests/proxy_unit_tests/test_proxy_reject_logging.py +++ b/tests/proxy_unit_tests/test_proxy_reject_logging.py @@ -129,7 +129,7 @@ async def test_chat_completion_request_with_redaction(route, body): 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) _test_logger = testLogger() diff --git a/tests/proxy_unit_tests/test_user_api_key_auth.py b/tests/proxy_unit_tests/test_user_api_key_auth.py index 911a7b06e5..c6345f74d0 100644 --- a/tests/proxy_unit_tests/test_user_api_key_auth.py +++ b/tests/proxy_unit_tests/test_user_api_key_auth.py @@ -287,7 +287,7 @@ async def test_auth_with_allowed_routes(route, should_raise_error): general_settings = {"allowed_routes": ["/embeddings"]} from fastapi import Request - from litellm.proxy import proxy_server + from litellm_proxy import proxy_server initial_general_settings = getattr(proxy_server, "general_settings")