rerankings

type RankedResult struct {
	Index  int // Index in the original input []string
	String string
	Rank   float32
}

type RerankedChromaResults struct {
	*chromago.QueryResultImpl
	QueryTexts []string               // Query texts used for reranking (not in V2 QueryResultImpl)
	Ranks      map[string][][]float32 // each reranker adds a rank for each result
}

type RerankingFunction interface {
	ID() string
	Rerank(ctx context.Context, query string, results []Result) (map[string][]RankedResult, error)
	RerankResults(ctx context.Context, queryTexts []string, queryResults *chromago.QueryResultImpl) (*RerankedChromaResults, error)
}

type RerankingModel string

type Result struct {
	Text   *string
	Object *any
}