phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) proxy - exception mapping openai.NotFoundError

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 13:34:51 +05:30
parent 13eafa52e9
commit d797bb100f
1 changed files with 56 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

57
litellm/tests/test_proxy_exception_mapping.py
View file

@ -69,4 +69,59 @@ def test_chat_completion_exception_azure(client):
assert isinstance(openai_exception, openai.AuthenticationError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
# raise openai.BadRequestError
def test_exception_openai_bad_model(client):
try:
# Your test data
test_data = {
"model": "openai/GPT-12",
"messages": [
{
"role": "user",
"content": "hi"
},
],
"max_tokens": 10,
}
response = client.post("/chat/completions", json=test_data)
# make an openai client to call _make_status_error_from_response
openai_client = openai.OpenAI(api_key="anything")
openai_exception = openai_client._make_status_error_from_response(response=response)
print("Type of exception=", type(openai_exception))
assert isinstance(openai_exception, openai.NotFoundError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
def test_chat_completion_exception_any_model(client):
try:
# Your test data
test_data = {
"model": "Lite-GPT-12",
"messages": [
{
"role": "user",
"content": "hi"
},
],
"max_tokens": 10,
}
response = client.post("/chat/completions", json=test_data)
# make an openai client to call _make_status_error_from_response
openai_client = openai.OpenAI(api_key="anything")
openai_exception = openai_client._make_status_error_from_response(response=response)
print("Exception raised=", openai_exception)
assert isinstance(openai_exception, openai.NotFoundError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")