airflow.providers.qdrant.operators.qdrant

Module Contents

Classes

QdrantIngestOperator

Upload points to a Qdrant collection.
class airflow.providers.qdrant.operators.qdrant.QdrantIngestOperator(*, conn_id=QdrantHook.default_conn_name, collection_name, vectors, payload=None, ids=None, batch_size=64, parallel=1, method=None, max_retries=3, wait=True, **kwargs)[source]

Bases: airflow.models.BaseOperator

Upload points to a Qdrant collection.

See also

For more information on how to use this operator, take a look at the guide: QdrantIngestOperator

Parameters

  • conn_id (str) – The connection id to connect to a Qdrant instance.

  • collection_name (str) – The name of the collection to ingest data into.

  • vectors (Iterable[qdrant_client.models.VectorStruct]) – An iterable over vectors to upload.

  • payload (Iterable[dict[str, Any]] | None) – Iterable of vector payloads, Optional. Defaults to None.

  • ids (Iterable[int | str] | None) – Iterable of custom vector ids, Optional. Defaults to None.

  • batch_size (int) – Number of points to upload per-request. Defaults to 64.

  • parallel (int) – Number of parallel upload processes. Defaults to 1.

  • method (str | None) – Start method for parallel processes. Defaults to ‘forkserver’.

  • max_retries (int) – Number of retries for failed requests. Defaults to 3.

  • wait (bool) – Await for the results to be applied on the server side. Defaults to True.

  • kwargs (Any) – Additional keyword arguments passed to the BaseOperator constructor.

template_fields: Sequence[str] = ('collection_name', 'vectors', 'payload', 'ids', 'batch_size', 'parallel', 'method',...[source]
hook()[source]

Return an instance of QdrantHook.

execute(context)[source]

Upload points to a Qdrant collection.

Was this entry helpful?