fix: use CUSTOM_CONNECT_TIMEOUT for gemini too
feat: add grok-4 to openrouter_models.json
This commit is contained in:
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user