langchain: add RankLLM Reranker (#21171)

Integrate RankLLM reranker (https://github.com/castorini/rank_llm) into
LangChain

An example notebook is given in
`docs/docs/integrations/retrievers/rankllm-reranker.ipynb`

---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
This commit is contained in:
Eric Zhang
2024-05-22 16:12:55 -04:00
committed by GitHub
parent 14a9c7c44e
commit e7e41eaabe
7 changed files with 927 additions and 2 deletions

View File

@@ -0,0 +1 @@
"""Test document compressor integrations."""

View File

@@ -0,0 +1,8 @@
"""Test rankllm reranker."""
from langchain_community.document_compressors.rankllm_rerank import RankLLMRerank
def test_rankllm_reranker_init() -> None:
"""Test the RankLLM reranker initializes correctly."""
RankLLMRerank()

View File

@@ -4,6 +4,7 @@ EXPECTED_ALL = [
"LLMLinguaCompressor",
"OpenVINOReranker",
"JinaRerank",
"RankLLMRerank",
"FlashrankRerank",
]