dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-04 03:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: Add reasoning content handling in Coder and Model classes

Browse source
This commit is contained in:
Paul Gauthier 2025-03-06 18:16:44 -08:00 • committed by Paul Gauthier (aider)
parent 52162a5604
commit d1d40a9a76
2 changed files with 34 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

27
aider/coders/base_coder.py
View file

@ -36,6 +36,8 @@ from aider.utils import format_content, format_messages, format_tokens, is_image
from ..dump import dump # noqa: F401
from .chat_chunks import ChatChunks
REASONING_TAG = "reasoning-content-" + "7bbeb8e1441453ad999a0bbba8a46d4b"
class UnknownEditFormat(ValueError):
def __init__(self, edit_format, valid_formats):
@ -1306,6 +1308,8 @@ class Coder:
litellm_ex = LiteLLMExceptions()
self.got_reasoning_content = False
self.ended_reasoning_content = False
self.usage_report = None
exhausted = False
interrupted = False
@ -1372,8 +1376,15 @@ class Coder:
self.mdstream = None
self.partial_response_content = self.get_multi_response_content_in_progress(True)
if self.got_reasoning_content:
reasoning_tag = REASONING_TAG
else:
reasoning_tag = None
self.partial_response_content = self.main_model.remove_reasoning_content(
self.partial_response_content
self.partial_response_content,
reasoning_tag=reasoning_tag,
)
self.multi_response_content = ""
@ -1744,9 +1755,23 @@ class Coder:
except AttributeError:
pass
try:
text = chunk.choices[0].delta.reasoning_content
if text:
self.got_reasoning_content = True
self.partial_response_content += text
received_content = True
except AttributeError:
text = None
try:
text = chunk.choices[0].delta.content
if text:
if self.got_reasoning_content and not self.ended_reasoning_content:
tag = f"\n\n------\n\n</{REASONING_TAG}>\n\n"
self.partial_response_content += tag
self.ended_reasoning_content = True
self.partial_response_content += text
received_content = True
except AttributeError: