From fd93a68b0d21c84ea5f5ca99f91d95847583137d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 15 Mar 2024 14:50:34 -0700 Subject: [PATCH] (fix) fix error message raised on passing invalid tokens --- litellm/proxy/proxy_server.py | 3 +++ litellm/tests/test_key_generate_prisma.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) 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