langchain-zerank 0.1.0

LangChain integration for ZeroEntropy zerank rerankers

langchain-zeroentropy

LangChain document compressor that reranks retrieved results using ZeroEntropy zerank models.

Installation

pip install langchain-zeroentropy

Usage

from langchain_zeroentropy import ZeroEntropyRerank
from langchain.retrievers import ContextualCompressionRetriever

compressor = ZeroEntropyRerank(model="zerank-1-small", top_n=3)

compression_retriever = ContextualCompressionRetriever(
    base_compressor=compressor,
    base_retriever=your_retriever,
)

docs = compression_retriever.invoke("what is the rebate on electronics?")

Set your API key via the ZEROENTROPY_API_KEY environment variable, or pass it directly:

compressor = ZeroEntropyRerank(api_key="ze-...", model="zerank-1-small", top_n=3)

Supported models

Model	Notes
zerank-1-small	Fast, Apache-2.0 licensed
zerank-1	Higher accuracy

See the ZeroEntropy docs for the full model list and free-tier limits.

Parameters

Parameter	Default	Description
model	zerank-1-small	ZeroEntropy reranker model
top_n	3	Number of top documents to return
api_key	None	API key (falls back to ZEROENTROPY_API_KEY env var)

License

MIT