make UpdateCollectionClusterSetupRequest.operation required

src/builder_ext.rs

@@ -1,5 +1,6 @@
 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,
@@ -306,9 +307,10 @@ impl DeleteFieldIndexCollectionBuilder {
 }
 
 impl UpdateCollectionClusterSetupRequestBuilder {
-    pub fn new(collection_name: impl Into<String>) -> Self {
+    pub fn new(collection_name: impl Into<String>, operation: impl Into<Operation>) -> Self {
         let mut builder = Self::empty();
         builder.collection_name = Some(collection_name.into());
+        builder.operation = Some(Some(operation.into()));
         builder
     }
 }

src/qdrant.rs

@@ -1158,7 +1158,7 @@ pub struct UpdateCollectionClusterSetupRequest {
         oneof = "update_collection_cluster_setup_request::Operation",
         tags = "2, 3, 4, 5, 7, 8, 9"
     )]
-    #[builder(default, setter(into, strip_option), field(vis = "pub(crate)"))]
+    #[builder(field(vis = "pub(crate)"))]
     pub operation: ::core::option::Option<
         update_collection_cluster_setup_request::Operation,
     >,

src/qdrant_client/collection.rs

@@ -399,15 +399,16 @@ impl Qdrant {
     ///# async fn update_collection_cluster_setup(client: &Qdrant)
     ///# -> Result<(), QdrantError> {
     /// client
-    ///     .update_collection_cluster_setup(
-    ///         UpdateCollectionClusterSetupRequestBuilder::new("my_collection")
-    ///             .operation(MoveShardBuilder::new(
-    ///                 0, // Shard ID
-    ///                 0, // From peer ID
-    ///                 1, // To peer ID
-    ///             ))
-    ///     )
-    ///     .await?;
+    ///     .update_collection_cluster_setup(UpdateCollectionClusterSetupRequestBuilder::new(
+    ///         "my_collection",
+    ///         MoveShardBuilder::new(
+    ///             0, // Shard ID
+    ///             0, // From peer ID
+    ///             1, // To peer ID
+    ///         )
+    ///         .build(),
+    ///     ))
+    ///     .await
     ///# Ok(())
     ///# }
     /// ```

tests/builder_coverage.rs

@@ -1,5 +1,6 @@
 use std::collections::HashMap;
 
+use qdrant_client::qdrant::update_collection_cluster_setup_request::Operation;
 use qdrant_client::qdrant::{
     AbortShardTransferBuilder, BinaryQuantizationBuilder, ClearPayloadPointsBuilder,
     ContextInputBuilder, ContextInputPairBuilder, CountPointsBuilder, CreateAliasBuilder,
@@ -58,7 +59,11 @@ fn builder_coverage() {
     UpsertPointsBuilder::new("my_collection", []).build();
     CreateFieldIndexCollectionBuilder::new("my_collection", " myfield", FieldType::Integer).build();
     DeleteFieldIndexCollectionBuilder::new("my_collection", " myfield").build();
-    UpdateCollectionClusterSetupRequestBuilder::new("my_collection").build();
+    UpdateCollectionClusterSetupRequestBuilder::new(
+        "my_collection",
+        Operation::MoveShard(MoveShardBuilder::new(0, 0, 0).build()),
+    )
+    .build();
     MoveShardBuilder::new(0, 0, 0).build();
     ReplicateShardBuilder::new(0, 0, 0).build();
     ReplicaBuilder::new(0, 0).build();

tests/protos.rs

@@ -612,10 +612,7 @@ fn configure_builder(builder: Builder) -> Builder {
                 "UpdateCollectionClusterSetupRequest.timeout",
                 DEFAULT_OPTION,
             ),
-            (
-                "UpdateCollectionClusterSetupRequest.operation",
-                DEFAULT_OPTION_INTO,
-            ),
+            ("UpdateCollectionClusterSetupRequest.operation", PUBLIC_ONLY),
             // MoveShard
             ("MoveShard.shard_id", PUBLIC_ONLY),
             ("MoveShard.to_shard_id", DEFAULT_OPTION),