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

Revert "rename simple_send_with_retries -> send_with_retries"

Browse source 
This reverts commit d619edf6e9.
This commit is contained in:
Paul Gauthier 2024-08-02 10:49:44 -03:00
parent d619edf6e9
commit da3e507ec4
5 changed files with 26 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/basic/test_history.py
View file

@ -34,7 +34,7 @@ class TestChatSummary(TestCase):
tokenized = self.chat_summary.tokenize(messages)
self.assertEqual(tokenized, [(2, messages[0]), (2, messages[1])])
@mock.patch("aider.history.send_with_retries")
@mock.patch("aider.history.simple_send_with_retries")
def test_summarize_all(self, mock_send):
mock_send.return_value = "This is a summary"
messages = [
@ -69,7 +69,7 @@ class TestChatSummary(TestCase):
self.assertGreater(len(result), 0)
self.assertLessEqual(len(result), len(messages))
@mock.patch("aider.history.send_with_retries")
@mock.patch("aider.history.simple_send_with_retries")
def test_fallback_to_second_model(self, mock_send):
mock_model1 = mock.Mock(spec=Model)
mock_model1.name = "gpt-4"

12
tests/basic/test_repo.py
View file

@ -106,7 +106,7 @@ class TestRepo(unittest.TestCase):
diffs = git_repo.diff_commits(False, "HEAD~1", "HEAD")
self.assertIn("two", diffs)
@patch("aider.repo.send_with_retries")
@patch("aider.repo.simple_send_with_retries")
def test_get_commit_message(self, mock_send):
mock_send.side_effect = ["", "a good commit message"]
@ -120,14 +120,14 @@ class TestRepo(unittest.TestCase):
# Assert that the returned message is the expected one from the second model
self.assertEqual(result, "a good commit message")
# Check that send_with_retries was called twice
# Check that simple_send_with_retries was called twice
self.assertEqual(mock_send.call_count, 2)
# Check that it was called with the correct model names
mock_send.assert_any_call(model1.name, mock_send.call_args[0][1])
mock_send.assert_any_call(model2.name, mock_send.call_args[0][1])
@patch("aider.repo.send_with_retries")
@patch("aider.repo.simple_send_with_retries")
def test_get_commit_message_strip_quotes(self, mock_send):
mock_send.return_value = '"a good commit message"'
@ -138,7 +138,7 @@ class TestRepo(unittest.TestCase):
# Assert that the returned message is the expected one
self.assertEqual(result, "a good commit message")
@patch("aider.repo.send_with_retries")
@patch("aider.repo.simple_send_with_retries")
def test_get_commit_message_no_strip_unmatched_quotes(self, mock_send):
mock_send.return_value = 'a good "commit message"'
@ -149,7 +149,7 @@ class TestRepo(unittest.TestCase):
# Assert that the returned message is the expected one
self.assertEqual(result, 'a good "commit message"')
@patch("aider.repo.send_with_retries")
@patch("aider.repo.simple_send_with_retries")
def test_get_commit_message_with_custom_prompt(self, mock_send):
mock_send.return_value = "Custom commit message"
custom_prompt = "Generate a commit message in the style of Shakespeare"
@ -349,7 +349,7 @@ class TestRepo(unittest.TestCase):
fnames = git_repo.get_tracked_files()
self.assertIn(str(fname), fnames)
@patch("aider.repo.send_with_retries")
@patch("aider.repo.simple_send_with_retries")
def test_noop_commit(self, mock_send):
mock_send.return_value = '"a good commit message"'