修复一个潜在的bug

domonic18 · Sep 13, 2024 · f6293a9 · f6293a9
1 parent 9a88fa3
commit f6293a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)