From 80de3335b7b4639d3bde77702d1db2ccc400d2fb Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 7 Mar 2025 17:15:44 -0800 Subject: [PATCH] refactor: Move reasoning content flags initialization to send method --- aider/coders/base_coder.py | 5 +++-- tests/basic/test_reasoning.py | 2 ++ 2 files changed, 5 insertions(+), 2 deletions(-) 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