fix weight rerank of knowledge retrieval

langgenius · Nov 21, 2024 · 2b68d92 · 2b68d92
1 parent 94c9cad
commit 2b68d92
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/api/core/rag/rerank/weight_rerank.py b/api/core/rag/rerank/weight_rerank.py
@@ -39,9 +39,8 @@ def run(
         unique_documents = []
         doc_id = set()
         for document in documents:
-            doc_id = document.metadata.get("doc_id")
-            if doc_id not in doc_id:
-                doc_id.add(doc_id)
+            if document.metadata["doc_id"] not in doc_id:
+                doc_id.add(document.metadata["doc_id"])
                 unique_documents.append(document)
 
         documents = unique_documents