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

(feat) add BadRequestError for Azure

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 09:53:38 +05:30
parent 8451badaa1
commit 191c1d2887
1 changed files with 57 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

57
litellm/tests/test_exceptions.py
View file

@ -233,6 +233,63 @@ async def asynctest_completion_azure_exception():
# )
def asynctest_completion_openai_exception_bad_model():
try:
import openai
import litellm, asyncio
print("azure exception bad model\n\n")
litellm.set_verbose=True
## Test azure call
async def test():
response = await litellm.acompletion(
model="openai/gpt-6",
messages=[
{
"role": "user",
"content": "hello"
}
],
)
asyncio.run(test())
except openai.BadRequestError:
print("Good job this is a bad request error for a model that does not exist!")
print("Passed")
except Exception as e:
print("Raised wrong type of exception", type(e))
assert isinstance(e, openai.BadRequestError)
pytest.fail(f"Error occurred: {e}")
# asynctest_completion_openai_exception_bad_model()
def asynctest_completion_azure_exception_bad_model():
try:
import openai
import litellm, asyncio
print("azure exception bad model\n\n")
litellm.set_verbose=True
## Test azure call
async def test():
response = await litellm.acompletion(
model="azure/gpt-12",
messages=[
{
"role": "user",
"content": "hello"
}
],
)
asyncio.run(test())
except openai.BadRequestError:
print("Good job this is a bad request error for a model that does not exist!")
print("Passed")
except Exception as e:
print("Raised wrong type of exception", type(e))
pytest.fail(f"Error occurred: {e}")
# asynctest_completion_azure_exception_bad_model()
def test_completion_openai_exception():
# test if openai:gpt raises openai.AuthenticationError
try: