From 4015bb0f7cebd59d474624c4a570a4c119cdd826 Mon Sep 17 00:00:00 2001
From: Kudo Khang <hnkhang.dev@gmail.com>
Date: Sat, 28 Mar 2026 23:29:42 +0700
Subject: [PATCH] fix: update default Cohere rerank model to rerank-v4.0-fast

---
 flowsettings.py                                   | 2 +-
 libs/kotaemon/kotaemon/indices/rankings/cohere.py | 2 +-
 libs/kotaemon/kotaemon/rerankings/cohere.py       | 7 ++++---
 libs/ktem/ktem/pages/setup.py                     | 2 +-
 4 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/flowsettings.py b/flowsettings.py
index 2ec2de8a..8bd6db31 100644
--- a/flowsettings.py
+++ b/flowsettings.py
@@ -310,7 +310,7 @@ KH_EMBEDDINGS["mistral"] = {
 KH_RERANKINGS["cohere"] = {
     "spec": {
         "__type__": "kotaemon.rerankings.CohereReranking",
-        "model_name": "rerank-multilingual-v2.0",
+        "model_name": "rerank-v4.0-fast",
         "cohere_api_key": config("COHERE_API_KEY", default=""),
     },
     "default": True,
diff --git a/libs/kotaemon/kotaemon/indices/rankings/cohere.py b/libs/kotaemon/kotaemon/indices/rankings/cohere.py
index 9515d12f..039a00db 100644
--- a/libs/kotaemon/kotaemon/indices/rankings/cohere.py
+++ b/libs/kotaemon/kotaemon/indices/rankings/cohere.py
@@ -8,7 +8,7 @@ from .base import BaseReranking
 
 
 class CohereReranking(BaseReranking):
-    model_name: str = "rerank-multilingual-v2.0"
+    model_name: str = "rerank-v4.0-fast"
     cohere_api_key: str = config("COHERE_API_KEY", "")
     use_key_from_ktem: bool = False
 
diff --git a/libs/kotaemon/kotaemon/rerankings/cohere.py b/libs/kotaemon/kotaemon/rerankings/cohere.py
index ab357007..ffc3baf5 100644
--- a/libs/kotaemon/kotaemon/rerankings/cohere.py
+++ b/libs/kotaemon/kotaemon/rerankings/cohere.py
@@ -13,10 +13,11 @@ class CohereReranking(BaseReranking):
     """Cohere Reranking model"""
 
     model_name: str = Param(
-        "rerank-multilingual-v2.0",
+        "rerank-v4.0-fast",
         help=(
-            "ID of the model to use. You can go to [Supported Models]"
-            "(https://docs.cohere.com/docs/rerank-2) to see the supported models"
+            "ID of the model to use. See [Cohere Rerank models]"
+            "(https://docs.cohere.com/docs/models#rerank) for supported IDs "
+            "(e.g. rerank-v4.0-fast, rerank-v4.0-pro, rerank-multilingual-v3.0)."
         ),
         required=True,
     )
diff --git a/libs/ktem/ktem/pages/setup.py b/libs/ktem/ktem/pages/setup.py
index 09e3b6a1..be5c917a 100644
--- a/libs/ktem/ktem/pages/setup.py
+++ b/libs/ktem/ktem/pages/setup.py
@@ -226,7 +226,7 @@ class SetupPage(BasePage):
                     name="cohere",
                     spec={
                         "__type__": "kotaemon.rerankings.CohereReranking",
-                        "model_name": "rerank-multilingual-v2.0",
+                        "model_name": "rerank-v4.0-fast",
                         "cohere_api_key": cohere_api_key,
                     },
                     default=True,