Cleanup and confirm tests pass

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

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."""