From 854567283948ed8f3abe6dc5edfb3ecc0a459363 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 7 Mar 2025 16:54:24 -0800 Subject: [PATCH] fix: Add missing attributes to MockCompletion for test compatibility --- tests/basic/test_reasoning.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index efba7a8df..c2e283f32 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -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)