diff --git a/tests/test_repo.py b/tests/test_repo.py index 41a8a791d..efa18bf99 100644 --- a/tests/test_repo.py +++ b/tests/test_repo.py @@ -15,10 +15,13 @@ from tests.utils import GitTemporaryDirectory class TestRepo(unittest.TestCase): - @patch("aider.sendchat.send_with_retries") + @patch("aider.repo.send_with_retries") def test_get_commit_message(self, mock_send): # Set the return value of the mocked function - mock_send.return_value = "a good commit message" + mock_send.return_value = ( + None, + "a good commit message" + ) repo = AiderRepo(InputOutput(), None) # Call the get_commit_message method with dummy diff and context