search pipeline updated

RAG/rag_DocumentSearcher.py

@@ -310,11 +310,11 @@ def query_(awsauth,inputs, session_id,search_types):
     
     #print("re-rank")
     
-    if(st.session_state.input_is_rerank == True and len(total_context)):
-        ques = [{"question":question}]
-        ans = [{"answer":total_context}]
+    # if(st.session_state.input_is_rerank == True and len(total_context)):
+    #     ques = [{"question":question}]
+    #     ans = [{"answer":total_context}]
         
-        total_context = re_ranker.re_rank('rag','Cross Encoder',"",ques, ans)
+    #     total_context = re_ranker.re_rank('rag','Cross Encoder',"",ques, ans)
 
     llm_prompt = prompt_template.format(context=total_context[0],question=question)
     output = invoke_models.invoke_llm_model( "\n\nHuman: {input}\n\nAssistant:".format(input=llm_prompt) ,False)