fix: Update test_thinking_tokens_option to check correct thinking tokens location

This commit is contained in:
Paul Gauthier (aider) 2025-03-18 12:22:13 -07:00
parent 50d8a19397
commit 14761ebec2

View file

@ -1012,5 +1012,5 @@ class TestMain(TestCase):
return_coder=True,
)
self.assertEqual(
coder.main_model.extra_params.get("extra_body", {}).get("thinking_tokens"), "1000"
coder.main_model.extra_params.get("thinking", {}).get("budget_tokens"), 1000
)