Fix CrossEncoderRanker (#1531)

Co-authored-by: dong <dongzhancai@iie2.com> Co-authored-by: Zhancai Dong <dongzhancai1@jd.com>
eosphoros-ai · May 20, 2024 · 04af30e · 04af30e
1 parent dcb1bd3
commit 04af30e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dbgpt/rag/retriever/rerank.py b/dbgpt/rag/retriever/rerank.py
@@ -239,6 +239,8 @@ def rank(
         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 = [
             [