torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Gemini model rename

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_auto_mode_provider_selection.py
View File

@@ -59,9 +59,9 @@ class TestAutoModeProviderSelection:
balanced = ModelProviderRegistry.get_preferred_fallback_model(ToolModelCategory.BALANCED)
# Should select appropriate Gemini models
assert extended_reasoning in ["gemini-2.5-pro-preview-06-05", "pro"]
assert fast_response in ["gemini-2.5-flash-preview-05-20", "flash"]
assert balanced in ["gemini-2.5-flash-preview-05-20", "flash"]
assert extended_reasoning in ["gemini-2.5-pro", "pro"]
assert fast_response in ["gemini-2.5-flash", "flash"]
assert balanced in ["gemini-2.5-flash", "flash"]
finally:
# Restore original environment
@@ -229,8 +229,8 @@ class TestAutoModeProviderSelection:
assert "o3-mini" not in available_models
# Should include all Gemini models (no restrictions)
assert "gemini-2.5-flash-preview-05-20" in available_models
assert available_models["gemini-2.5-flash-preview-05-20"] == ProviderType.GOOGLE
assert "gemini-2.5-flash" in available_models
assert available_models["gemini-2.5-flash"] == ProviderType.GOOGLE
finally:
# Restore original environment
@@ -316,8 +316,8 @@ class TestAutoModeProviderSelection:
# Test that providers resolve aliases correctly
test_cases = [
("flash", ProviderType.GOOGLE, "gemini-2.5-flash-preview-05-20"),
("pro", ProviderType.GOOGLE, "gemini-2.5-pro-preview-06-05"),
("flash", ProviderType.GOOGLE, "gemini-2.5-flash"),
("pro", ProviderType.GOOGLE, "gemini-2.5-pro"),
("mini", ProviderType.OPENAI, "o4-mini"),
("o3mini", ProviderType.OPENAI, "o3-mini"),
("grok", ProviderType.XAI, "grok-3"),