mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-28 16:25:00 +00:00
fix: Add missing attributes to MockCompletion for test compatibility
This commit is contained in:
parent
4a6c4b95f1
commit
8545672839
1 changed files with 4 additions and 0 deletions
|
@ -26,6 +26,10 @@ class TestReasoning(unittest.TestCase):
|
|||
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.finish_reason = "stop"
|
||||
|
||||
mock_completion = MockCompletion(main_content, reasoning_content)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue