[Proposal] Builder generation for CreateCollection and VectorParams (#…

…108)

* Builder generation for CreateCollection and VectorParams

Cargo.toml

@@ -22,6 +22,7 @@ serde = { version = "1", features = ["derive"], optional = true }
 serde_json = { version = "1", optional = true }
 reqwest = { version = "0.12.3", optional = true, default-features = false, features = ["stream", "rustls-tls", "http2"] }
 futures-util = { version = "0.3.30", optional = true }
+derive_builder = { version = "0.20" }
 
 [dev-dependencies]
 tonic-build = { version = "0.11.0", features = ["prost"] }

examples/search.rs

@@ -1,8 +1,8 @@
 use anyhow::Result;
 use qdrant_client::prelude::*;
-use qdrant_client::qdrant::vectors_config::Config;
+use qdrant_client::qdrant::Distance;
 use qdrant_client::qdrant::{
-    Condition, CreateCollection, Filter, SearchPoints, VectorParams, VectorsConfig,
+    Condition, CreateCollectionBuilder, Filter, SearchPoints, VectorParamsBuilder,
 };
 use serde_json::json;
 
@@ -28,17 +28,16 @@ async fn main() -> Result<()> {
     client.delete_collection(collection_name).await?;
 
     client
-        .create_collection(&CreateCollection {
-            collection_name: collection_name.into(),
-            vectors_config: Some(VectorsConfig {
-                config: Some(Config::Params(VectorParams {
-                    size: 10,
-                    distance: Distance::Cosine.into(),
-                    ..Default::default()
-                })),
-            }),
-            ..Default::default()
-        })
+        .create_collection(
+            &CreateCollectionBuilder::default()
+                .collection_name(collection_name)
+                .vectors_config(
+                    VectorParamsBuilder::default()
+                        .distance(Distance::Cosine)
+                        .size(10),
+                )
+                .build(),
+        )
         .await?;
 
     let collection_info = client.collection_info(collection_name).await?;

src/grpc_ext.rs

@@ -16,9 +16,9 @@ use crate::qdrant::{
     ProductQuantization, QuantizationConfig, QuantizationConfigDiff, ReadConsistency, RenameAlias,
     RepeatedIntegers, RepeatedStrings, ScalarQuantization, ShardKey, ShardKeySelector,
     SparseIndexConfig, SparseIndices, SparseVectorConfig, SparseVectorParams, StartFrom, Struct,
-    TargetVector, TextIndexParams, Value, Vector, VectorExample, VectorParams, VectorParamsDiff,
-    VectorParamsDiffMap, VectorParamsMap, Vectors, VectorsConfig, VectorsConfigDiff,
-    VectorsSelector, WithPayloadSelector, WithVectorsSelector,
+    TargetVector, TextIndexParams, Value, Vector, VectorExample, VectorParams, VectorParamsBuilder,
+    VectorParamsDiff, VectorParamsDiffMap, VectorParamsMap, Vectors, VectorsConfig,
+    VectorsConfigDiff, VectorsSelector, WithPayloadSelector, WithVectorsSelector,
 };
 use std::collections::HashMap;
 
@@ -771,3 +771,9 @@ impl From<Vec<PointId>> for PointsIdsList {
         Self { ids: value }
     }
 }
+
+impl From<&mut VectorParamsBuilder> for vectors_config::Config {
+    fn from(value: &mut VectorParamsBuilder) -> Self {
+        value.build().into()
+    }
+}