cluster fix

app.py

@@ -21,7 +21,7 @@ def fake_search(question, retriever_top_k, reader_top_k):
   return result
 
 def search(question, retriever_top_k, reader_top_k):
-  query_result = extractive_query.search_by_query(query = question, retriever_top_k = retriever_top_k, reader_top_k = reader_top_k, es_index = "informecomisionverdad")
+  query_result = extractive_query.search_by_query(query = question, retriever_top_k = retriever_top_k, reader_top_k = reader_top_k)
   result = []
   for i in range(0, len(query_result)):
     item = query_result[i]

config.py

@@ -1,7 +1,7 @@
 class Config():
-    es_host = "3pvrzh9tl:4yl4vk9ijr@saimon-kb-5550035374.us-east-1.bonsaisearch.net"
-    es_user = "3pvrzh9tl"
-    es_password = "4yl4vk9ijr"
+    es_host = "saimon-askwdemocracy.es.us-central1.gcp.cloud.es.io"
+    es_user = "elastic"
+    es_password = "NY5Gqg83peXwagrfolBFUhhy"
     proposals_index = "informecomisionverdad"
     reader_model_name_or_path = "deepset/xlm-roberta-base-squad2-distilled"
     #reader_model_name_or_path = "deepset/xlm-roberta-base-squad2"

document_quieries.py

@@ -34,8 +34,8 @@ class ExtractiveProposalQueries(DocumentQueries):
 
     def search_by_query(self, query : str, retriever_top_k: int, reader_top_k: int, es_index: str = None) :
         self.log.write_log(query, "hfspace-informecomision")
-        if es_index is not None:
-            self._initialize_pipeline(self.es_host, es_index, self.es_user, self.es_password)
+        #if es_index is not None:
+            #self._initialize_pipeline(self.es_host, es_index, self.es_user, self.es_password)
         params = {"Retriever": {"top_k": retriever_top_k}, "Reader": {"top_k": reader_top_k}}
         prediction = self.pipe.run( query = query, params = params)      
         return prediction["answers"]
@@ -44,7 +44,7 @@ class Log():
     
     def __init__(self, es_host: str, es_index: str, es_user, es_password) -> None:
         self.elastic_endpoint = f"https://{es_host}:443/{es_index}/_doc"
-        self.credentials = b64encode(b"elastic:Xmir2X1f5twb8OJbpjbiIgXt").decode("ascii")
+        self.credentials = b64encode(b"3pvrzh9tl:4yl4vk9ijr").decode("ascii")
         self.auth_header = { 'Authorization' : 'Basic %s' %  self.credentials }
                       
     def write_log(self, message: str, source: str) -> None: