From c472e6e1603da5085f2b2a30df4d004716d4de6c Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 8 Nov 2024 10:00:30 -0800 Subject: [PATCH] fix: Update test_exceptions.py to provide required arguments for litellm exceptions --- tests/basic/test_exceptions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/basic/test_exceptions.py b/tests/basic/test_exceptions.py index f5bcae850..905726093 100644 --- a/tests/basic/test_exceptions.py +++ b/tests/basic/test_exceptions.py @@ -21,7 +21,7 @@ def test_get_ex_info(): # Test with a known exception type from litellm import AuthenticationError - auth_error = AuthenticationError() + auth_error = AuthenticationError(message="Invalid API key", llm_provider="openai", model="gpt-4") ex_info = ex.get_ex_info(auth_error) assert isinstance(ex_info, ExInfo) assert ex_info.name == "AuthenticationError" @@ -45,7 +45,7 @@ def test_rate_limit_error(): ex = LiteLLMExceptions() from litellm import RateLimitError - rate_error = RateLimitError() + rate_error = RateLimitError(message="Rate limit exceeded", llm_provider="openai", model="gpt-4") ex_info = ex.get_ex_info(rate_error) assert ex_info.retry is True assert "rate limited" in ex_info.description.lower() @@ -56,6 +56,6 @@ def test_context_window_error(): ex = LiteLLMExceptions() from litellm import ContextWindowExceededError - ctx_error = ContextWindowExceededError() + ctx_error = ContextWindowExceededError(message="Context length exceeded", model="gpt-4", llm_provider="openai") ex_info = ex.get_ex_info(ctx_error) assert ex_info.retry is False