From 05aa99fde301858ba7be112d1a788b36fecd455a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 12 Aug 2023 16:49:55 -0700 Subject: [PATCH] fixing exception handling test --- litellm/tests/test_exceptions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py index fa62b1970c..16f7ae65d4 100644 --- a/litellm/tests/test_exceptions.py +++ b/litellm/tests/test_exceptions.py @@ -37,7 +37,7 @@ def test_context_window(model): try: azure = model == "chatgpt-test" print(f"model: {model}") - response = completion(model=model, messages=messages, custom_llm_provider=custom_llm_provider, logger_fn=logging_fn) + response = completion(model=model, messages=messages, custom_llm_provider="azure", logger_fn=logging_fn) print(f"response: {response}") except InvalidRequestError: print("InvalidRequestError")