Tests for Grok 4.

This commit is contained in:
Raymond Lucke
2025-07-10 18:15:31 -07:00
parent 3ff6fef086
commit 39e2bc61b6
3 changed files with 105 additions and 28 deletions

View File

@@ -75,19 +75,25 @@ class TestSupportedModelsAliases:
assert isinstance(config.aliases, list), f"{model_name} aliases must be a list"
# Test specific aliases
assert "grok" in provider.SUPPORTED_MODELS["grok-3"].aliases
assert "grok" in provider.SUPPORTED_MODELS["grok-4-0709"].aliases
assert "grok-4" in provider.SUPPORTED_MODELS["grok-4-0709"].aliases
assert "grok-4-latest" in provider.SUPPORTED_MODELS["grok-4-0709"].aliases
assert "grok4" in provider.SUPPORTED_MODELS["grok-4-0709"].aliases
assert "grok3" in provider.SUPPORTED_MODELS["grok-3"].aliases
assert "grok3fast" in provider.SUPPORTED_MODELS["grok-3-fast"].aliases
assert "grokfast" in provider.SUPPORTED_MODELS["grok-3-fast"].aliases
# Test alias resolution
assert provider._resolve_model_name("grok") == "grok-3"
assert provider._resolve_model_name("grok") == "grok-4-0709"
assert provider._resolve_model_name("grok4") == "grok-4-0709"
assert provider._resolve_model_name("grok-4") == "grok-4-0709"
assert provider._resolve_model_name("grok3") == "grok-3"
assert provider._resolve_model_name("grok3fast") == "grok-3-fast"
assert provider._resolve_model_name("grokfast") == "grok-3-fast"
# Test case insensitive resolution
assert provider._resolve_model_name("Grok") == "grok-3"
assert provider._resolve_model_name("Grok") == "grok-4-0709"
assert provider._resolve_model_name("GROK4") == "grok-4-0709"
assert provider._resolve_model_name("GROKFAST") == "grok-3-fast"
def test_dial_provider_aliases(self):