Fix CrossEncoderRanker (#1531)

Co-authored-by: dong <dongzhancai@iie2.com> Co-authored-by: Zhancai Dong <dongzhancai1@jd.com>
2025-07-24 12:45:45 +00:00 · 2024-05-20 21:17:06 +08:00 · 2024-05-20 21:17:06 +08:00 · 04af30e3db
commit 04af30e3db
parent dcb1bd35d0
1 changed files with 2 additions and 0 deletions
--- a/dbgpt/rag/retriever/rerank.py
+++ b/dbgpt/rag/retriever/rerank.py
@ -239,6 +239,8 @@ class CrossEncoderRanker(Ranker):
        Returns:
            List[Chunk], reranked candidates
        """
+        if len(candidates_with_scores) <= 1:
+            return candidates_with_scores
        contents = [candidate.content for candidate in candidates_with_scores]
        query_content_pairs = [
            [