AnswerDotAI / rerankers

A lightweight, low-dependency, unified API to use all common reranking and cross-encoder models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Allow arbitrary "document" types

janheinrichmerker opened this issue · comments

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).