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

fix: Handle NotFoundError in send chat test with single print call

Browse source
This commit is contained in:
Paul Gauthier 2024-11-08 14:18:23 -08:00 • committed by Paul Gauthier (aider)
parent 7a8399571a
commit 9dd7b795ca
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/basic/test_sendchat.py
View file

@ -84,10 +84,10 @@ class TestSendChat(unittest.TestCase):
mock = MagicMock()
mock.status_code = 400
mock_completion.side_effect = litellm.InvalidRequestError(
mock_completion.side_effect = litellm.NotFoundError(
message="Invalid request", llm_provider="test_provider", model="test_model"
)
result = simple_send_with_retries(self.mock_model, self.mock_messages)
assert result is None
assert mock_print.call_count == 2 # Error message and description
assert mock_print.call_count == 1