From d2f39cbdd3af3227b980ae15b097b0ba9c6b555b Mon Sep 17 00:00:00 2001 From: Alisson Date: Mon, 12 Aug 2024 09:53:40 -0300 Subject: [PATCH] check if index exists --- app/store/elasticsearch_vector_store.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/app/store/elasticsearch_vector_store.py b/app/store/elasticsearch_vector_store.py index cafa1fa..113cae6 100644 --- a/app/store/elasticsearch_vector_store.py +++ b/app/store/elasticsearch_vector_store.py @@ -60,7 +60,13 @@ class ContentBaseElasticsearchVectorStoreIndex(ElasticsearchVectorStoreIndex): def save(self, docs: list[Document]) -> list[str]: index = os.environ.get("INDEX_CONTENTBASES_NAME", "content_bases") - res = self.vectorstore.from_documents( + + if self.vectorstore.client.indices.exists(index=index): + index_documents = self.vectorstore.add_documents + else: + index_documents = self.vectorstore.from_documents + + res = index_documents( docs, self.vectorstore.embeddings, es_url=os.environ.get("ELASTICSEARCH_URL"),