Update app.py

app.py

@@ -23,6 +23,7 @@ from langchain_community.chat_models import ChatPerplexity
 from langchain.retrievers.document_compressors import CrossEncoderReranker
 from langchain_community.cross_encoders import HuggingFaceCrossEncoder
 from langchain_core.prompts import PromptTemplate
+from langchain.retrievers.document_compressors import FlashrankRerank
 import re
 
 # Load environment variables
@@ -91,9 +92,14 @@ llm = ChatPerplexity(temperature=0, pplx_api_key=GROQ_API_KEY, model="llama-3.1-
 # compression_retriever = ContextualCompressionRetriever(
 #     base_compressor=compressor, base_retriever=retriever
 # )
-from langchain.retrievers.document_compressors import LLMChainExtractor
+# from langchain.retrievers.document_compressors import LLMChainExtractor
 
-compressor = LLMChainExtractor.from_llm(llm)
+# compressor = LLMChainExtractor.from_llm(llm)
+# compression_retriever = ContextualCompressionRetriever(
+#     base_compressor=compressor, base_retriever=retriever
+# )
+
+compressor = FlashrankRerank()
 compression_retriever = ContextualCompressionRetriever(
     base_compressor=compressor, base_retriever=retriever
 )