diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 355ddeef1..972627763 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -277,7 +277,7 @@ class TestModels(unittest.TestCase): self.assertTrue(model.use_repo_map) self.assertTrue(model.examples_as_sys_msg) self.assertFalse(model.use_temperature) - self.assertEqual(model.remove_reasoning, "think") + self.assertEqual(model.reasoning_tag, "think") # Test provider/deepseek-r1 case model = Model("someprovider/deepseek-r1") diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index 161da61b6..9da646831 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -186,7 +186,7 @@ class TestReasoning(unittest.TestCase): # Setup model and coder model = Model("gpt-3.5-turbo") - model.remove_reasoning = "think" # Set to remove tags + model.reasoning_tag = "think" # Set to remove tags coder = Coder.create(model, None, io=io, stream=False) # Test data @@ -256,7 +256,7 @@ class TestReasoning(unittest.TestCase): # Setup model and coder model = Model("gpt-3.5-turbo") - model.remove_reasoning = "think" # Set to remove tags + model.reasoning_tag = "think" # Set to remove tags coder = Coder.create(model, None, io=io, stream=True) # Ensure the coder shows pretty output @@ -384,7 +384,7 @@ End""" @patch("aider.models.litellm.completion") def test_simple_send_with_retries_removes_reasoning(self, mock_completion): """Test that simple_send_with_retries correctly removes reasoning content.""" - model = Model("deepseek-r1") # This model has remove_reasoning="think" + model = Model("deepseek-r1") # This model has reasoning_tag="think" # Mock the completion response mock_response = MagicMock()