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

Cleanup and confirm tests pass

Browse Source
This commit is contained in:
Fahad
2025-06-13 16:09:40 +04:00
parent 048ebf90bf
commit a7b27b285c
3 changed files with 15 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
conf/custom_models.json
View File

@@ -103,27 +103,9 @@
"supports_function_calling": false,
"description": "Meta's Llama 3 70B model"
},
{
"model_name": "openai/gpt-4o",
"aliases": ["gpt4o", "4o", "gpt-4o"],
"context_window": 128000,
"supports_extended_thinking": false,
"supports_json_mode": true,
"supports_function_calling": true,
"description": "OpenAI's GPT-4o model"
},
{
"model_name": "deepseek/deepseek-coder",
"aliases": ["deepseek-coder", "deepseek", "coder"],
"context_window": 65536,
"supports_extended_thinking": false,
"supports_json_mode": true,
"supports_function_calling": true,
"description": "DeepSeek Coder model for programming tasks"
},
{
"model_name": "deepseek/deepseek-r1-0528",
"aliases": ["deepseek-r1", "r1", "deepseek-thinking"],
"aliases": ["deepseek-r1", "deepseek", "r1", "deepseek-thinking"],
"context_window": 65536,
"supports_extended_thinking": true,
"supports_json_mode": true,

20
tests/test_openrouter_provider.py
View File

@@ -52,9 +52,9 @@ class TestOpenRouterProvider:
provider = OpenRouterProvider(api_key="test-key")
# Test with a model in the registry (using alias)
caps = provider.get_capabilities("gpt4o")
caps = provider.get_capabilities("o3")
assert caps.provider == ProviderType.OPENROUTER
assert caps.model_name == "openai/gpt-4o" # Resolved name
assert caps.model_name == "openai/o3" # Resolved name
assert caps.friendly_name == "OpenRouter"
# Test with a model not in registry - should get generic capabilities
@@ -71,22 +71,22 @@ class TestOpenRouterProvider:
# Test alias resolution
assert provider._resolve_model_name("opus") == "anthropic/claude-3-opus"
assert provider._resolve_model_name("sonnet") == "anthropic/claude-3-sonnet"
assert provider._resolve_model_name("gpt4o") == "openai/gpt-4o"
assert provider._resolve_model_name("4o") == "openai/gpt-4o"
assert provider._resolve_model_name("o3") == "openai/o3"
assert provider._resolve_model_name("o3-mini") == "openai/o3-mini-high"
assert provider._resolve_model_name("claude") == "anthropic/claude-3-sonnet"
assert provider._resolve_model_name("mistral") == "mistral/mistral-large"
assert provider._resolve_model_name("deepseek") == "deepseek/deepseek-coder"
assert provider._resolve_model_name("coder") == "deepseek/deepseek-coder"
assert provider._resolve_model_name("deepseek") == "deepseek/deepseek-r1-0528"
assert provider._resolve_model_name("r1") == "deepseek/deepseek-r1-0528"
# Test case-insensitive
assert provider._resolve_model_name("OPUS") == "anthropic/claude-3-opus"
assert provider._resolve_model_name("GPT4O") == "openai/gpt-4o"
assert provider._resolve_model_name("O3") == "openai/o3"
assert provider._resolve_model_name("Mistral") == "mistral/mistral-large"
assert provider._resolve_model_name("CLAUDE") == "anthropic/claude-3-sonnet"
# Test direct model names (should pass through unchanged)
assert provider._resolve_model_name("anthropic/claude-3-opus") == "anthropic/claude-3-opus"
assert provider._resolve_model_name("openai/gpt-4o") == "openai/gpt-4o"
assert provider._resolve_model_name("openai/o3") == "openai/o3"
# Test unknown models pass through
assert provider._resolve_model_name("unknown-model") == "unknown-model"
@@ -120,13 +120,13 @@ class TestOpenRouterRegistry:
models = registry.list_models()
assert len(models) > 0
assert "anthropic/claude-3-opus" in models
assert "openai/gpt-4o" in models
assert "openai/o3" in models
# Should have loaded aliases
aliases = registry.list_aliases()
assert len(aliases) > 0
assert "opus" in aliases
assert "gpt4o" in aliases
assert "o3" in aliases
assert "claude" in aliases
def test_registry_capabilities(self):

8
tests/test_openrouter_registry.py
View File

@@ -77,8 +77,8 @@ class TestOpenRouterModelRegistry:
("opus", "anthropic/claude-3-opus"),
("OPUS", "anthropic/claude-3-opus"), # Case insensitive
("claude", "anthropic/claude-3-sonnet"),
("gpt4o", "openai/gpt-4o"),
("4o", "openai/gpt-4o"),
("o3", "openai/o3"),
("deepseek", "deepseek/deepseek-r1-0528"),
("mistral", "mistral/mistral-large"),
]
@@ -96,9 +96,9 @@ class TestOpenRouterModelRegistry:
assert config is not None
assert config.model_name == "anthropic/claude-3-opus"
config = registry.resolve("openai/gpt-4o")
config = registry.resolve("openai/o3")
assert config is not None
assert config.model_name == "openai/gpt-4o"
assert config.model_name == "openai/o3"
def test_unknown_model_resolution(self):
"""Test resolution of unknown models."""