(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 response=response
) )
print("Type of exception=", type(openai_exception)) print("Type of exception=", type(openai_exception))
assert isinstance(openai_exception, openai.BadRequestError) assert isinstance(openai_exception, openai.NotFoundError)
except Exception as e: except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}") pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")