voyage-ai / voyageai-python

Voyage AI Official Python Library
https://www.voyageai.com
MIT License
37 stars 5 forks source link

Reranker API #5

Closed Amiya-Lahiri-ML closed 3 months ago

Amiya-Lahiri-ML commented 4 months ago

AttributeError Traceback (most recent call last) in <cell line: 3>() 1 import voyageai 2 vo = voyageai.Client(api_key=VOYAGE_API_KEY) ----> 3 reranking = vo.rerank(query, RerankerSummaryList, model="rerank-lite-1", top_k=5) 4 for r in reranking.results: 5 print(f"Document: {r.document}")

AttributeError: 'Client' object has no attribute 'rerank' how to resolve it

thomas0809 commented 4 months ago

Hi,

Please upgrade to the latest version of the package through pip install -U voyageai, and you should be able to use the rerank method in Client. https://github.com/voyage-ai/voyageai-python/blob/e565fb60b854e80ead526a57ea0e6eb1db9efc33/voyageai/client.py#L81.