diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 48e3689dd..0d1f488e1 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -941,6 +941,8 @@ def test_generate_and_call_with_expired_key(prisma_client): print("Got Exception", e) print(e.message) assert "Authentication Error" in e.message + assert e.type == ProxyErrorTypes.expired_key + pass