elasticsearch url in from_documents

weni-ai · May 20, 2024 · 4785bac · 4785bac
1 parent d54846c
commit 4785bac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/app/store/elasticsearch_vector_store.py b/app/store/elasticsearch_vector_store.py
@@ -60,6 +60,7 @@ def save(self, docs: list[Document])-> list[str]:
         res = self.vectorstore.from_documents(
             docs,
             self.vectorstore.embeddings,
+            es_url=os.environ.get("ELASTICSEARCH_URL"),
             index_name=index,
             bulk_kwargs={
                 "chunk_size": os.environ.get("DEFAULT_CHUNK_SIZE", 75),