diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index c2e283f32..10a8ad8ce 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -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)