From 94aa6ec543aa7c30da9ce7c66913c1e82e1c5ce5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 15 Jun 2024 10:51:44 -0700 Subject: [PATCH] test fix _get_bearer_token --- litellm/tests/test_key_generate_prisma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 27a986b9c..f53e241ce 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -1779,7 +1779,7 @@ async def test_upperbound_key_params(prisma_client): def test_get_bearer_token(): - from litellm.proxy.proxy_server import _get_bearer_token + from litellm.proxy.auth.user_api_key_auth import _get_bearer_token # Test valid Bearer token api_key = "Bearer valid_token"