mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 14:25:00 +00:00
refactor: rename remove_reasoning to reasoning_tag in test files
This commit is contained in:
parent
cac9b4460e
commit
072ce87051
2 changed files with 4 additions and 4 deletions
|
@ -186,7 +186,7 @@ class TestReasoning(unittest.TestCase):
|
|||
|
||||
# Setup model and coder
|
||||
model = Model("gpt-3.5-turbo")
|
||||
model.remove_reasoning = "think" # Set to remove <think> tags
|
||||
model.reasoning_tag = "think" # Set to remove <think> tags
|
||||
coder = Coder.create(model, None, io=io, stream=False)
|
||||
|
||||
# Test data
|
||||
|
@ -256,7 +256,7 @@ class TestReasoning(unittest.TestCase):
|
|||
|
||||
# Setup model and coder
|
||||
model = Model("gpt-3.5-turbo")
|
||||
model.remove_reasoning = "think" # Set to remove <think> tags
|
||||
model.reasoning_tag = "think" # Set to remove <think> tags
|
||||
coder = Coder.create(model, None, io=io, stream=True)
|
||||
|
||||
# Ensure the coder shows pretty output
|
||||
|
@ -384,7 +384,7 @@ End"""
|
|||
@patch("aider.models.litellm.completion")
|
||||
def test_simple_send_with_retries_removes_reasoning(self, mock_completion):
|
||||
"""Test that simple_send_with_retries correctly removes reasoning content."""
|
||||
model = Model("deepseek-r1") # This model has remove_reasoning="think"
|
||||
model = Model("deepseek-r1") # This model has reasoning_tag="think"
|
||||
|
||||
# Mock the completion response
|
||||
mock_response = MagicMock()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue