From 3aa0f6ed7296892d4dee75d404c183eba021426e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 11 Oct 2025 08:54:15 -0700 Subject: [PATCH] test_generate_key_with_object_permission --- .../management_endpoints/test_key_management_endpoints.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py index fcf2ac2453b..74544cb882a 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py @@ -1874,15 +1874,15 @@ async def test_generate_key_with_object_permission(): # Patch the prisma_client and other dependencies with patch( - "litellm.proxy.management_endpoints.key_management_endpoints.prisma_client", + "litellm.proxy.proxy_server.prisma_client", mock_prisma_client, ), patch( - "litellm.proxy.management_endpoints.key_management_endpoints.llm_router", None + "litellm.proxy.proxy_server.llm_router", None ), patch( - "litellm.proxy.management_endpoints.key_management_endpoints.premium_user", + "litellm.proxy.proxy_server.premium_user", False, ), patch( - "litellm.proxy.management_endpoints.key_management_endpoints.litellm_proxy_admin_name", + "litellm.proxy.proxy_server.litellm_proxy_admin_name", "admin", ): # Execute