Update app.py

app.py

@@ -95,12 +95,12 @@ llm = ChatOpenAI(temperature=0, model_name="gpt-4o-mini", max_tokens=512)
 #     base_compressor=compressor, base_retriever=retriever
 # )
 
-from langchain_community.document_compressors.rankllm_rerank import RankLLMRerank
+# from langchain_community.document_compressors.rankllm_rerank import RankLLMRerank
 
-compressor = RankLLMRerank(top_n=3, model="gpt", gpt_model="gpt-4o-mini")
-compression_retriever = ContextualCompressionRetriever(
-    base_compressor=compressor, base_retriever=retriever
-)
+# compressor = RankLLMRerank(top_n=3, model="gpt", gpt_model="gpt-4o-mini")
+# compression_retriever = ContextualCompressionRetriever(
+#     base_compressor=compressor, base_retriever=retriever
+# )
 
 # Contextualization prompt and retriever
 contextualize_q_system_prompt = """Given a chat history and the latest user question \
@@ -115,7 +115,7 @@ contextualize_q_prompt = ChatPromptTemplate.from_messages(
         ("human", "{input}")
     ]
 )
-history_aware_retriever = create_history_aware_retriever(llm, compression_retriever, contextualize_q_prompt)
+history_aware_retriever = create_history_aware_retriever(llm, retriever, contextualize_q_prompt)
 
 # QA system prompt and chain
 qa_system_prompt = """ You are a highly skilled information retrieval assistant. Use the following context to answer questions effectively.