From f6293a93240d599c3c0a04137471780d3a38e3ec Mon Sep 17 00:00:00 2001 From: Deadwalk Date: Fri, 13 Sep 2024 23:52:26 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=B8=80=E4=B8=AA=E6=BD=9C?= =?UTF-8?q?=E5=9C=A8=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/rag/retrievers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)