From 10e2b5fdef3a543ce807bdbf1c1c6a309fa82145 Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Sun, 29 Mar 2026 18:55:39 +0000 Subject: [PATCH] fix: update Cohere models to current (command-r retired Sept 2025) --- studio/backend/core/inference/providers.py | 6 +++--- studio/backend/tests/test_providers_api.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) 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"),