dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 20:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

style: Format code with linter

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-08 10:00:33 -08:00
parent c472e6e160
commit 203634314c
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_exceptions.py
View file

@ -21,7 +21,9 @@ def test_get_ex_info():
# Test with a known exception type
from litellm import AuthenticationError
auth_error = AuthenticationError(message="Invalid API key", llm_provider="openai", model="gpt-4")
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"
@ -56,6 +58,8 @@ def test_context_window_error():
ex = LiteLLMExceptions()
from litellm import ContextWindowExceededError
ctx_error = ContextWindowExceededError(message="Context length exceeded", model="gpt-4", llm_provider="openai")
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