test string checked for model access control

This commit is contained in:
Ishaan Jaff 2025-03-10 20:04:18 -07:00
parent aa5ac6ba3d
commit 91a7fb0a23
2 changed files with 2 additions and 2 deletions

View file

@ -94,7 +94,7 @@ async def test_model_access_patterns(key_models, test_model, expect_success):
assert _error_body["type"] == "key_model_access_denied" assert _error_body["type"] == "key_model_access_denied"
assert _error_body["param"] == "model" assert _error_body["param"] == "model"
assert _error_body["code"] == "401" assert _error_body["code"] == "401"
assert "API Key not allowed to access model" in _error_body["message"] assert "key not allowed to access model" in _error_body["message"]
@pytest.mark.asyncio @pytest.mark.asyncio

View file

@ -308,7 +308,7 @@ async def test_chat_completion():
model="gpt-4", model="gpt-4",
messages=[{"role": "user", "content": "Hello!"}], messages=[{"role": "user", "content": "Hello!"}],
) )
assert "API Key not allowed to access model." in str(e) assert "key not allowed to access model." in str(e)
@pytest.mark.asyncio @pytest.mark.asyncio