AnswerDotAI / rerankers

A lightweight, low-dependency, unified API to use all common reranking and cross-encoder models.
Apache License 2.0
784 stars 40 forks source link

Allow arbitrary "document" types #1

Open janheinrichmerker opened 6 months ago

janheinrichmerker commented 6 months ago

Just a quick thought: Why restrict the basic re-ranking interface to just strings? For example, one might want to re-rank images and there might be models that support that. I believe, the two main model classes of this library, Result and RankedResults, can pretty easily be relaxed to support generic document types (i.e., by extending Generic[T] and swapping str for T).