Reranker

org.llm4s.reranker.Reranker
trait Reranker

Cross-encoder reranker interface.

Rerankers improve search quality by re-scoring candidate documents using a cross-encoder model that sees both query and document together. This produces more accurate relevance scores than bi-encoder (embedding) similarity alone.

Usage:

val reranker = RerankerFactory.cohere(config)
val request = RerankRequest(
 query = "What is Scala?",
 documents = Seq("Scala is a programming language", "Python is popular"),
 topK = Some(5)
)
val response = reranker.rerank(request)
response.foreach { r =>
 r.results.foreach(rr => println(s"Score $${rr.score}: $${rr.document.take(50)}..."))
}

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Known subtypes

Members list

Value members

Abstract methods

Rerank documents against a query.

Rerank documents against a query.

Value parameters

request

The rerank request containing query and documents

Attributes

Returns

Reranked results or error