rerank model

pages/Semantic_Search.py

@@ -242,7 +242,7 @@ if "KendraResourcePlanID" not in st.session_state:
     st.session_state.KendraResourcePlanID= ""
 
 if "SAGEMAKER_CrossEncoder_MODEL_ID" not in st.session_state:
-    st.session_state.SAGEMAKER_CrossEncoder_MODEL_ID = "eUoo-ZMBTp0efWqBQ-5g" 
+    st.session_state.SAGEMAKER_CrossEncoder_MODEL_ID = "deBS3pYB5VHEj-qVuPHT" 
     
     
 if "SAGEMAKER_SPARSE_MODEL_ID" not in st.session_state:
@@ -910,7 +910,7 @@ if(search_all_type == True or 1==1):
         if(st.session_state.re_ranker == "true"):
             st.subheader(':blue[Re-ranking]')
             reranker = st.selectbox('Choose a Re-Ranker',
-            ('None','Cross Encoder'#'Kendra Rescore'
+            ('None','Cohere Rerank'#'Kendra Rescore'
 
             ),
 

semantic_search/all_search_execute.py

@@ -136,8 +136,6 @@ def handler(input_,session_id):
 
         
     opensearch_search_pipeline = (requests.get(host+'_search/pipeline/hybrid_search_pipeline', auth=awsauth,headers=headers)).text
-    print("opensearch_search_pipeline")
-    print(opensearch_search_pipeline)
     if(opensearch_search_pipeline!='{}'):
         path = "_search/pipeline/hybrid_search_pipeline" 
         url = host + path
@@ -145,11 +143,6 @@ def handler(input_,session_id):
         print("Hybrid Search Pipeline updated: "+str(r.status_code))
         ######## Combining hybrid+rerank pipeline ####### 
         opensearch_rerank_pipeline = (requests.get(host+'_search/pipeline/rerank_pipeline', auth=awsauth,headers=headers)).text
-        print("opensearch_rerank_pipeline")
-        print(opensearch_rerank_pipeline)
-        
-            
-    
     ######## start of Applying LLM filters ####### 
     if(st.session_state.input_rewritten_query!=""):
             filter_ = {"filter": {
@@ -419,7 +412,7 @@ def handler(input_,session_id):
         single_query = hybrid_payload["query"]["hybrid"]["queries"][0]
         del hybrid_payload["query"]["hybrid"]
         hybrid_payload["query"] = single_query
-        if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cross Encoder'):
+        if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
             path = "demostore-search-index/_search?search_pipeline=rerank_pipeline" 
             url = host + path
             hybrid_payload["ext"] = {"rerank": {
@@ -439,7 +432,7 @@ def handler(input_,session_id):
         if( st.session_state.input_hybridType == "OpenSearch Hybrid Query"):
             url_ = url + "?search_pipeline=hybrid_search_pipeline" 
             
-            if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cross Encoder'):
+            if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
                 
                 url_ = url + "?search_pipeline=hybrid_rerank_pipeline"