Integrate VectorStore from Elasticsearch client

run-llama · May 13, 2024 · e1ef2ea · e1ef2ea
1 parent 48e0a26
commit e1ef2ea
Show file tree

Hide file tree

Showing 8 changed files with 375 additions and 419 deletions.
diff --git a/llama-index-finetuning/tests/test_base.py b/llama-index-finetuning/tests/test_base.py
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-elasticsearch/Makefile b/llama-index-integrations/vector_stores/llama-index-vector-stores-elasticsearch/Makefile
@@ -11,7 +11,7 @@ lint:	## Run linters: pre-commit (black, ruff, codespell) and mypy
 	pre-commit install && git ls-files | xargs pre-commit run --show-diff-on-failure --files
 
 test:	## Run tests via pytest.
-	pytest tests
+	poetry run pytest tests
 
 watch-docs:	## Build and watch documentation.
 	sphinx-autobuild docs/ docs/_build/html --open-browser --watch $(GIT_ROOT)/llama_index/
diff --git a/...ama-index-vector-stores-elasticsearch/llama_index/vector_stores/elasticsearch/__init__.py b/...ama-index-vector-stores-elasticsearch/llama_index/vector_stores/elasticsearch/__init__.py
@@ -1,3 +1,16 @@
 from llama_index.vector_stores.elasticsearch.base import ElasticsearchStore
 
-__all__ = ["ElasticsearchStore"]
+from elasticsearch.helpers.vectorstore import (
+    AsyncBM25Strategy,
+    AsyncSparseVectorStrategy,
+    AsyncDenseVectorStrategy,
+    AsyncRetrievalStrategy,
+)
+
+__all__ = [
+    "AsyncBM25Strategy",
+    "AsyncDenseVectorStrategy",
+    "AsyncRetrievalStrategy",
+    "AsyncSparseVectorStrategy",
+    "ElasticsearchStore",
+]