mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54: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:
|
if not self.remove_reasoning:
|
||||||
return res
|
return res
|
||||||
|
|
||||||
|
# Try to match the complete tag pattern first
|
||||||
pattern = f"<{self.remove_reasoning}>.*?</{self.remove_reasoning}>"
|
pattern = f"<{self.remove_reasoning}>.*?</{self.remove_reasoning}>"
|
||||||
res = re.sub(pattern, "", res, flags=re.DOTALL).strip()
|
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
|
return res
|
||||||
|
|
||||||
def simple_send_with_retries(self, messages):
|
def simple_send_with_retries(self, messages):
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue