From 83b31141c6ecae29991e42ab1a54a715743287d3 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 5 Jan 2024 12:29:03 +0530 Subject: [PATCH] (ci/cd) raise correct exception proxy --- 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 59f59e7ac..fcc0ad98c 100644 --- a/litellm/tests/test_proxy_exception_mapping.py +++ b/litellm/tests/test_proxy_exception_mapping.py @@ -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.BadRequestError) + assert isinstance(openai_exception, openai.NotFoundError) except Exception as e: pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")