diff --git a/studio/backend/core/inference/providers.py b/studio/backend/core/inference/providers.py index 0fb86431f5..7c98c2dc90 100644 --- a/studio/backend/core/inference/providers.py +++ b/studio/backend/core/inference/providers.py @@ -61,9 +61,9 @@ PROVIDER_REGISTRY: dict[str, dict[str, Any]] = { "display_name": "Cohere", "base_url": "https://api.cohere.ai/compatibility/v1", "default_models": [ - "command-r-plus", - "command-r", - "command-a", + "command-a-03-2025", + "command-r-plus-08-2024", + "command-r-08-2024", ], "supports_streaming": True, "supports_vision": False, diff --git a/studio/backend/tests/test_providers_api.py b/studio/backend/tests/test_providers_api.py index c355a87aa5..d044f24c33 100644 --- a/studio/backend/tests/test_providers_api.py +++ b/studio/backend/tests/test_providers_api.py @@ -44,7 +44,7 @@ _PROVIDER_CONFIGS: dict[str, tuple[str, str]] = { "openai": ("OPENAI_API_KEY", "gpt-4o-mini"), "mistral": ("MISTRAL_API_KEY", "mistral-small-latest"), "google": ("GOOGLE_API_KEY", "gemini-2.5-flash"), - "cohere": ("COHERE_API_KEY", "command-r"), + "cohere": ("COHERE_API_KEY", "command-a-03-2025"), "together": ("TOGETHER_API_KEY", "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"), "fireworks": ("FIREWORKS_API_KEY", "accounts/fireworks/models/llama4-maverick-instruct-basic"), "perplexity": ("PERPLEXITY_API_KEY", "sonar"),