diff --git a/litellm/tests/test_key_generate_dynamodb.py b/litellm/tests/test_key_generate_dynamodb.py index 96da4e9df2..559cc9ff52 100644 --- a/litellm/tests/test_key_generate_dynamodb.py +++ b/litellm/tests/test_key_generate_dynamodb.py @@ -190,8 +190,9 @@ def test_call_with_key_under_budget(): asyncio.run(test()) except Exception as e: - pass - print("Got Exception", e) + error_detail = e.detail + assert "Authentication Error, ExceededBudget:" in error_detail + print(vars(e)) # def test_call_with_key_over_budget():