refactor: simplify litellm exception imports

This commit is contained in:
Paul Gauthier 2024-10-28 14:27:19 -07:00 committed by Paul Gauthier (aider)
parent e2dff0a74b
commit bf63e7045b
2 changed files with 15 additions and 16 deletions

View file

@ -10,7 +10,7 @@ from aider.sendchat import simple_send_with_retries
class PrintCalled(Exception):
pass
#ai add a test that simply invokes retry_exceptions() and ensures no error raised!
class TestSendChat(unittest.TestCase):
@patch("litellm.completion")
@patch("builtins.print")