diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index b9b5363bd..b9db9c6c4 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -90,7 +90,9 @@ class TestReasoning(unittest.TestCase): # Mock streaming response chunks class MockStreamingChunk: - def __init__(self, content=None, reasoning_content=None, reasoning=None, finish_reason=None): + def __init__( + self, content=None, reasoning_content=None, reasoning=None, finish_reason=None + ): self.choices = [MagicMock()] self.choices[0].delta = MagicMock() self.choices[0].finish_reason = finish_reason @@ -108,7 +110,7 @@ class TestReasoning(unittest.TestCase): else: # Need to handle attribute access that would raise AttributeError delattr(self.choices[0].delta, "reasoning_content") - + # Set reasoning if provided if reasoning is not None: self.choices[0].delta.reasoning = reasoning @@ -271,7 +273,9 @@ class TestReasoning(unittest.TestCase): # Mock streaming response chunks class MockStreamingChunk: - def __init__(self, content=None, reasoning_content=None, reasoning=None, finish_reason=None): + def __init__( + self, content=None, reasoning_content=None, reasoning=None, finish_reason=None + ): self.choices = [MagicMock()] self.choices[0].delta = MagicMock() self.choices[0].finish_reason = finish_reason @@ -289,7 +293,7 @@ class TestReasoning(unittest.TestCase): else: # Need to handle attribute access that would raise AttributeError delattr(self.choices[0].delta, "reasoning_content") - + # Set reasoning if provided if reasoning is not None: self.choices[0].delta.reasoning = reasoning