From a1bb880872d2ed354fe54d35bc38b8fbb04dde22 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 23 Nov 2023 16:28:05 -0800 Subject: [PATCH] (test) router raise exceptions --- litellm/tests/test_router.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index ca8e8c7f30..5c78c929c0 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -157,13 +157,11 @@ def test_exception_raising(): ] ) except openai.AuthenticationError: - print("Caught an OPENAI AUTH Error, Good job. This is what we needed!") - except Exception as e: + print("Test Passed: Caught an OPENAI AUTH Error, Good job. This is what we needed!") os.environ["AZURE_API_KEY"] = old_api_key - print("Got exception on router! Good job ", e) - print(e) - -# test_exception_raising() + except Exception as e: + print("Got unexpected exception on router!", e) +test_exception_raising() ### FUNCTION CALLING