fix test_team_disable_guardrails

This commit is contained in:
Ishaan Jaff 2024-07-31 11:48:36 -07:00
parent 7db85fbdb7
commit bcc4adff46

View file

@ -227,7 +227,7 @@ async def test_team_disable_guardrails(mock_acompletion, client_no_auth):
await user_api_key_auth(request=request, api_key="Bearer " + user_key)
pytest.fail("Expected to raise 403 forbidden error.")
except ProxyException as e:
assert e.code == 403
assert e.code == "403"
from litellm.tests.test_custom_callback_input import CompletionCustomHandler