Gemini model rename

This commit is contained in:
Fahad
2025-06-19 05:37:40 +04:00
parent b6ad76b39a
commit d0da6ce9e4
37 changed files with 187 additions and 187 deletions

View File

@@ -30,7 +30,7 @@ class TestModelThinkingConfig(BaseSimulatorTest):
"chat",
{
"prompt": "What is 2 + 2? Please think carefully and explain.",
"model": "pro", # Should resolve to gemini-2.5-pro-preview-06-05
"model": "pro", # Should resolve to gemini-2.5-pro
"thinking_mode": "high", # Should use thinking_config
},
)
@@ -55,7 +55,7 @@ class TestModelThinkingConfig(BaseSimulatorTest):
"chat",
{
"prompt": "What is 3 + 3? Give a quick answer.",
"model": "flash", # Should resolve to gemini-2.5-flash-preview-05-20
"model": "flash", # Should resolve to gemini-2.5-flash
"thinking_mode": "high", # Should be ignored for Flash model
},
)
@@ -79,8 +79,8 @@ class TestModelThinkingConfig(BaseSimulatorTest):
test_cases = [
("pro", "should work with Pro model"),
("flash", "should work with Flash model"),
("gemini-2.5-pro-preview-06-05", "should work with full Pro model name"),
("gemini-2.5-flash-preview-05-20", "should work with full Flash model name"),
("gemini-2.5-pro", "should work with full Pro model name"),
("gemini-2.5-flash", "should work with full Flash model name"),
]
success_count = 0