class CohereRerankRetriever(BaseRetriever):
BaseRetriever
using the Cohere Re-ranking
model.
Parameters:
def __init__(
self,
model_name: str = 'rerank-multilingual-v2.0',
api_key: Optional[str] = None
):
def query(
self,
query: str,
retrieved_result: List[Dict[str, Any]],
top_k: int = DEFAULT_TOP_K_RESULTS
):
BaseRetriever
like VectorRetriever
.DEFAULT_TOP_K_RESULTS
.