fix: use CUSTOM_CONNECT_TIMEOUT for gemini too

feat: add grok-4 to openrouter_models.json
This commit is contained in:
Fahad
2025-10-06 23:23:24 +04:00
parent a65485a1e5
commit a33efbde52
5 changed files with 68 additions and 11 deletions

View File

@@ -47,6 +47,7 @@ class TestOpenRouterProvider:
assert provider.validate_model_name("anthropic/claude-3-opus") is True
assert provider.validate_model_name("google/any-model-name") is True
assert provider.validate_model_name("groq/llama-3.1-8b") is True
assert provider.validate_model_name("grok-4") is True
# Unknown models without provider prefix are rejected
assert provider.validate_model_name("gpt-4") is False
@@ -88,6 +89,9 @@ class TestOpenRouterProvider:
assert provider._resolve_model_name("o4-mini") == "openai/o4-mini"
assert provider._resolve_model_name("haiku") == "anthropic/claude-3.5-haiku"
assert provider._resolve_model_name("mistral") == "mistralai/mistral-large-2411"
assert provider._resolve_model_name("grok-4") == "x-ai/grok-4"
assert provider._resolve_model_name("grok4") == "x-ai/grok-4"
assert provider._resolve_model_name("grok") == "x-ai/grok-4"
assert provider._resolve_model_name("deepseek") == "deepseek/deepseek-r1-0528"
assert provider._resolve_model_name("r1") == "deepseek/deepseek-r1-0528"