Updated With Base Retriever

app/spark.py

@@ -47,10 +47,10 @@ def init():
     index_name=index_name, embedding=embeddings
     )
     retriever = docsearch.as_retriever(search_kwargs={"k": 4})
-    compressor = CohereRerank()
-    reranker = ContextualCompressionRetriever(
-        base_compressor=compressor, base_retriever=retriever
-    )
+    # compressor = CohereRerank()
+    # reranker = ContextualCompressionRetriever(
+    #     base_compressor=compressor, base_retriever=retriever
+    # )
     messages = [SystemMessagePromptTemplate.from_template(spark)]
     # print('mem', user_session.get('memory'))
     messages.append(HumanMessagePromptTemplate.from_template("{question}"))
@@ -60,7 +60,7 @@ def init():
     doc_chain = load_qa_with_sources_chain(llm, chain_type="stuff", verbose=True,prompt=prompt)
 
     chain = ConversationalRetrievalChain(
-            retriever=reranker,
+            retriever=retriever,
             question_generator=question_generator,
             combine_docs_chain=doc_chain,
             verbose=True,