From 456ce692df013c72101d5dfda29bbe7dbe0ff3d1 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 12 Feb 2024 16:25:35 -0800 Subject: [PATCH] test(test_proxy_exception_mapping.py): fix test --- litellm/tests/test_proxy_exception_mapping.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_proxy_exception_mapping.py b/litellm/tests/test_proxy_exception_mapping.py index 31829586d8..a2c7952586 100644 --- a/litellm/tests/test_proxy_exception_mapping.py +++ b/litellm/tests/test_proxy_exception_mapping.py @@ -139,7 +139,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)}")