Skip to content

Commit 4124812

Browse files
author
Jay Bhambhani
committed
feat: adding opensearch integration
rename files and make some other fixes missed the pytest marks pytest marker one more spot? test undo docker for local testing fixes and cleanup Signed-off-by: Jay Bhambhani <[email protected]> fixup pyproject.toml Signed-off-by: Jay Bhambhani <[email protected]>
1 parent 83ebef6 commit 4124812

File tree

11 files changed

+2218
-4
lines changed

11 files changed

+2218
-4
lines changed

‎.github/workflows/ci.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ jobs:
193193
fail-fast: false
194194
matrix:
195195
python-version: [3.8]
196-
db_test_folder: [base_classes, elastic, epsilla, hnswlib, qdrant, weaviate, redis, milvus]
196+
db_test_folder: [base_classes, elastic, epsilla, hnswlib, qdrant, weaviate, redis, milvus, opensearch]
197197
pydantic-version: ["pydantic-v2", "pydantic-v1"]
198198
steps:
199199
- uses: actions/[email protected]

‎docarray/index/__init__.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from docarray.index.backends.mongodb_atlas import ( # noqa: F401
1717
MongoDBAtlasDocumentIndex,
1818
)
19+
from docarray.index.backends.opensearchv2 import OpenSearchDocIndex # noqa: F401
1920
from docarray.index.backends.qdrant import QdrantDocumentIndex # noqa: F401
2021
from docarray.index.backends.redis import RedisDocumentIndex # noqa: F401
2122
from docarray.index.backends.weaviate import WeaviateDocumentIndex # noqa: F401
@@ -30,6 +31,7 @@
3031
'RedisDocumentIndex',
3132
'MilvusDocumentIndex',
3233
'MongoDBAtlasDocumentIndex',
34+
'OpenSearchDocIndex',
3335
]
3436

3537

@@ -41,6 +43,9 @@ def __getattr__(name: str):
4143
elif name == 'ElasticDocIndex':
4244
import_library('elasticsearch', raise_error=True)
4345
import docarray.index.backends.elastic as lib
46+
elif name == 'OpenSearchDocIndex':
47+
import_library('elasticsearch', raise_error=True)
48+
import docarray.index.backends.opensearchv2 as lib
4449
elif name == 'ElasticV7DocIndex':
4550
import_library('elasticsearch', raise_error=True)
4651
import docarray.index.backends.elasticv7 as lib

0 commit comments

Comments
 (0)