From cddc67ad69547851a1ea03293aae713988258a03 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 7 Mar 2025 17:30:15 -0800 Subject: [PATCH] refactor: Extract reasoning content removal to helper method --- aider/coders/base_coder.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index aca39f656..719c5af69 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -1382,12 +1382,10 @@ class Coder: self.partial_response_content = self.get_multi_response_content_in_progress(True) - # ai: make a helper method... - self.partial_response_content = self.main_model.remove_reasoning_content( + self.partial_response_content = self.remove_reasoning_content( self.partial_response_content, self.reasoning_tag_name, ) - # ... ai! self.multi_response_content = "" ### @@ -1835,6 +1833,10 @@ class Coder: def render_incremental_response(self, final): return self.get_multi_response_content_in_progress() + def remove_reasoning_content(self, content, reasoning_tag_name): + """Remove reasoning content from the model's response.""" + return self.main_model.remove_reasoning_content(content, reasoning_tag_name) + def calculate_and_show_tokens_and_cost(self, messages, completion=None): prompt_tokens = 0 completion_tokens = 0