multilingual

semantic_search/all_search_execute.py

@@ -160,21 +160,21 @@ def handler(input_,session_id):
             print("Hybrid Rerank Search Pipeline updated: "+str(r.status_code))
         
             ######## Updating opensearch_translation_pipeline Search pipeline #######   
-            opensearch_translation_pipeline = json.loads((requests.get(host+'_search/pipeline/ml_inference_for_vector_search_and_language_translation', auth=awsauth,headers=headers)).text)
-            path = "_search/pipeline/ml_inference_for_vector_search_and_language_translation" 
-            url = host + path
-            opensearch_translation_pipeline["ml_inference_for_vector_search_and_language_translation"]["phase_results_processors"] = hybrid_search_processor
-            print(opensearch_translation_pipeline)
-            r = requests.put(url, auth=awsauth, json=opensearch_translation_pipeline["ml_inference_for_vector_search_and_language_translation"], headers=headers)
-            print("translation hybrid Search Pipeline updated: "+str(r.status_code))
+            # opensearch_translation_pipeline = json.loads((requests.get(host+'_search/pipeline/ml_inference_for_vector_search_and_language_translation', auth=awsauth,headers=headers)).text)
+            # path = "_search/pipeline/ml_inference_for_vector_search_and_language_translation" 
+            # url = host + path
+            # opensearch_translation_pipeline["ml_inference_for_vector_search_and_language_translation"]["phase_results_processors"] = hybrid_search_processor
+            # print(opensearch_translation_pipeline)
+            # r = requests.put(url, auth=awsauth, json=opensearch_translation_pipeline["ml_inference_for_vector_search_and_language_translation"], headers=headers)
+            # print("translation hybrid Search Pipeline updated: "+str(r.status_code))
 
             ######## Updating opensearch_translation_pipeline_with_rerank Search pipeline #######
-            opensearch_translation_pipeline_with_rerank = json.loads((requests.get(host+'_search/pipeline/ml_inference_for_vector_search_and_language_translation_with_rerank', auth=awsauth,headers=headers)).text)
-            path = "_search/pipeline/ml_inference_for_vector_search_and_language_translation_with_rerank" 
-            url = host + path
-            opensearch_translation_pipeline_with_rerank["ml_inference_for_vector_search_and_language_translation_with_rerank"]["phase_results_processors"] = hybrid_search_processor
-            r = requests.put(url, auth=awsauth, json=opensearch_translation_pipeline_with_rerank["ml_inference_for_vector_search_and_language_translation_with_rerank"], headers=headers)
-            print("translation hybrid rerank Search Pipeline updated: "+str(r.status_code))
+            # opensearch_translation_pipeline_with_rerank = json.loads((requests.get(host+'_search/pipeline/ml_inference_for_vector_search_and_language_translation_with_rerank', auth=awsauth,headers=headers)).text)
+            # path = "_search/pipeline/ml_inference_for_vector_search_and_language_translation_with_rerank" 
+            # url = host + path
+            # opensearch_translation_pipeline_with_rerank["ml_inference_for_vector_search_and_language_translation_with_rerank"]["phase_results_processors"] = hybrid_search_processor
+            # r = requests.put(url, auth=awsauth, json=opensearch_translation_pipeline_with_rerank["ml_inference_for_vector_search_and_language_translation_with_rerank"], headers=headers)
+            # print("translation hybrid rerank Search Pipeline updated: "+str(r.status_code))
         ######## start of Applying LLM filters ####### 
     if(st.session_state.input_rewritten_query!=""):
             filter_ = {"filter": {
@@ -482,41 +482,37 @@ def handler(input_,session_id):
                                                 }}
                 
             r = requests.get(url, auth=awsauth, json=hybrid_payload, headers=headers)
-            print("url: "+url)
-            print("payload: "+json.dumps(hybrid_payload))
             response_ = json.loads(r.text)
             docs = response_['hits']['hits']
     
     
     else:
         if( st.session_state.input_hybridType == "OpenSearch Hybrid Query"):
-            if(st.session_state.input_multilingual):
-                if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
-                    path = "demostore-search-index-reindex-new/_search?search_pipeline=ml_inference_for_vector_search_and_language_translation_with_rerank" 
-                    url = host + path
-                    hybrid_payload["ext"] = {"rerank": {
-                                                "query_context": {
-                                                    "query_text": query
-                                                }
-                                                }}
-                else:
-                    path = "demostore-search-index-reindex-new/_search?search_pipeline=ml_inference_for_vector_search_and_language_translation" 
-                    url = host + path
-            else:
-                path = "demostore-search-index-reindex-new/_search?search_pipeline=hybrid_search_pipeline" 
+            # if(st.session_state.input_multilingual):
+            #     if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
+            #         path = "demostore-search-index-reindex-new/_search?search_pipeline=ml_inference_for_vector_search_and_language_translation_with_rerank" 
+            #         url = host + path
+            #         hybrid_payload["ext"] = {"rerank": {
+            #                                     "query_context": {
+            #                                         "query_text": query
+            #                                     }
+            #                                     }}
+            #     else:
+            #         path = "demostore-search-index-reindex-new/_search?search_pipeline=ml_inference_for_vector_search_and_language_translation" 
+            #         url = host + path
+            #else:
+            path = "demostore-search-index-reindex-new/_search?search_pipeline=hybrid_search_pipeline" 
+            url = host + path
+            if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
+                
+                path = "demostore-search-index-reindex-new/_search?search_pipeline=hybrid_rerank_pipeline" 
                 url = host + path
-                if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
-                    
-                    path = "demostore-search-index-reindex-new/_search?search_pipeline=hybrid_rerank_pipeline" 
-                    url = host + path
-                    hybrid_payload["ext"] = {"rerank": {
-                                            "query_context": {
-                                                "query_text": query
-                                            }
-                                            }}
+                hybrid_payload["ext"] = {"rerank": {
+                                        "query_context": {
+                                            "query_text": query
+                                        }
+                                        }}
             r = requests.get(url, auth=awsauth, json=hybrid_payload, headers=headers)
-            print(hybrid_payload)
-            print(r.text)
             response_ = json.loads(r.text)
             docs = response_['hits']['hits']