Use the new flash model

Updated tests
2025-06-12 13:44:09 +04:00
parent 8b8d966d33
commit 79af2654b9
20 changed files with 297 additions and 63 deletions
--- a/tests/mock_helpers.py
+++ b/tests/mock_helpers.py
@@ -5,7 +5,7 @@ from unittest.mock import Mock
 from providers.base import ModelCapabilities, ProviderType, RangeTemperatureConstraint


-def create_mock_provider(model_name="gemini-2.0-flash-exp", max_tokens=1_048_576):
+def create_mock_provider(model_name="gemini-2.0-flash", max_tokens=1_048_576):
    """Create a properly configured mock provider."""
    mock_provider = Mock()