集成到检索系统

📅 2026/8/21 9:50:01
集成到检索系统
classHybridSearchWithRerank:def__init__(self,embedding_model):self.vector_retrieverVectorRetriever(embedding_model)self.keyword_retrieverBM25Retriever()self.fusion_rerankerDocumentFusionReranker()defsearch(self,query,top_k10):# 并行检索vector_resultsself.vector_retriever.search(query,top_k50)keyword_resultsself.keyword_retriever.search(query,top_k50)# 融合与重排final_resultsself.fusion_reranker.fuse_and_rerank(query,vector_results,keyword_results,top_ktop_k)returnfinal_results