feat: Support for api_key

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

Author: Anush008

docs/docs/ops/storages.md

@@ -21,10 +21,12 @@ Exports data to a [Qdrant](https://qdrant.tech/) collection.
 
 The spec takes the following fields:
 
-*   `grpc_url` (type: `str`, required): The [gRPC URL](https://qdrant.tech/documentation/interfaces/#grpc-interface) of the Qdrant instance. Defaults to `http://localhost:6334/`.
-
 *   `collection_name` (type: `str`, required): The name of the collection to export the data to.
 
+*   `grpc_url` (type: `str`, optional): The [gRPC URL](https://qdrant.tech/documentation/interfaces/#grpc-interface) of the Qdrant instance. Defaults to `http://localhost:6334/`.
+
+*   `api_key` (type: `str`, optional). API key to authenticate requests with.
+
 The field name for the vector embeddings must match the [vector name](https://qdrant.tech/documentation/concepts/vectors/#named-vectors) used when the collection was created.
 
 If no primary key is set during export, a random UUID is used as the Qdrant point ID.

python/cocoindex/storages.py

@@ -17,6 +17,7 @@ class Qdrant(op.StorageSpec):
 
     collection_name: str
     grpc_url: str = "http://localhost:6334/"
+    api_key: str | None = None
 
 @dataclass
 class Neo4jConnectionSpec:

src/ops/storages/qdrant.rs

@@ -37,6 +37,7 @@ fn key_value_fields_iter<'a>(
 pub struct Spec {
     collection_name: String,
     grpc_url: String,
+    api_key: Option<String>,
 }
 
 pub struct Executor {
@@ -51,6 +52,7 @@ impl Executor {
     fn new(
         url: String,
         collection_name: String,
+        api_key: Option<String>,
         key_fields_schema: Vec<FieldSchema>,
         value_fields_schema: Vec<FieldSchema>,
     ) -> Result<Self> {
@@ -60,7 +62,10 @@ impl Executor {
             .cloned()
             .collect::<Vec<_>>();
         Ok(Self {
-            client: Qdrant::from_url(&url).build()?,
+            client: Qdrant::from_url(&url)
+                .api_key(api_key)
+                .skip_compatibility_check()
+                .build()?,
             key_fields_schema,
             value_fields_schema,
             all_fields,
@@ -339,6 +344,7 @@ impl StorageFactoryBase for Arc<Factory> {
             let executor = Arc::new(Executor::new(
                 spec.grpc_url,
                 spec.collection_name.clone(),
+                spec.api_key,
                 key_fields_schema,
                 value_fields_schema,
             )?);