From bcb05b56319bcc75d985e517c67df7c383178c52 Mon Sep 17 00:00:00 2001
From: George Panchuk <george.panchuk@qdrant.tech>
Date: Tue, 24 Sep 2024 18:47:20 +0200
Subject: [PATCH] fix: add sharding method to collection params conversion

---
 qdrant_client/conversions/conversion.py | 8 ++++++++
 tests/conversions/fixtures.py           | 1 +
 2 files changed, 9 insertions(+)

diff --git a/qdrant_client/conversions/conversion.py b/qdrant_client/conversions/conversion.py
index 8de442e0..0f529e99 100644
--- 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
index 8338ed6d..241eaed2 100644
--- 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(