feat: Add existence checks for thinking mode and budget toggles to support models without them.

Author: NikkeTryHard

browser_utils/page_controller_modules/thinking.py

@@ -444,6 +444,23 @@ async def _control_thinking_mode_toggle(
 
         try:
             toggle_locator = self.page.locator(toggle_selector)
+
+            # First check if element exists at all (for non-thinking models like gemini-2.0-flash)
+            element_count = await toggle_locator.count()
+            if element_count == 0:
+                if not should_be_enabled:
+                    # Trying to disable on a model without thinking toggle - just skip
+                    self.logger.info(
+                        " 主思考开关不存在（当前模型不支持思考模式），无需关闭。"
+                    )
+                    return True
+                else:
+                    # User wants to enable but toggle doesn't exist
+                    self.logger.warning(
+                        " 主思考开关不存在（当前模型可能不支持思考模式），无法开启。"
+                    )
+                    return False
+
             await expect_async(toggle_locator).to_be_visible(timeout=5000)
             try:
                 await toggle_locator.scroll_into_view_if_needed()
@@ -537,6 +554,23 @@ async def _control_thinking_budget_toggle(
 
         try:
             toggle_locator = self.page.locator(toggle_selector)
+
+            # First check if element exists at all (for non-thinking models)
+            element_count = await toggle_locator.count()
+            if element_count == 0:
+                if not should_be_checked:
+                    # Trying to disable on a model without budget toggle - just skip
+                    self.logger.info(
+                        " 思考预算开关不存在（当前模型不支持），无需禁用。"
+                    )
+                    return
+                else:
+                    # User wants to enable but toggle doesn't exist
+                    self.logger.warning(
+                        " 思考预算开关不存在（当前模型可能不支持），无法启用。"
+                    )
+                    return
+
             await expect_async(toggle_locator).to_be_visible(timeout=5000)
             try:
                 await toggle_locator.scroll_into_view_if_needed()

tests/browser_utils/page_controller_modules/test_thinking.py

@@ -628,6 +628,7 @@ async def test_set_thinking_level_errors(mock_controller):
 async def test_control_thinking_mode_toggle_fallback(mock_controller):
     # Test fallback to aria-label based toggle click
     toggle = MagicMock()
+    toggle.count = AsyncMock(return_value=1)  # Element exists
     toggle.get_attribute = AsyncMock(return_value="false")
     toggle.click = AsyncMock(side_effect=Exception("Click failed"))
 
@@ -666,6 +667,7 @@ def locator_side_effect(selector):
 async def test_control_thinking_budget_toggle_fallback(mock_controller):
     # Test fallback to aria-label based toggle click
     toggle = MagicMock()
+    toggle.count = AsyncMock(return_value=1)  # Element exists
     toggle.get_attribute = AsyncMock(return_value="false")
     toggle.click = AsyncMock(side_effect=Exception("Click failed"))