diff --git a/api/core/rag/rerank/rerank_model.py b/api/core/rag/rerank/rerank_model.py index 7f06732f31..693535413a 100644 --- a/api/core/rag/rerank/rerank_model.py +++ b/api/core/rag/rerank/rerank_model.py @@ -63,5 +63,5 @@ class RerankModelRunner(BaseRerankRunner): rerank_document.metadata["score"] = result.score rerank_documents.append(rerank_document) - rerank_documents.sort(key=lambda x: x.metadata.get('score', 0.0), reverse=True) + rerank_documents.sort(key=lambda x: x.metadata.get("score", 0.0), reverse=True) return rerank_documents[:top_n] if top_n else rerank_documents