diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index e3e778dc7c..52dac7524c 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -889,6 +889,9 @@ async def user_api_key_auth( raise Exception( f"This key is made for LiteLLM UI, Tried to access route: {route}. Not allowed" ) + if valid_token is None: + # No token was found when looking up in the DB + raise Exception("Invalid token passed") if valid_token_dict is not None: return UserAPIKeyAuth(api_key=api_key, **valid_token_dict) else: diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 151781beb2..103b344f50 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -158,7 +158,7 @@ def test_call_with_invalid_key(prisma_client): async def test(): await litellm.proxy.proxy_server.prisma_client.connect() - generated_key = "bad-key" + generated_key = "sk-126666" bearer_token = "Bearer " + generated_key request = Request(scope={"type": "http"}, receive=None) @@ -173,7 +173,7 @@ def test_call_with_invalid_key(prisma_client): except Exception as e: print("Got Exception", e) print(e.message) - assert "Authentication Error" in e.message + assert "Authentication Error, Invalid token passed" in e.message pass