Skip to content

Cohere

Reranking via Cohere's API.

Overview

Field Value
Type API
Provider Cohere
Default Model rerank-v3.5
Env Variable COHERE_API_KEY

Installation

pip install "autorag-research[reranker]"
# or
uv add "autorag-research[reranker]"

Configuration

_target_: autorag_research.rerankers.cohere.CohereReranker
model_name: rerank-v3.5

Options

Option Type Default Description
model_name str rerank-v3.5 Cohere model name
api_key str None API key (or use env var)
batch_size int 64 Batch size for multiple queries

Models

Model Description
rerank-v3.5 Latest, best quality
rerank-english-v3.0 English optimized
rerank-multilingual-v3.0 Multilingual

Usage

from autorag_research.rerankers import CohereReranker

reranker = CohereReranker(model_name="rerank-v3.5")
results = reranker.rerank("What is RAG?", ["doc1", "doc2", "doc3"], top_k=2)

for r in results:
    print(f"[{r.index}] {r.score:.3f}: {r.text[:50]}...")