fix: support qdrant v8.0 on docarray

dong xiang · dong xiang · commit c10449c2b923 · 2022-10-19T16:37:08.000+08:00
diff --git a/docarray/array/storage/qdrant/backend.py b/docarray/array/storage/qdrant/backend.py
@@ -81,6 +81,7 @@ def _init_storage(
             config.collection_name = self._tmp_collection_name()
 
         self._n_dim = config.n_dim
+        self._distance = config.distance
         self._serialize_config = config.serialize_config
 
         self._client = QdrantClient(host=config.host, port=config.port)
@@ -134,17 +135,17 @@ def _initialize_qdrant_schema(self):
                 m=self._config.m,
             )
             self.client.http.collections_api.create_collection(
-                self.collection_name,
-                CreateCollection(
-                    vector_size=self.n_dim,
-                    distance=self.distance,
+                collection_name=self.collection_name,
+                create_collection=CreateCollection(
+                    vector_size=self._n_dim,
+                    distance=DISTANCES[self._distance],
                     hnsw_config=hnsw_config,
                 ),
             )
 
     def _collection_exists(self, collection_name):
-        resp = self.client.http.collections_api.get_collections()
-        collections = [collection.name for collection in resp.result.collections]
+        resp = self.client.get_collections()
+        collections = [collection.name for collection in resp.collections]
         return collection_name in collections
 
     @staticmethod
@@ -170,15 +171,14 @@ def __setstate__(self, state):
     def _get_offset2ids_meta(self) -> List[str]:
         if not self._collection_exists(self.collection_name_meta):
             return []
-        return self.client.http.points_api.get_point(
-            self.collection_name_meta, id=1
-        ).result.payload['offset2id']
+        return self.client.retrieve(collection_name=self.collection_name_meta, ids=[1])
 
     def _update_offset2ids_meta(self):
         if not self._collection_exists(self.collection_name_meta):
-            self.client.http.collections_api.create_collection(
+            self.client.recreate_collection(
                 self.collection_name_meta,
-                CreateCollection(vector_size=1, distance=Distance.COSINE),
+                vector_size=1,
+                distance=Distance.COSINE,
             )
 
         self.client.http.points_api.upsert_points(
diff --git a/docs/advanced/document-store/qdrant.md b/docs/advanced/document-store/qdrant.md
@@ -19,7 +19,7 @@ server. Create `docker-compose.yml` as follows:
 version: '3.4'
 services:
   qdrant:
-    image: qdrant/qdrant:v0.7.0
+    image: qdrant/qdrant:v0.8.0
     ports:
       - "6333:6333"
     ulimits: # Only required for tests, as there are a lot of collections created
diff --git a/scripts/docker-compose.yml b/scripts/docker-compose.yml
@@ -10,7 +10,7 @@ services:
       AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED: 'true'
       PERSISTENCE_DATA_PATH: '/var/lib/weaviate'
   qdrant:
-    image: qdrant/qdrant:v0.7.0
+    image: qdrant/qdrant:v0.8.0
     ports:
       - "41233:41233"
     ulimits: # Only required for tests, as there are a lot of collections created
diff --git a/setup.py b/setup.py
@@ -65,7 +65,7 @@
             'strawberry-graphql',
         ],
         'qdrant': [
-            'qdrant-client~=0.7.3',
+            'qdrant-client==0.8.0',
         ],
         'annlite': [
             'annlite>=0.3.12',
diff --git a/tests/unit/array/docker-compose.yml b/tests/unit/array/docker-compose.yml
@@ -10,7 +10,7 @@ services:
       AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED: 'true'
       PERSISTENCE_DATA_PATH: '/var/lib/weaviate'
   qdrant:
-    image: qdrant/qdrant:v0.7.0
+    image: qdrant/qdrant:v0.8.0
     ports:
       - "6333:6333"
     ulimits: # Only required for tests, as there are a lot of collections created
