fmt

qdrant · Jun 19, 2024 · 243e313 · 243e313
1 parent d77dddd
commit 243e313
Show file tree

Hide file tree

Showing 8 changed files with 67 additions and 40 deletions.
diff --git a/src/builder_ext.rs b/src/builder_ext.rs
@@ -1,7 +1,23 @@
 use std::collections::HashMap;
 
 use crate::builder_types::RecommendExample;
-use crate::qdrant::{BinaryQuantizationBuilder, ClearPayloadPointsBuilder, ContextExamplePair, CountPointsBuilder, CreateAliasBuilder, CreateCollectionBuilder, CreateFieldIndexCollectionBuilder, CreateShardKeyRequestBuilder, DeleteCollectionBuilder, DeleteFieldIndexCollectionBuilder, DeletePayloadPointsBuilder, DeletePointsBuilder, DeletePointVectorsBuilder, DeleteShardKey, DeleteShardKeyRequestBuilder, DiscoverBatchPointsBuilder, DiscoverPoints, DiscoverPointsBuilder, Distance, GetPointsBuilder, LookupLocationBuilder, OrderByBuilder, PayloadExcludeSelector, PayloadIncludeSelector, PointId, PointStruct, PointsUpdateOperation, PointVectors, ProductQuantizationBuilder, QuantizationType, QueryPointsBuilder, RecommendBatchPointsBuilder, RecommendPointGroups, RecommendPointGroupsBuilder, RecommendPoints, RecommendPointsBuilder, RenameAliasBuilder, ScalarQuantizationBuilder, ScrollPointsBuilder, SearchBatchPointsBuilder, SearchPointGroupsBuilder, SearchPoints, SearchPointsBuilder, SetPayloadPointsBuilder, shard_key, ShardKey, TextIndexParamsBuilder, TokenizerType, UpdateBatchPointsBuilder, UpdateCollectionBuilder, UpdateCollectionClusterSetupRequestBuilder, UpdatePointVectorsBuilder, UpsertPointsBuilder, Value, VectorParamsBuilder, VectorsSelector, WithLookupBuilder};
+use crate::qdrant::{
+    shard_key, BinaryQuantizationBuilder, ClearPayloadPointsBuilder, ContextExamplePair,
+    CountPointsBuilder, CreateAliasBuilder, CreateCollectionBuilder,
+    CreateFieldIndexCollectionBuilder, CreateShardKeyRequestBuilder, DeleteCollectionBuilder,
+    DeleteFieldIndexCollectionBuilder, DeletePayloadPointsBuilder, DeletePointVectorsBuilder,
+    DeletePointsBuilder, DeleteShardKey, DeleteShardKeyRequestBuilder, DiscoverBatchPointsBuilder,
+    DiscoverPoints, DiscoverPointsBuilder, Distance, GetPointsBuilder, LookupLocationBuilder,
+    OrderByBuilder, PayloadExcludeSelector, PayloadIncludeSelector, PointId, PointStruct,
+    PointVectors, PointsUpdateOperation, ProductQuantizationBuilder, QuantizationType,
+    QueryPointsBuilder, RecommendBatchPointsBuilder, RecommendPointGroups,
+    RecommendPointGroupsBuilder, RecommendPoints, RecommendPointsBuilder, RenameAliasBuilder,
+    ScalarQuantizationBuilder, ScrollPointsBuilder, SearchBatchPointsBuilder,
+    SearchPointGroupsBuilder, SearchPoints, SearchPointsBuilder, SetPayloadPointsBuilder, ShardKey,
+    TextIndexParamsBuilder, TokenizerType, UpdateBatchPointsBuilder, UpdateCollectionBuilder,
+    UpdateCollectionClusterSetupRequestBuilder, UpdatePointVectorsBuilder, UpsertPointsBuilder,
+    Value, VectorParamsBuilder, VectorsSelector, WithLookupBuilder,
+};
 
 impl VectorParamsBuilder {
     pub fn new(size: u64, distance: Distance) -> Self {
@@ -461,4 +477,4 @@ impl QueryPointsBuilder {
         builder.collection_name = Some(collection_name.into());
         builder
     }
-}
+}
diff --git a/src/grpc_conversions/query.rs b/src/grpc_conversions/query.rs
@@ -6,4 +6,4 @@
 //             variant: Some(query::Variant::Nearest()),
 //         }
 //     }
-// }
+// }
diff --git a/src/lib.rs b/src/lib.rs
@@ -134,9 +134,9 @@ mod tests {
         SetPayloadPointsBuilder, SnapshotDownloadBuilder, Struct, UpsertPointsBuilder, Value,
         VectorParamsBuilder,
     };
+    use crate::qdrant_client::config::QdrantConfig;
     use crate::qdrant_client::Qdrant;
     use std::collections::HashMap;
-    use crate::qdrant_client::config::QdrantConfig;
 
     #[test]
     fn display() {

diff --git a/src/qdrant_client/collection.rs b/src/qdrant_client/collection.rs
@@ -219,9 +219,9 @@ mod tests {
         CountPointsBuilder, CreateCollectionBuilder, PointStruct, SearchPointsBuilder,
         UpsertPointsBuilder, VectorParamsBuilder,
     };
+    use crate::qdrant_client::config::QdrantConfig;
     use std::time::Duration;
     use tokio::time::sleep;
-    use crate::qdrant_client::config::QdrantConfig;
 
     #[tokio::test]
     async fn create_collection_and_do_the_search() -> Result<()> {

diff --git a/src/qdrant_client/config.rs b/src/qdrant_client/config.rs
@@ -1,6 +1,6 @@
+use crate::prelude::QdrantError;
 use crate::qdrant_client::Qdrant;
 use std::time::Duration;
-use crate::prelude::QdrantError;
 
 pub struct QdrantConfig {
     pub uri: String,

diff --git a/src/qdrant_client/mod.rs b/src/qdrant_client/mod.rs
@@ -1,10 +1,10 @@
 pub mod collection;
+pub mod config;
 pub mod errors;
 mod points;
+mod query;
 pub mod sharding_keys;
 pub mod snapshot;
-mod query;
-pub mod config;
 
 use crate::channel_pool::ChannelPool;
 use crate::qdrant::{qdrant_client, HealthCheckReply, HealthCheckRequest};

diff --git a/src/qdrant_client/points.rs b/src/qdrant_client/points.rs
@@ -1,19 +1,28 @@
 use std::future::Future;
 
 use tonic::codegen::InterceptedService;
-use tonic::Status;
 use tonic::transport::Channel;
+use tonic::Status;
 
 use crate::auth::TokenInterceptor;
 use crate::prelude::SearchPoints;
-use crate::qdrant::{ClearPayloadPoints, CountPoints, CountResponse, CreateFieldIndexCollection, DeleteFieldIndexCollection, DeletePayloadPoints, DeletePoints, DeletePointVectors, DiscoverBatchPoints, DiscoverBatchResponse, DiscoverPoints, DiscoverResponse, GetPoints, GetResponse, PointsOperationResponse, RecommendBatchPoints, RecommendBatchResponse, RecommendGroupsResponse, RecommendPointGroups, RecommendPoints, RecommendResponse, ScrollPoints, ScrollResponse, SearchBatchPoints, SearchBatchResponse, SearchGroupsResponse, SearchPointGroups, SearchResponse, SetPayloadPoints, UpdateBatchPoints, UpdateBatchResponse, UpdatePointVectors, UpsertPoints};
 use crate::qdrant::points_client::PointsClient;
+use crate::qdrant::{
+    ClearPayloadPoints, CountPoints, CountResponse, CreateFieldIndexCollection,
+    DeleteFieldIndexCollection, DeletePayloadPoints, DeletePointVectors, DeletePoints,
+    DiscoverBatchPoints, DiscoverBatchResponse, DiscoverPoints, DiscoverResponse, GetPoints,
+    GetResponse, PointsOperationResponse, RecommendBatchPoints, RecommendBatchResponse,
+    RecommendGroupsResponse, RecommendPointGroups, RecommendPoints, RecommendResponse,
+    ScrollPoints, ScrollResponse, SearchBatchPoints, SearchBatchResponse, SearchGroupsResponse,
+    SearchPointGroups, SearchResponse, SetPayloadPoints, UpdateBatchPoints, UpdateBatchResponse,
+    UpdatePointVectors, UpsertPoints,
+};
 use crate::qdrant_client::{Qdrant, Result};
 
 impl Qdrant {
     pub(crate) async fn with_points_client<
         T,
-        O: Future<Output=std::result::Result<T, Status>>,
+        O: Future<Output = std::result::Result<T, Status>>,
     >(
         &self,
         f: impl Fn(PointsClient<InterceptedService<Channel, TokenInterceptor>>) -> O,
@@ -45,7 +54,7 @@ impl Qdrant {
             let result = points_api.search(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn search_batch_points(
@@ -58,7 +67,7 @@ impl Qdrant {
             let result = points_api.search_batch(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn search_groups(
@@ -71,7 +80,7 @@ impl Qdrant {
             let result = points_api.search_groups(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn batch_updates(
@@ -83,7 +92,7 @@ impl Qdrant {
         self.with_points_client(|mut points_api| async move {
             Ok(points_api.update_batch(request.clone()).await?.into_inner())
         })
-            .await
+        .await
     }
 
     /// Update or insert points into the collection.
@@ -98,7 +107,7 @@ impl Qdrant {
         self.with_points_client(|mut points_api| async move {
             Ok(points_api.upsert(request.clone()).await?.into_inner())
         })
-            .await
+        .await
     }
 
     /// Update or insert points into the collection, splitting in chunks.
@@ -137,7 +146,7 @@ impl Qdrant {
 
             Ok(resp)
         })
-            .await
+        .await
     }
 
     pub async fn set_payload(
@@ -150,7 +159,7 @@ impl Qdrant {
             let result = points_api.set_payload(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn overwrite_payload(
@@ -163,7 +172,7 @@ impl Qdrant {
             let result = points_api.overwrite_payload(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn delete_payload(
@@ -176,7 +185,7 @@ impl Qdrant {
             let result = points_api.delete_payload(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn clear_payload(
@@ -189,7 +198,7 @@ impl Qdrant {
             let result = points_api.clear_payload(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn get_points(&self, request: impl Into<GetPoints>) -> Result<GetResponse> {
@@ -199,7 +208,7 @@ impl Qdrant {
             let result = points_api.get(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn delete_points(
@@ -212,7 +221,7 @@ impl Qdrant {
             let result = points_api.delete(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn delete_vectors(
@@ -225,7 +234,7 @@ impl Qdrant {
             let result = points_api.delete_vectors(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn update_vectors(
@@ -238,7 +247,7 @@ impl Qdrant {
             let result = points_api.update_vectors(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn scroll(&self, request: impl Into<ScrollPoints>) -> Result<ScrollResponse> {
@@ -248,7 +257,7 @@ impl Qdrant {
             let result = points_api.scroll(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn recommend(
@@ -261,7 +270,7 @@ impl Qdrant {
             let result = points_api.recommend(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn recommend_batch(
@@ -274,7 +283,7 @@ impl Qdrant {
             let result = points_api.recommend_batch(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn recommend_groups(
@@ -287,7 +296,7 @@ impl Qdrant {
             let result = points_api.recommend_groups(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn discover(&self, request: impl Into<DiscoverPoints>) -> Result<DiscoverResponse> {
@@ -297,7 +306,7 @@ impl Qdrant {
             let result = points_api.discover(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn discover_batch(
@@ -308,7 +317,7 @@ impl Qdrant {
             let result = points_api.discover_batch(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn count(&self, request: impl Into<CountPoints>) -> Result<CountResponse> {
@@ -318,7 +327,7 @@ impl Qdrant {
             let result = points_api.count(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn update_batch_points(
@@ -331,7 +340,7 @@ impl Qdrant {
             let result = points_api.update_batch(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     /// Create index for a payload field
@@ -345,7 +354,7 @@ impl Qdrant {
             let result = client.create_field_index(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 
     pub async fn delete_field_index(
@@ -358,6 +367,6 @@ impl Qdrant {
             let result = client.delete_field_index(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 }
diff --git a/src/qdrant_client/query.rs b/src/qdrant_client/query.rs
@@ -2,22 +2,24 @@ use crate::qdrant::{QueryPoints, QueryResponse};
 use crate::qdrant_client::Qdrant;
 
 impl Qdrant {
-    pub async fn query(&self, request: impl Into<QueryPoints>) -> crate::qdrant_client::Result<QueryResponse> {
+    pub async fn query(
+        &self,
+        request: impl Into<QueryPoints>,
+    ) -> crate::qdrant_client::Result<QueryResponse> {
         let request = &request.into();
 
         self.with_points_client(|mut points_api| async move {
             let result = points_api.query(request.clone()).await?;
             Ok(result.into_inner())
         })
-            .await
+        .await
     }
 }
 
-
 #[cfg(test)]
 mod tests {
-    use crate::qdrant::{CreateCollectionBuilder, QueryPointsBuilder};
     use super::*;
+    use crate::qdrant::{CreateCollectionBuilder, QueryPointsBuilder};
 
     #[tokio::test]
     async fn test_query() {
@@ -33,4 +35,4 @@ mod tests {
         let response = client.query(request).await.unwrap();
         assert_eq!(response.result.len(), 0);
     }
-}
+}