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

test(test_proxy_exception_mapping.py): fix exception checking

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-04 22:45:16 +05:30
parent 8283c4f7c7
commit 6506fba3bc
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_proxy_exception_mapping.py
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.NotFoundError)
assert isinstance(openai_exception, openai.BadRequestError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
@ -149,7 +149,7 @@ def test_chat_completion_exception_any_model(client):
response=response
)
print("Exception raised=", openai_exception)
assert isinstance(openai_exception, openai.NotFoundError)
assert isinstance(openai_exception, openai.BadRequestError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
@ -170,7 +170,7 @@ def test_embedding_exception_any_model(client):
response=response
)
print("Exception raised=", openai_exception)
assert isinstance(openai_exception, openai.NotFoundError)
assert isinstance(openai_exception, openai.BadRequestError)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")