Introduce DisableQuantizationBuilder

Author: KShivendu

examples/search.rs

@@ -1,6 +1,7 @@
 use qdrant_client::qdrant::{
-    Condition, CreateCollectionBuilder, Distance, Filter, PointStruct, ScalarQuantizationBuilder,
-    SearchParamsBuilder, SearchPointsBuilder, UpsertPointsBuilder, VectorParamsBuilder,
+    Condition, CreateCollectionBuilder, DisabledQuantizationBuilder, Distance, Filter, PointStruct,
+    ScalarQuantizationBuilder, SearchParamsBuilder, SearchPointsBuilder, UpdateCollectionBuilder,
+    UpsertPointsBuilder, VectorParamsBuilder,
 };
 use qdrant_client::{Payload, Qdrant, QdrantError};
 
@@ -81,5 +82,13 @@ async fn main() -> Result<(), QdrantError> {
     println!("baz: {}", baz_payload);
     // baz: {"qux":"quux"}
 
+    // Disable quantization:
+    client
+        .update_collection(
+            UpdateCollectionBuilder::new(collection_name)
+                .quantization_config(DisabledQuantizationBuilder::new()),
+        )
+        .await?;
+
     Ok(())
 }

src/builder_ext.rs

@@ -2,23 +2,7 @@ use std::collections::HashMap;
 
 use crate::qdrant::update_collection_cluster_setup_request::Operation;
 use crate::qdrant::{
-    shard_key, AbortShardTransferBuilder, BinaryQuantizationBuilder, ClearPayloadPointsBuilder,
-    ContextExamplePair, CountPointsBuilder, CreateAliasBuilder, CreateCollectionBuilder,
-    CreateFieldIndexCollectionBuilder, CreateShardKeyRequestBuilder, DeleteCollectionBuilder,
-    DeleteFieldIndexCollectionBuilder, DeletePayloadPointsBuilder, DeletePointVectorsBuilder,
-    DeletePointsBuilder, DeleteShardKey, DeleteShardKeyRequestBuilder,
-    DeleteSnapshotRequestBuilder, DiscoverBatchPointsBuilder, DiscoverPoints,
-    DiscoverPointsBuilder, Distance, FacetCountsBuilder, FieldType, GetPointsBuilder,
-    LookupLocationBuilder, MoveShardBuilder, PayloadExcludeSelector, PayloadIncludeSelector,
-    PointId, PointStruct, PointVectors, PointsUpdateOperation, ProductQuantizationBuilder,
-    QuantizationType, QueryBatchPointsBuilder, QueryPointGroupsBuilder, QueryPoints,
-    QueryPointsBuilder, RecommendBatchPointsBuilder, RecommendExample, RecommendPointGroupsBuilder,
-    RecommendPoints, RecommendPointsBuilder, RenameAliasBuilder, ReplicaBuilder,
-    ReplicateShardBuilder, ScalarQuantizationBuilder, ScrollPointsBuilder,
-    SearchBatchPointsBuilder, SearchMatrixPointsBuilder, SearchPointGroupsBuilder, SearchPoints,
-    SearchPointsBuilder, SetPayloadPointsBuilder, ShardKey, UpdateBatchPointsBuilder,
-    UpdateCollectionBuilder, UpdateCollectionClusterSetupRequestBuilder, UpdatePointVectorsBuilder,
-    UpsertPointsBuilder, Value, VectorParamsBuilder, VectorsSelector, WithLookupBuilder,
+    shard_key, AbortShardTransferBuilder, BinaryQuantizationBuilder, ClearPayloadPointsBuilder, ContextExamplePair, CountPointsBuilder, CreateAliasBuilder, CreateCollectionBuilder, CreateFieldIndexCollectionBuilder, CreateShardKeyRequestBuilder, DeleteCollectionBuilder, DeleteFieldIndexCollectionBuilder, DeletePayloadPointsBuilder, DeletePointVectorsBuilder, DeletePointsBuilder, DeleteShardKey, DeleteShardKeyRequestBuilder, DeleteSnapshotRequestBuilder, DisabledQuantizationBuilder, DiscoverBatchPointsBuilder, DiscoverPoints, DiscoverPointsBuilder, Distance, FacetCountsBuilder, FieldType, GetPointsBuilder, LookupLocationBuilder, MoveShardBuilder, PayloadExcludeSelector, PayloadIncludeSelector, PointId, PointStruct, PointVectors, PointsUpdateOperation, ProductQuantizationBuilder, QuantizationType, QueryBatchPointsBuilder, QueryPointGroupsBuilder, QueryPoints, QueryPointsBuilder, RecommendBatchPointsBuilder, RecommendExample, RecommendPointGroupsBuilder, RecommendPoints, RecommendPointsBuilder, RenameAliasBuilder, ReplicaBuilder, ReplicateShardBuilder, ScalarQuantizationBuilder, ScrollPointsBuilder, SearchBatchPointsBuilder, SearchMatrixPointsBuilder, SearchPointGroupsBuilder, SearchPoints, SearchPointsBuilder, SetPayloadPointsBuilder, ShardKey, UpdateBatchPointsBuilder, UpdateCollectionBuilder, UpdateCollectionClusterSetupRequestBuilder, UpdatePointVectorsBuilder, UpsertPointsBuilder, Value, VectorParamsBuilder, VectorsSelector, WithLookupBuilder
 };
 
 impl VectorParamsBuilder {
@@ -54,6 +38,13 @@ impl BinaryQuantizationBuilder {
     }
 }
 
+impl DisabledQuantizationBuilder {
+    pub fn new() -> Self {
+        let builder = Self::empty();
+        builder
+    }
+}
+
 impl SearchPointsBuilder {
     pub fn new(
         collection_name: impl Into<String>,

src/builders/disabled_quantization_builder.rs

@@ -0,0 +1,27 @@
+use crate::qdrant::*;
+
+#[derive(Clone)]
+pub struct DisabledQuantizationBuilder {}
+
+impl DisabledQuantizationBuilder {
+    pub fn empty() -> Self {
+        Self {}
+    }
+
+    fn build_inner(self) -> Disabled {
+        Disabled {}
+    }
+}
+
+impl From<DisabledQuantizationBuilder> for Disabled {
+    fn from(value: DisabledQuantizationBuilder) -> Self {
+        value.build_inner()
+    }
+}
+
+impl DisabledQuantizationBuilder {
+    /// Builds the desired type. Can often be omitted.
+    pub fn build(self) -> Disabled {
+        self.build_inner()
+    }
+}

src/builders/mod.rs

@@ -22,6 +22,9 @@ pub use wal_config_diff_builder::WalConfigDiffBuilder;
 mod optimizers_config_diff_builder;
 pub use optimizers_config_diff_builder::OptimizersConfigDiffBuilder;
 
+mod disabled_quantization_builder;
+pub use disabled_quantization_builder::DisabledQuantizationBuilder;
+
 mod scalar_quantization_builder;
 pub use scalar_quantization_builder::ScalarQuantizationBuilder;
 

src/grpc_conversions/mod.rs

@@ -11,17 +11,17 @@ use crate::qdrant::{
     update_collection_cluster_setup_request, vector_example, vectors_config, vectors_config_diff,
     with_payload_selector, with_vectors_selector, AbortShardTransfer, AbortShardTransferBuilder,
     AliasOperations, BinaryQuantization, BinaryQuantizationBuilder, Condition, CreateAlias,
-    CreateShardKey, DeleteAlias, DeleteShardKey, Disabled, FieldCondition, Filter, GeoLineString,
-    GeoPoint, GroupId, HasIdCondition, IsEmptyCondition, IsNullCondition, ListValue, Match,
-    MoveShard, MoveShardBuilder, NestedCondition, PayloadExcludeSelector, PayloadIncludeSelector,
-    PointId, PointsIdsList, PointsSelector, PointsUpdateOperation, ProductQuantization,
-    ProductQuantizationBuilder, QuantizationConfig, QuantizationConfigDiff, ReadConsistency,
-    RenameAlias, Replica, ReplicateShard, ReplicateShardBuilder, RestartTransfer,
-    ScalarQuantization, ScalarQuantizationBuilder, ShardKey, ShardKeySelector, SparseIndexConfig,
-    SparseVectorParams, StartFrom, Struct, TargetVector, Value, Vector, VectorExample,
-    VectorParams, VectorParamsBuilder, VectorParamsDiff, VectorParamsDiffBuilder,
-    VectorParamsDiffMap, VectorParamsMap, VectorsConfig, VectorsConfigDiff, VectorsSelector,
-    WithPayloadSelector, WithVectorsSelector,
+    CreateShardKey, DeleteAlias, DeleteShardKey, Disabled, DisabledQuantizationBuilder,
+    FieldCondition, Filter, GeoLineString, GeoPoint, GroupId, HasIdCondition, IsEmptyCondition,
+    IsNullCondition, ListValue, Match, MoveShard, MoveShardBuilder, NestedCondition,
+    PayloadExcludeSelector, PayloadIncludeSelector, PointId, PointsIdsList, PointsSelector,
+    PointsUpdateOperation, ProductQuantization, ProductQuantizationBuilder, QuantizationConfig,
+    QuantizationConfigDiff, ReadConsistency, RenameAlias, Replica, ReplicateShard,
+    ReplicateShardBuilder, RestartTransfer, ScalarQuantization, ScalarQuantizationBuilder,
+    ShardKey, ShardKeySelector, SparseIndexConfig, SparseVectorParams, StartFrom, Struct,
+    TargetVector, Value, Vector, VectorExample, VectorParams, VectorParamsBuilder,
+    VectorParamsDiff, VectorParamsDiffBuilder, VectorParamsDiffMap, VectorParamsMap, VectorsConfig,
+    VectorsConfigDiff, VectorsSelector, WithPayloadSelector, WithVectorsSelector,
 };
 
 impl From<Vec<PointId>> for PointsSelector {
@@ -450,6 +450,12 @@ impl From<BinaryQuantizationBuilder> for quantization_config_diff::Quantization
     }
 }
 
+impl From<DisabledQuantizationBuilder> for quantization_config_diff::Quantization {
+    fn from(value: DisabledQuantizationBuilder) -> Self {
+        Self::Disabled(value.build())
+    }
+}
+
 impl From<ScalarQuantizationBuilder> for quantization_config::Quantization {
     fn from(value: ScalarQuantizationBuilder) -> Self {
         Self::Scalar(value.build())