fix _assemble_key_object

tests/local_testing/test_add_update_models.py

@@ -14,7 +14,9 @@ import os, io, time
 sys.path.insert(
     0, os.path.abspath("../..")
 )  # Adds the parent directory to the system path
+
 import pytest, logging, asyncio
+import litellm_proxy
 import litellm
 from litellm_proxy.management_endpoints.model_management_endpoints import (
     add_new_model,

tests/local_testing/test_blocked_user_list.py

@@ -25,6 +25,7 @@ import logging
 import pytest
 
 import litellm
+import litellm_proxy
 from litellm import Router, mock_completion
 from litellm._logging import verbose_proxy_logger
 from litellm.caching.caching import DualCache

tests/local_testing/test_lakera_ai_prompt_injection.py

@@ -24,6 +24,7 @@ from unittest.mock import patch
 import pytest
 
 import litellm
+import litellm_proxy
 from litellm._logging import verbose_proxy_logger
 from litellm.caching.caching import DualCache
 from litellm_proxy._types import UserAPIKeyAuth

tests/local_testing/test_openai_moderations_hook.py

@@ -23,6 +23,7 @@ from litellm import Router, mock_completion
 from litellm_proxy.utils import ProxyLogging, hash_token
 from litellm_proxy._types import UserAPIKeyAuth
 from litellm.caching.caching import DualCache
+import litellm_proxy
 
 ### UNIT TESTS FOR OpenAI Moderation ###
 

tests/local_testing/test_spend_calculate_endpoint.py

@@ -7,6 +7,7 @@ from fastapi import Request
 from fastapi.routing import APIRoute
 
 import litellm
+import litellm_proxy
 from litellm_proxy._types import SpendCalculateRequest
 from litellm_proxy.spend_tracking.spend_management_endpoints import calculate_spend
 from litellm.router import Router

tests/local_testing/test_update_spend.py

@@ -25,6 +25,7 @@ import logging
 import pytest
 
 import litellm
+import litellm_proxy
 from litellm import Router, mock_completion
 from litellm._logging import verbose_proxy_logger
 from litellm.caching.caching import DualCache

tests/proxy_admin_ui_tests/test_key_management.py

@@ -25,6 +25,7 @@ import logging
 
 import pytest
 import litellm
+import litellm_proxy
 from litellm._logging import verbose_proxy_logger
 from litellm_proxy.management_endpoints.team_endpoints import list_team
 from litellm_proxy._types import *

tests/proxy_admin_ui_tests/test_role_based_access.py

@@ -28,6 +28,7 @@ import logging
 import pytest
 
 import litellm
+import litellm_proxy
 from litellm._logging import verbose_proxy_logger
 from litellm_proxy.auth.auth_checks import get_user_object
 from litellm_proxy.management_endpoints.key_management_endpoints import (

tests/proxy_admin_ui_tests/test_sso_sign_in.py

@@ -10,6 +10,7 @@ sys.path.insert(
     0, os.path.abspath("../..")
 )  # Adds the parent directory to the system path
 import litellm
+import litellm_proxy
 from litellm_proxy.proxy_server import app
 from litellm_proxy.utils import PrismaClient, ProxyLogging
 from litellm_proxy.management_endpoints.ui_sso import auth_callback

tests/proxy_admin_ui_tests/test_usage_endpoints.py

@@ -39,6 +39,7 @@ import logging
 import pytest
 
 import litellm
+import litellm_proxy
 from litellm._logging import verbose_proxy_logger
 from litellm_proxy.management_endpoints.internal_user_endpoints import (
     new_user,

tests/proxy_unit_tests/test_aproxy_startup.py

@@ -14,6 +14,7 @@ sys.path.insert(
 )  # Adds the parent directory to the system path
 import pytest, logging, asyncio
 import litellm
+import litellm_proxy
 from litellm_proxy.proxy_server import (
     router,
     save_worker_config,

tests/proxy_unit_tests/test_audit_logs_proxy.py

@@ -26,6 +26,7 @@ load_dotenv()
 import pytest
 import uuid
 import litellm
+import litellm_proxy
 from litellm._logging import verbose_proxy_logger
 
 from litellm_proxy.proxy_server import (

tests/proxy_unit_tests/test_jwt.py

@@ -25,6 +25,7 @@ from fastapi import Request, HTTPException
 from fastapi.routing import APIRoute
 from fastapi.responses import Response
 import litellm
+import litellm_proxy
 from litellm.caching.caching import DualCache
 from litellm_proxy._types import (
     LiteLLM_JWTAuth,

tests/proxy_unit_tests/test_proxy_token_counter.py

@@ -18,6 +18,7 @@ sys.path.insert(
 )  # Adds the parent directory to the system path
 import pytest, logging, asyncio
 import litellm, asyncio
+import litellm_proxy
 from litellm_proxy.proxy_server import token_counter
 from litellm_proxy.utils import PrismaClient, ProxyLogging, hash_token, update_spend
 from litellm._logging import verbose_proxy_logger

tests/proxy_unit_tests/test_proxy_utils.py

@@ -12,6 +12,7 @@ sys.path.insert(
     0, os.path.abspath("../..")
 )  # Adds the parent directory to the system path
 import litellm
+import litellm_proxy
 from unittest.mock import MagicMock, patch, AsyncMock
 
 from litellm_proxy._types import LitellmUserRoles, UserAPIKeyAuth

tests/proxy_unit_tests/test_user_api_key_auth.py

@@ -16,6 +16,7 @@ from starlette.datastructures import URL
 from litellm._logging import verbose_proxy_logger
 import logging
 import litellm
+import litellm_proxy
 from litellm_proxy.auth.user_api_key_auth import (
     user_api_key_auth,
     UserAPIKeyAuth,