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

22
tests/test_auto_mode_comprehensive.py
View File

@@ -80,9 +80,9 @@ class TestAutoModeComprehensive:
"OPENROUTER_API_KEY": None,
},
{
"EXTENDED_REASONING": "gemini-2.5-pro-preview-06-05", # Pro for deep thinking
"FAST_RESPONSE": "gemini-2.5-flash-preview-05-20", # Flash for speed
"BALANCED": "gemini-2.5-flash-preview-05-20", # Flash as balanced
"EXTENDED_REASONING": "gemini-2.5-pro", # Pro for deep thinking
"FAST_RESPONSE": "gemini-2.5-flash", # Flash for speed
"BALANCED": "gemini-2.5-flash", # Flash as balanced
},
),
# Only OpenAI API available
@@ -313,8 +313,8 @@ class TestAutoModeComprehensive:
# Should include Gemini models
assert "flash" in available_models
assert "pro" in available_models
assert "gemini-2.5-flash-preview-05-20" in available_models
assert "gemini-2.5-pro-preview-06-05" in available_models
assert "gemini-2.5-flash" in available_models
assert "gemini-2.5-pro" in available_models
# Should also include other models (users might have OpenRouter configured)
# The schema should show all options; validation happens at runtime
@@ -476,8 +476,8 @@ class TestAutoModeComprehensive:
assert "o3-mini" not in available_models
# Should still include all Gemini models (no restrictions)
assert "gemini-2.5-flash-preview-05-20" in available_models
assert "gemini-2.5-pro-preview-06-05" in available_models
assert "gemini-2.5-flash" in available_models
assert "gemini-2.5-pro" in available_models
def test_openrouter_fallback_when_no_native_apis(self):
"""Test that OpenRouter provides fallback models when no native APIs are available."""
@@ -511,8 +511,8 @@ class TestAutoModeComprehensive:
# Mock OpenRouter registry to return known models
mock_registry = MagicMock()
mock_registry.list_models.return_value = [
"google/gemini-2.5-flash-preview-05-20",
"google/gemini-2.5-pro-preview-06-05",
"google/gemini-2.5-flash",
"google/gemini-2.5-pro",
"openai/o3",
"openai/o4-mini",
"anthropic/claude-3-opus",
@@ -564,11 +564,11 @@ class TestAutoModeComprehensive:
mock_provider = MagicMock()
mock_response = MagicMock()
mock_response.content = "test response"
mock_response.model_name = "gemini-2.5-flash-preview-05-20" # The resolved name
mock_response.model_name = "gemini-2.5-flash" # The resolved name
mock_response.usage = {"input_tokens": 10, "output_tokens": 5}
# Mock _resolve_model_name to simulate alias resolution
mock_provider._resolve_model_name = lambda alias: (
"gemini-2.5-flash-preview-05-20" if alias == "flash" else alias
"gemini-2.5-flash" if alias == "flash" else alias
)
mock_provider.generate_content.return_value = mock_response