mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-28 08:14:59 +00:00
refactor: Enhance remove_reasoning_content to handle incomplete tag patterns
This commit is contained in:
parent
2bb4db127c
commit
c0c960ec2e
1 changed files with 8 additions and 0 deletions
|
@ -613,8 +613,16 @@ class Model(ModelSettings):
|
|||
if not self.remove_reasoning:
|
||||
return res
|
||||
|
||||
# Try to match the complete tag pattern first
|
||||
pattern = f"<{self.remove_reasoning}>.*?</{self.remove_reasoning}>"
|
||||
res = re.sub(pattern, "", res, flags=re.DOTALL).strip()
|
||||
|
||||
# If closing tag exists but opening tag might be missing, remove everything before closing tag
|
||||
closing_tag = f"</{self.remove_reasoning}>"
|
||||
if closing_tag in res:
|
||||
parts = res.split(closing_tag, 1)
|
||||
res = parts[1].strip() if len(parts) > 1 else res
|
||||
|
||||
return res
|
||||
|
||||
def simple_send_with_retries(self, messages):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue