sort matched documents by similarity after merge with mongodb values (#…

…106) * sort matched documents by similarity after merge with mongodb values
jerpint · Jun 9, 2023 · 3d1bd0e · 3d1bd0e
1 parent fc97df5
commit 3d1bd0e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/buster/retriever/service.py b/buster/retriever/service.py
@@ -92,4 +92,7 @@ def get_topk_documents(self, query: str, source: str, top_k: int) -> pd.DataFram
         matched_documents["similarity"] = matched_documents["_id"].apply(lambda x: matching_scores[str(x)])
         matched_documents["embedding"] = matched_documents["_id"].apply(lambda x: matching_embeddings[str(x)])
 
+        # sort by similarity
+        matched_documents = matched_documents.sort_values(by="similarity", ascending=False, ignore_index=True)
+
         return matched_documents