Fixed tests
This commit is contained in:
@@ -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"""
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user