fix: upgrade rerankers and remove flashrank patch (#47)

Author: lsorber

poetry.lock

@@ -39,7 +39,7 @@ pydantic = ">=2.7.0"
 pynndescent = ">=0.5.12"
 # Reranking:
 langdetect = ">=1.0.9"
-rerankers = { extras = ["flashrank"], version = ">=0.5.3" }
+rerankers = { extras = ["flashrank"], version = ">=0.6.0" }
 # Storage:
 pg8000 = ">=1.31.2"
 sqlmodel-slim = ">=0.0.18"

pyproject.toml

@@ -8,11 +8,10 @@
 from llama_cpp import llama_supports_gpu_offload
 from sqlalchemy.engine import URL
 
-from raglite._flashrank import PatchedFlashRankRanker as FlashRankRanker
-
 # Suppress rerankers output on import until [1] is fixed.
 # [1] https://github.com/AnswerDotAI/rerankers/issues/36
 with contextlib.redirect_stdout(StringIO()):
+    from rerankers.models.flashrank_ranker import FlashRankRanker
     from rerankers.models.ranker import BaseRanker
 
 

src/raglite/_config.py

@@ -1,11 +1,11 @@
 """Test RAGLite's reranking functionality."""
 
 import pytest
+from rerankers.models.flashrank_ranker import FlashRankRanker
 from rerankers.models.ranker import BaseRanker
 
 from raglite import RAGLiteConfig, hybrid_search, rerank_chunks, retrieve_chunks
 from raglite._database import Chunk
-from raglite._flashrank import PatchedFlashRankRanker as FlashRankRanker
 
 
 @pytest.fixture(