Merge pull request #36 from prabhkaran/fix/rank-extraction-in-listwis…

…e-ranking

Fix: rank extraction by stripping square brackets in listwise ranking

flashrank/Ranker.py

@@ -210,7 +210,7 @@ def rerank(self, request: RerankRequest) -> List[Dict[str, Any]]:
             raw_ranks = self.llm_model.create_chat_completion(messages)
             results = []
             for rank in raw_ranks["choices"][0]["message"]["content"].split(" > "):
-                results.append(result_map[int(rank[1])])
+                results.append(result_map[int(rank.strip("[]"))])
             return results    
 
         # self.session will be instantiated for ONNX based pairwise CE models