Fixed tests

This commit is contained in:
Fahad
2025-06-12 21:00:53 +04:00
parent 3aedb16101
commit 22093bbf18
2 changed files with 3 additions and 1 deletions

View File

@@ -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"""

View File

@@ -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()