diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 17cca5cdd..56db85ed4 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1315,8 +1315,6 @@ class Coder: litellm_ex = LiteLLMExceptions() - self.got_reasoning_content = False - self.ended_reasoning_content = False self.usage_report = None exhausted = False interrupted = False @@ -1644,6 +1642,9 @@ class Coder: return prompts.added_files.format(fnames=", ".join(added_fnames)) def send(self, messages, model=None, functions=None): + self.got_reasoning_content = False + self.ended_reasoning_content = False + if not model: model = self.main_model diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index a4f0f11ed..23e65481d 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -131,6 +131,8 @@ class TestReasoning(unittest.TestCase): # Verify mdstream.update was called multiple times mock_mdstream.update.assert_called() + coder.live_incremental_response(True) + # Explicitly get all calls to update update_calls = mock_mdstream.update.call_args_list