From 9553478384de7eb2c8c933cdbda746d9933fa821 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 4 Feb 2025 12:06:59 -0800 Subject: [PATCH] test: Fix patch target for simple_send_with_retries in tests --- tests/basic/test_history.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_history.py b/tests/basic/test_history.py index e2ea194c0..b77869a8c 100644 --- a/tests/basic/test_history.py +++ b/tests/basic/test_history.py @@ -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.simple_send_with_retries") + @mock.patch("aider.models.Model.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.simple_send_with_retries") + @mock.patch("aider.models.Model.simple_send_with_retries") def test_fallback_to_second_model(self, mock_send): mock_model1 = mock.Mock(spec=Model) mock_model1.name = "gpt-4"