mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 14:55:00 +00:00
refactor: Move reasoning_content to choices[0].message in MockCompletion
This commit is contained in:
parent
8545672839
commit
0cd8e3701d
1 changed files with 1 additions and 1 deletions
|
@ -25,10 +25,10 @@ class TestReasoning(unittest.TestCase):
|
|||
class MockCompletion:
|
||||
def __init__(self, content, reasoning_content):
|
||||
self.content = content
|
||||
self.reasoning_content = reasoning_content
|
||||
# Add required attributes expected by show_send_output
|
||||
self.choices = [MagicMock()]
|
||||
self.choices[0].message.content = content
|
||||
self.choices[0].message.reasoning_content = reasoning_content
|
||||
self.finish_reason = "stop"
|
||||
|
||||
mock_completion = MockCompletion(main_content, reasoning_content)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue