fix: flashrank ordering (#46)

bclavie · web-flow · commit 89c76f23f050 · 2024-11-12T09:10:08.000+01:00
diff --git a/rerankers/models/flashrank_ranker.py b/rerankers/models/flashrank_ranker.py
@@ -45,12 +45,13 @@ def rank(
 
         ranked_results = [
             Result(
-                document=docs[idx],
+                document=docs[result["id"]],  # Returns reranked documents.
                 score=result["score"],
                 rank=idx + 1,
             )
             for idx, result in enumerate(flashrank_results)
         ]
+        
         return RankedResults(results=ranked_results, query=query, has_scores=True)
 
     def score(self, query: str, doc: str) -> float:

Original file line number	Diff line number	Diff line change
`@@ -45,12 +45,13 @@ def rank(`
`45`	`45`
`46`	`46`	`ranked_results = [`
`47`	`47`	`Result(`
`48`		`- document=docs[idx],`
	`48`	`+ document=docs[result["id"]], # Returns reranked documents.`
`49`	`49`	`score=result["score"],`
`50`	`50`	`rank=idx + 1,`
`51`	`51`	`)`
`52`	`52`	`for idx, result in enumerate(flashrank_results)`
`53`	`53`	`]`
	`54`	`+`
`54`	`55`	`return RankedResults(results=ranked_results, query=query, has_scores=True)`
`55`	`56`
`56`	`57`	`def score(self, query: str, doc: str) -> float:`