style: Format code with linter and improve readability

This commit is contained in:
Paul Gauthier (aider) 2025-01-31 13:13:10 -08:00
parent 476a0ad6ad
commit ee9d0c4a99

View file

@ -824,7 +824,14 @@ class TestMain(TestCase):
self.fail(f"main() raised an unexpected exception: {e}") self.fail(f"main() raised an unexpected exception: {e}")
self.assertIsNone(result, "main() should return None when called with --exit") self.assertIsNone(result, "main() should return None when called with --exit")
def test_reasoning_effort_option(self): def test_reasoning_effort_option(self):
coder = main(["--reasoning-effort", "3", "--yes", "--exit"], input=DummyInput(), output=DummyOutput(), return_coder=True) coder = main(
self.assertEqual(coder.main_model.extra_params.get("extra_body", {}).get("reasoning_effort"), "3") ["--reasoning-effort", "3", "--yes", "--exit"],
input=DummyInput(),
output=DummyOutput(),
return_coder=True,
)
self.assertEqual(
coder.main_model.extra_params.get("extra_body", {}).get("reasoning_effort"), "3"
)