fix: add sharding method to collection params conversion

qdrant · Sep 24, 2024 · bcb05b5 · bcb05b5
1 parent 8ad5574
commit bcb05b5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/qdrant_client/conversions/conversion.py b/qdrant_client/conversions/conversion.py
@@ -635,6 +635,9 @@ def convert_collection_params(cls, model: grpc.CollectionParams) -> rest.Collect
             sparse_vectors=cls.convert_sparse_vector_config(model.sparse_vectors_config)
             if model.HasField("sparse_vectors_config")
             else None,
+            sharding_method=cls.convert_sharding_method(model.sharding_method)
+            if model.HasField("sharding_method")
+            else None,
         )
 
     @classmethod
@@ -2153,6 +2156,11 @@ def convert_collection_params(cls, model: rest.CollectionParams) -> grpc.Collect
                 if model.sparse_vectors is not None
                 else None
             ),
+            sharding_method=(
+                cls.convert_sharding_method(model.sharding_method)
+                if model.sharding_method is not None
+                else None
+            ),
         )
 
     @classmethod

diff --git a/tests/conversions/fixtures.py b/tests/conversions/fixtures.py
@@ -218,6 +218,7 @@
     vectors_config=single_vector_config,
     shard_number=10,
     on_disk_payload=True,
+    sharding_method=grpc.ShardingMethod.Custom,
 )
 
 collection_params_2 = grpc.CollectionParams(