new: add read consistency to count points

Author: joein

qdrant_client/async_qdrant_remote.py

@@ -1525,13 +1525,16 @@ async def count(
         exact: bool = True,
         shard_key_selector: Optional[types.ShardKeySelector] = None,
         timeout: Optional[int] = None,
+        consistency: Optional[types.ReadConsistency] = None,
         **kwargs: Any,
     ) -> types.CountResult:
         if self._prefer_grpc:
             if isinstance(count_filter, models.Filter):
                 count_filter = RestToGrpc.convert_filter(model=count_filter)
             if isinstance(shard_key_selector, get_args_subscribed(models.ShardKeySelector)):
                 shard_key_selector = RestToGrpc.convert_shard_key_selector(shard_key_selector)
+            if isinstance(consistency, get_args_subscribed(models.ReadConsistency)):
+                consistency = RestToGrpc.convert_read_consistency(consistency)
             response = (
                 await self.grpc_points.Count(
                     grpc.CountPoints(
@@ -1540,6 +1543,7 @@ async def count(
                         exact=exact,
                         shard_key_selector=shard_key_selector,
                         timeout=timeout,
+                        read_consistency=consistency,
                     ),
                     timeout=timeout if timeout is not None else self._timeout,
                 )
@@ -1553,6 +1557,7 @@ async def count(
                 count_request=models.CountRequest(
                     filter=count_filter, exact=exact, shard_key=shard_key_selector
                 ),
+                consistency=consistency,
                 timeout=timeout,
             )
         ).result

qdrant_client/qdrant_remote.py

@@ -1742,6 +1742,7 @@ def count(
         exact: bool = True,
         shard_key_selector: Optional[types.ShardKeySelector] = None,
         timeout: Optional[int] = None,
+        consistency: Optional[types.ReadConsistency] = None,
         **kwargs: Any,
     ) -> types.CountResult:
         if self._prefer_grpc:
@@ -1751,13 +1752,17 @@ def count(
             if isinstance(shard_key_selector, get_args_subscribed(models.ShardKeySelector)):
                 shard_key_selector = RestToGrpc.convert_shard_key_selector(shard_key_selector)
 
+            if isinstance(consistency, get_args_subscribed(models.ReadConsistency)):
+                consistency = RestToGrpc.convert_read_consistency(consistency)
+
             response = self.grpc_points.Count(
                 grpc.CountPoints(
                     collection_name=collection_name,
                     filter=count_filter,
                     exact=exact,
                     shard_key_selector=shard_key_selector,
                     timeout=timeout,
+                    read_consistency=consistency,
                 ),
                 timeout=timeout if timeout is not None else self._timeout,
             ).result
@@ -1773,6 +1778,7 @@ def count(
                 exact=exact,
                 shard_key=shard_key_selector,
             ),
+            consistency=consistency,
             timeout=timeout,
         ).result
         assert count_result is not None, "Count points returned None result"