32 lines
966 B
Python
32 lines
966 B
Python
|
from haystack.document_stores.elasticsearch import ElasticsearchDocumentStore
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
sys.path.append('/root/home/BA_QA_HSMA/backendd')
|
||
|
|
|
||
|
|
class ElasticSearchData:
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.doc_store = ElasticsearchDocumentStore(port=9210)
|
||
|
|
pass
|
||
|
|
|
||
|
|
def write_data(self, data, index):
|
||
|
|
return self.doc_store.write_documents(documents=data, index=index)
|
||
|
|
|
||
|
|
def get_data(self, index):
|
||
|
|
return self.doc_store.get_all_documents(index=index)
|
||
|
|
|
||
|
|
def get_all_doc_count(self,index):
|
||
|
|
return self.doc_store.get_document_count(index=index)
|
||
|
|
|
||
|
|
def delete_all_docs(self, index):
|
||
|
|
return self.doc_store.delete_all_documents(index=index)
|
||
|
|
|
||
|
|
def remove_docs(self):
|
||
|
|
pass
|
||
|
|
def update_embeddings(self, retriever):
|
||
|
|
return self.doc_store.update_embeddings(retriever=retriever, index="ib")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
es = ElasticSearchData()
|
||
|
|
print(es.delete_all_docs(index="ib"))
|