scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

Merge pull request #2540 from BerriAI/litellm_fix_error_message_on_bad_key

Browse source 
(fix) Proxy - fix error message raised on passing invalid tokens
This commit is contained in:
Ishaan Jaff 2024-03-15 15:49:39 -07:00 • committed by GitHub
commit eb071d8101
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -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:

4
litellm/tests/test_key_generate_prisma.py
View file

@ -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