Skip to content

Commit 4015bb0

Browse files
authored
fix: update default Cohere rerank model to rerank-v4.0-fast
1 parent 155e590 commit 4015bb0

4 files changed

Lines changed: 7 additions & 6 deletions

File tree

flowsettings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@
310310
KH_RERANKINGS["cohere"] = {
311311
"spec": {
312312
"__type__": "kotaemon.rerankings.CohereReranking",
313-
"model_name": "rerank-multilingual-v2.0",
313+
"model_name": "rerank-v4.0-fast",
314314
"cohere_api_key": config("COHERE_API_KEY", default=""),
315315
},
316316
"default": True,

libs/kotaemon/kotaemon/indices/rankings/cohere.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99

1010
class CohereReranking(BaseReranking):
11-
model_name: str = "rerank-multilingual-v2.0"
11+
model_name: str = "rerank-v4.0-fast"
1212
cohere_api_key: str = config("COHERE_API_KEY", "")
1313
use_key_from_ktem: bool = False
1414

libs/kotaemon/kotaemon/rerankings/cohere.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@ class CohereReranking(BaseReranking):
1313
"""Cohere Reranking model"""
1414

1515
model_name: str = Param(
16-
"rerank-multilingual-v2.0",
16+
"rerank-v4.0-fast",
1717
help=(
18-
"ID of the model to use. You can go to [Supported Models]"
19-
"(https://docs.cohere.com/docs/rerank-2) to see the supported models"
18+
"ID of the model to use. See [Cohere Rerank models]"
19+
"(https://docs.cohere.com/docs/models#rerank) for supported IDs "
20+
"(e.g. rerank-v4.0-fast, rerank-v4.0-pro, rerank-multilingual-v3.0)."
2021
),
2122
required=True,
2223
)

libs/ktem/ktem/pages/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def update_model(
226226
name="cohere",
227227
spec={
228228
"__type__": "kotaemon.rerankings.CohereReranking",
229-
"model_name": "rerank-multilingual-v2.0",
229+
"model_name": "rerank-v4.0-fast",
230230
"cohere_api_key": cohere_api_key,
231231
},
232232
default=True,

0 commit comments

Comments
 (0)