rerank-2-lite

!pip install -qU voyageai pinecone

import voyageai

# These are the documents returned from a search using Pinecone
documents = [
		"Apple is a popular fruit known for its sweetness and crisp texture.",
		"The tech company Apple is known for its innovative products like the iPhone.",
		"Many people enjoy eating apples as a healthy snack.",
		"Apple Inc. has revolutionized the tech industry with its sleek designs and user-friendly interfaces.",
		"An apple a day keeps the doctor away, as the saying goes."
]

query = "Tell me about the tech company known as Apple"

co = voyageai.Client(api_key=VOYAGE_API_KEY)
reranked_docs = vo.rerank(
    query=query,
    documents=documents,
    model="rerank-2",
    top_k=3
)