Skip to content

Commit 45cf138

Browse files
committed
Make vectorstore conditional
1 parent 477ef42 commit 45cf138

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

singlestoredb/__init__.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,18 @@
3131
Date, Time, Timestamp, DateFromTicks, TimeFromTicks, TimestampFromTicks,
3232
Binary, STRING, BINARY, NUMBER, DATETIME, ROWID,
3333
)
34-
from .vectorstore import (
35-
vector_db, IndexInterface, IndexList, IndexModel, MatchTypedDict,
36-
Metric, IndexStatsTypedDict, NamespaceStatsTypedDict, Vector,
37-
VectorDictMetadataValue, VectorMetadataTypedDict, VectorTuple,
38-
VectorTupleWithMetadata, DeletionProtection, AndFilter, EqFilter,
39-
ExactMatchFilter, FilterTypedDict, GteFilter, GtFilter, InFilter,
40-
LteFilter, LtFilter, NeFilter, NinFilter, OrFilter, SimpleFilter,
41-
)
34+
# These are only loaded if the singlestore-vectorstore package is available
35+
try:
36+
from .vectorstore import (
37+
vector_db, IndexInterface, IndexList, IndexModel, MatchTypedDict,
38+
Metric, IndexStatsTypedDict, NamespaceStatsTypedDict, Vector,
39+
VectorDictMetadataValue, VectorMetadataTypedDict, VectorTuple,
40+
VectorTupleWithMetadata, DeletionProtection, AndFilter, EqFilter,
41+
ExactMatchFilter, FilterTypedDict, GteFilter, GtFilter, InFilter,
42+
LteFilter, LtFilter, NeFilter, NinFilter, OrFilter, SimpleFilter,
43+
)
44+
except (ImportError, ModuleNotFoundError):
45+
pass
4246

4347

4448
#

0 commit comments

Comments
 (0)