mirror of
https://github.com/hwchase17/langchain.git
synced 2025-04-29 20:35:43 +00:00
a70f31de5f
# community: Fix AttributeError in RankLLMRerank (`list` object has no attribute `candidates`) ## **Description** This PR fixes an issue in `RankLLMRerank` where reranking fails with the following error: ``` AttributeError: 'list' object has no attribute 'candidates' ``` The issue arises because `rerank_batch()` returns a `List[Result]` instead of an object containing `.candidates`. ### **Changes Introduced** - Adjusted `compress_documents()` to support both: - Old API format: `rerank_results.candidates` - New API format: `rerank_results` as a list - Also fix wrong .txt location parsing while I was at it. --- ## **Issue** Fixes **AttributeError** in `RankLLMRerank` when using `compression_retriever.invoke()`. The issue is observed when `rerank_batch()` returns a list instead of an object with `.candidates`. **Relevant log:** ``` AttributeError: 'list' object has no attribute 'candidates' ``` ## **Dependencies** - No additional dependencies introduced. --- ## **Checklist** - [x] **Backward compatible** with previous API versions - [x] **Tested** locally with different RankLLM models - [x] **No new dependencies introduced** - [x] **Linted** with `make format && make lint` - [x] **Ready for review** --- ## **Testing** - Ran `compression_retriever.invoke(query)` ## **Reviewers** If no review within a few days, please **@mention** one of: - @baskaryan - @efriis - @eyurtsev - @ccurme - @vbarda - @hwchase17 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| dashscope_rerank.py | ||
| flashrank_rerank.py | ||
| infinity_rerank.py | ||
| jina_rerank.py | ||
| llmlingua_filter.py | ||
| openvino_rerank.py | ||
| rankllm_rerank.py | ||
| volcengine_rerank.py | ||