feat: Optionally delete points

Signed-off-by: Anush008 <[email protected]>

Author: Anush008

examples/text_embedding_qdrant/pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-name = "text-embedding"
+name = "text-embedding-qdrant"
 version = "0.1.0"
 description = "Simple example for cocoindex: build embedding index based on local text files."
 requires-python = ">=3.10"

src/ops/storages/qdrant.rs

@@ -10,7 +10,8 @@ use anyhow::{bail, Result};
 use futures::FutureExt;
 use qdrant_client::qdrant::vectors_output::VectorsOptions;
 use qdrant_client::qdrant::{
-    NamedVectors, PointId, PointStruct, UpsertPointsBuilder, Value as QdrantValue,
+    DeletePointsBuilder, NamedVectors, PointId, PointStruct, PointsIdsList, UpsertPointsBuilder,
+    Value as QdrantValue,
 };
 use qdrant_client::qdrant::{Query, QueryPointsBuilder, ScoredPoint};
 use qdrant_client::Qdrant;
@@ -77,6 +78,23 @@ impl ExportTargetExecutor for Executor {
         self.client
             .upsert_points(UpsertPointsBuilder::new(&self.collection_name, points).wait(true))
             .await?;
+
+        let ids = mutation
+            .delete_keys
+            .iter()
+            .map(key_to_point_id)
+            .collect::<Result<Vec<_>>>()?;
+
+        if !ids.is_empty() {
+            self.client
+                .delete_points(
+                    DeletePointsBuilder::new(&self.collection_name)
+                        .points(PointsIdsList { ids })
+                        .wait(true),
+                )
+                .await?;
+        }
+
         Ok(())
     }
 }