(ci/cd) raise correct exception proxy

This commit is contained in:
ishaan-jaff 2024-01-05 12:29:03 +05:30
parent bcf22725a6
commit 83b31141c6

View file

@ -123,7 +123,7 @@ def test_exception_openai_bad_model(client):
response=response
)
print("Type of exception=", type(openai_exception))
assert isinstance(openai_exception, openai.BadRequestError)
assert isinstance(openai_exception, openai.NotFoundError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")