Commit 582deb2e by tinywell

chore: Update GetSimilarity class to use compressed documents for reranking

parent d5d48338
...@@ -9,7 +9,7 @@ class GetSimilarity: ...@@ -9,7 +9,7 @@ class GetSimilarity:
self.similarity_doc_txt = self.faiss_db.join_document(self.similarity_docs) self.similarity_doc_txt = self.faiss_db.join_document(self.similarity_docs)
def get_rerank(self,reranker:BgeRerank ,top_k = 5): def get_rerank(self,reranker:BgeRerank ,top_k = 5):
rerank_docs = reranker.bge_rerank(self.question,self.similarity_docs) rerank_docs = reranker.compress_documents(self.similarity_docs,self.question)
self.rerank_docs = rerank_docs self.rerank_docs = rerank_docs
return self.faiss_db.join_document([d[1] for d in rerank_docs[:top_k]]) return self.faiss_db.join_document([d[1] for d in rerank_docs[:top_k]])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment