diff --git a/tests/test_providers.py b/tests/test_providers.py index ea35301..f1827f5 100644 --- a/tests/test_providers.py +++ b/tests/test_providers.py @@ -85,7 +85,7 @@ class TestGeminiProvider: assert capabilities.provider == ProviderType.GOOGLE assert capabilities.model_name == "gemini-2.5-flash-preview-05-20" assert capabilities.max_tokens == 1_048_576 - assert not capabilities.supports_extended_thinking + assert capabilities.supports_extended_thinking def test_get_capabilities_pro_model(self): """Test getting capabilities for Pro model with thinking support""" diff --git a/tests/test_thinking_modes.py b/tests/test_thinking_modes.py index 0fb4a68..4a623e6 100644 --- a/tests/test_thinking_modes.py +++ b/tests/test_thinking_modes.py @@ -23,6 +23,7 @@ def setup_test_env(): class TestThinkingModes: """Test thinking modes across all tools""" + @patch("config.DEFAULT_THINKING_MODE_THINKDEEP", "high") def test_default_thinking_modes(self): """Test that tools have correct default thinking modes""" tools = [ @@ -169,6 +170,7 @@ class TestThinkingModes: @pytest.mark.asyncio @patch("tools.base.BaseTool.get_model_provider") + @patch("config.DEFAULT_THINKING_MODE_THINKDEEP", "high") async def test_thinking_mode_max(self, mock_get_provider): """Test max thinking mode (default for thinkdeep)""" mock_provider = create_mock_provider()