superduper-qdrant 0.7.0__tar.gz → 0.9.1__tar.gz

{superduper_qdrant-0.7.0 → superduper_qdrant-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superduper_qdrant
-Version: 0.7.0
+Version: 0.9.1
 Summary: SuperDuper Lance is a Python library that provides a high-level API for working with Lance vector search database.
 Maintainer-email: "superduper.io, Inc." <opensource@superduper.io>
 License: 

{superduper_qdrant-0.7.0 → superduper_qdrant-0.9.1}/superduper_qdrant/__init__.py

@@ -1,5 +1,5 @@
 from .qdrant import QdrantVectorSearcher as VectorSearcher
 
-__version__ = "0.7.0"
+__version__ = "0.9.1"
 
 __all__ = ['VectorSearcher']

{superduper_qdrant-0.7.0 → superduper_qdrant-0.9.1}/superduper_qdrant/qdrant.py

@@ -28,31 +28,48 @@ class QdrantVectorSearcher(BaseVectorSearcher):
 
     def __init__(
         self,
-        uuid: str,
+        identifier: str,
         dimensions: int,
         measure: t.Optional[str] = None,
+        component: str = 'VectorIndex',
     ):
         config_dict = deepcopy(CFG.vector_search_kwargs)
-        self.vector_name: t.Optional[str] = config_dict.pop("vector_name", None)
+        try:
+            plugin, uri = CFG.vector_search_engine.split("://")
+            if not uri.startswith("http"):
+                uri = f"http://{uri}"
+            if uri:
+                config_dict['location'] = uri
+        except ValueError as e:
+            if 'not enough values to unpack' in str(e):
+                plugin = CFG.vector_search_engine
+            else:
+                raise e
+
+        assert (
+            plugin == "qdrant"
+        ), "Only 'qdrant' vector search engine is supported in QdrantVectorSearcher."
+
         # Use an in-memory instance by default
         # https://github.com/qdrant/qdrant-client#local-mode
         config_dict = config_dict or {"location": ":memory:"}
         self.client = QdrantClient(**config_dict)
-        self.collection_name = uuid
+        self.identifier = identifier
         self.measure = measure
 
-        self.collection_name = re.sub("\W+", "", uuid)
-        if not self.client.collection_exists(self.collection_name):
+        self.identifier = re.sub("\W+", "", identifier)
+        if not self.client.collection_exists(self.identifier):
             measure = (
                 measure.name if isinstance(measure, VectorIndexMeasureType) else measure
             )
             distance = self._distance_mapping(measure)
             self.client.create_collection(
-                collection_name=self.collection_name,
+                collection_name=self.identifier,
                 vectors_config=models.VectorParams(size=dimensions, distance=distance),
             )
+        self.component = component
 
-    def initialize(self, db):
+    def initialize(self):
         """Initialize the vector index.
 
         :param db: Datalayer instance
@@ -60,7 +77,7 @@ class QdrantVectorSearcher(BaseVectorSearcher):
         pass
 
     def __len__(self):
-        return self.client.get_collection(self.collection_name).vectors_count
+        return self.client.get_collection(self.identifier).vectors_count
 
     def _create_collection(self):
         measure = (
@@ -70,7 +87,7 @@ class QdrantVectorSearcher(BaseVectorSearcher):
         )
         distance = self._distance_mapping(measure)
         self.client.create_collection(
-            collection_name=self.collection_name,
+            collection_name=self.identifier,
             vectors_config=models.VectorParams(size=self.dimensions, distance=distance),
         )
 
@@ -82,22 +99,22 @@ class QdrantVectorSearcher(BaseVectorSearcher):
         """
         points = []
         for item in items:
+            if hasattr(item.vector, "tolist"):
+                vector = item.vector.tolist()
+            else:
+                vector = item.vector
             point = models.PointStruct(
                 id=self._convert_id(item.id),
-                vector=(
-                    {self.vector_name: item.vector.tolist()}
-                    if self.vector_name
-                    else item.vector.tolist()
-                ),
+                vector=vector,
                 payload={ID_PAYLOAD_KEY: item.id},
             )
             points.append(point)
-        self.client.upsert(collection_name=self.collection_name, points=points)
+        self.client.upsert(collection_name=self.identifier, points=points)
 
     def drop(self):
         """Drop the vector index."""
-        if self.client.collection_exists(self.collection_name):
-            self.client.delete_collection(self.collection_name)
+        if self.client.collection_exists(self.identifier):
+            self.client.delete_collection(self.identifier)
 
     def delete(self, ids: t.Sequence[str]) -> None:
         """Delete vectors from the index.
@@ -105,7 +122,7 @@ class QdrantVectorSearcher(BaseVectorSearcher):
         :param ids: List of IDs to delete
         """
         self.client.delete(
-            collection_name=self.collection_name,
+            collection_name=self.identifier,
             points_selector=models.Filter(
                 must=[
                     models.FieldCondition(
@@ -160,12 +177,12 @@ class QdrantVectorSearcher(BaseVectorSearcher):
             )
 
         search_result = self.client.query_points(
-            collection_name=self.collection_name,
+            collection_name=self.identifier,
             query=query,
             limit=n,
             query_filter=query_filter,
             with_payload=[ID_PAYLOAD_KEY],
-            using=self.vector_name,
+            using=None,
         ).points
 
         ids = [hit.payload[ID_PAYLOAD_KEY] for hit in search_result if hit.payload]

{superduper_qdrant-0.7.0 → superduper_qdrant-0.9.1}/superduper_qdrant.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superduper_qdrant
-Version: 0.7.0
+Version: 0.9.1
 Summary: SuperDuper Lance is a Python library that provides a high-level API for working with Lance vector search database.
 Maintainer-email: "superduper.io, Inc." <opensource@superduper.io>
 License: