feat: Qdrant storage support

Cargo.toml

@@ -89,4 +89,5 @@ rustls = { version = "0.23.25" }
 http-body-util = "0.1.3"
 yaml-rust2 = "0.10.0"
 urlencoding = "2.1.3"
+qdrant-client = "1.13.0"
 uuid = { version = "1.16.0", features = ["serde", "v4", "v8"] }

docs/docs/ops/storages.md

@@ -14,3 +14,13 @@ The spec takes the following fields:
 *   `database_url` (type: `str`, optional): The URL of the Postgres database to use as the internal storage, e.g. `postgres://cocoindex:cocoindex@localhost/cocoindex`. If unspecified, will use the same database as the [internal storage](/docs/core/basics#internal-storage).
 
 *   `table_name` (type: `str`, optional): The name of the table to store to. If unspecified, will generate a new automatically. We recommend specifying a name explicitly if you want to directly query the table. It can be omitted if you want to use CocoIndex's query handlers to query the table.
+
+## Qdrant
+
+`Qdrant` exports data to a [Qdrant](https://qdrant.tech/) collection.
+
+The spec takes the following fields:
+
+*   `qdrant_url` (type: `str`, required): The [gRPC URL](https://qdrant.tech/documentation/interfaces/#grpc-interface) of the Qdrant instance. Defaults to http://localhost:6334/.
+
+*   `collection` (type: `str`, required): The name of the collection to export the data to.

examples/pdf_embedding/main.py

@@ -61,7 +61,7 @@ def pdf_embedding_flow(flow_builder: cocoindex.FlowBuilder, data_scope: cocoinde
 
     doc_embeddings.export(
         "doc_embeddings",
-        cocoindex.storages.Postgres(),
+        cocoindex.storages.Qdrant(collection_name="cocoindex"),
         primary_key_fields=["id"],
         vector_index=[("embedding", cocoindex.VectorSimilarityMetric.COSINE_SIMILARITY)])
 

python/cocoindex/storages.py

@@ -6,3 +6,8 @@ class Postgres(op.StorageSpec):
 
     database_url: str | None = None
     table_name: str | None = None
+
+class Qdrant(op.StorageSpec):
+    """Storage powered by Qdrant - https://qdrant.tech/."""
+
+    collection_name: str

src/ops/interface.rs

@@ -141,6 +141,7 @@ pub enum ExecutorFactory {
     ExportTarget(Arc<dyn ExportTargetFactory + Send + Sync>),
 }
 
+#[derive(Debug)]
 pub struct VectorMatchQuery {
     pub vector_field_name: String,
     pub vector: Vec<f32>,

src/ops/registration.rs

@@ -13,6 +13,7 @@ fn register_executor_factories(registry: &mut ExecutorFactoryRegistry) -> Result
     functions::extract_by_llm::Factory.register(registry)?;
 
     Arc::new(storages::postgres::Factory::default()).register(registry)?;
+    Arc::new(storages::qdrant::Factory::default()).register(registry)?;
 
     Ok(())
 }

src/ops/storages/mod.rs

@@ -1 +1,2 @@
 pub mod postgres;
+pub mod qdrant;