From 14761ebec21ddf8477b415a1a0a167ed26f324e4 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Tue, 18 Mar 2025 12:22:13 -0700
Subject: [PATCH] fix: Update test_thinking_tokens_option to check correct
 thinking tokens location

---
 tests/basic/test_main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index bc989475a..5d7a8ebc9 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -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
         )