fix import loc

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()

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")
 

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,

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()

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")