diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 3c62339dd..d8750ddb7 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -301,16 +301,14 @@ End""" @patch("aider.models.litellm.completion") def test_simple_send_with_retries_removes_reasoning(self, mock_completion): model = Model("deepseek-r1") # This model has remove_reasoning="think" - + # Mock the completion response mock_response = MagicMock() - mock_response.choices = [ - MagicMock(message=MagicMock(content="""Here is some text + mock_response.choices = [MagicMock(message=MagicMock(content="""Here is some text This reasoning should be removed -And this text should remain""")) - ] +And this text should remain"""))] mock_completion.return_value = mock_response messages = [{"role": "user", "content": "test"}]