Skip to content

Commit ff00b60

Browse files
tonyyangatonyyanga
authored
feat(index): add epsilla connector (docarray#1835)
Signed-off-by: Tony Yang <[email protected]>
1 parent 21e107b commit ff00b60

File tree

14 files changed

+1292
-6
lines changed

14 files changed

+1292
-6
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, hnswlib, qdrant, weaviate, redis, milvus]
196+
db_test_folder: [base_classes, elastic, epsilla, hnswlib, qdrant, weaviate, redis, milvus]
197197
pydantic-version: ["pydantic-v2", "pydantic-v1"]
198198
steps:
199199
- uses: actions/[email protected]

‎docarray/index/__init__.py‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,18 @@
1010
if TYPE_CHECKING:
1111
from docarray.index.backends.elastic import ElasticDocIndex # noqa: F401
1212
from docarray.index.backends.elasticv7 import ElasticV7DocIndex # noqa: F401
13+
from docarray.index.backends.epsilla import EpsillaDocumentIndex # noqa: F401
1314
from docarray.index.backends.hnswlib import HnswDocumentIndex # noqa: F401
15+
from docarray.index.backends.milvus import MilvusDocumentIndex # noqa: F401
1416
from docarray.index.backends.qdrant import QdrantDocumentIndex # noqa: F401
15-
from docarray.index.backends.weaviate import WeaviateDocumentIndex # noqa: F401
1617
from docarray.index.backends.redis import RedisDocumentIndex # noqa: F401
17-
from docarray.index.backends.milvus import MilvusDocumentIndex # noqa: F401
18+
from docarray.index.backends.weaviate import WeaviateDocumentIndex # noqa: F401
1819

1920
__all__ = [
2021
'InMemoryExactNNIndex',
2122
'ElasticDocIndex',
2223
'ElasticV7DocIndex',
24+
'EpsillaDocumentIndex',
2325
'QdrantDocumentIndex',
2426
'WeaviateDocumentIndex',
2527
'RedisDocumentIndex',
@@ -38,6 +40,9 @@ def __getattr__(name: str):
3840
elif name == 'ElasticV7DocIndex':
3941
import_library('elasticsearch', raise_error=True)
4042
import docarray.index.backends.elasticv7 as lib
43+
elif name == 'EpsillaDocumentIndex':
44+
import_library('pyepsilla', raise_error=True)
45+
import docarray.index.backends.epsilla as lib
4146
elif name == 'QdrantDocumentIndex':
4247
import_library('qdrant_client', raise_error=True)
4348
import docarray.index.backends.qdrant as lib

0 commit comments

Comments
 (0)