diff --git a/app/rag/retrievers.py b/app/rag/retrievers.py index 7edf6f6..92cc8ac 100644 --- a/app/rag/retrievers.py +++ b/app/rag/retrievers.py @@ -61,7 +61,7 @@ def create_retriever(self): # 如果开启了压缩检索器,就将ES检索器的权重调整为0.3 es_compressor = LLMChainExtractor.from_llm(llm=self.llm) es_compression_retriever = ContextualCompressionRetrieverWrapper( - base_compressor=compressor, base_retriever=es_retriever + base_compressor=es_compressor, base_retriever=es_retriever ) # 开启开关就使用压缩检索器 retrievers.append(es_compression_retriever)