feat: Qdrant storage support

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

Author: Anush008

docs/docs/ops/storages.md

@@ -11,6 +11,16 @@ description: CocoIndex Built-in Storages
 
 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).
+* `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.
+* `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

@@ -60,7 +60,7 @@ def pdf_embedding_flow(flow_builder: cocoindex.FlowBuilder, data_scope: cocoinde
 
     doc_embeddings.export(
         "doc_embeddings",
-        cocoindex.storages.Postgres(),
+        cocoindex.storages.Qdrant(qdrant_url="http://localhost:6333", collection_name="cocoindex"),
         primary_key_fields=["filename", "location"],
         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

@@ -132,6 +132,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(())
 }