diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index 1ebb1d332..7e455099e 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -32,7 +32,7 @@ class TestReasoning(unittest.TestCase): # Create a mock hash object mock_hash = MagicMock() mock_hash.hexdigest.return_value = "mock_hash_digest" - + # Mock the model's send_completion method to return the expected tuple format with patch.object(model, "send_completion", return_value=(mock_hash, mock_completion)): # Call send with a simple message