Gemini model rename
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -43,7 +43,7 @@ class OpenRouterModelsTest(BaseSimulatorTest):
|
||||
self.setup_test_files()
|
||||
|
||||
# Test 1: Flash alias mapping to OpenRouter
|
||||
self.logger.info(" 1: Testing 'flash' alias (should map to google/gemini-2.5-flash-preview-05-20)")
|
||||
self.logger.info(" 1: Testing 'flash' alias (should map to google/gemini-2.5-flash)")
|
||||
|
||||
response1, continuation_id = self.call_mcp_tool(
|
||||
"chat",
|
||||
@@ -63,7 +63,7 @@ class OpenRouterModelsTest(BaseSimulatorTest):
|
||||
self.logger.info(f" ✅ Got continuation_id: {continuation_id}")
|
||||
|
||||
# Test 2: Pro alias mapping to OpenRouter
|
||||
self.logger.info(" 2: Testing 'pro' alias (should map to google/gemini-2.5-pro-preview-06-05)")
|
||||
self.logger.info(" 2: Testing 'pro' alias (should map to google/gemini-2.5-pro)")
|
||||
|
||||
response2, _ = self.call_mcp_tool(
|
||||
"chat",
|
||||
|
||||
Reference in New Issue
Block a user