QdrantIngestOperator

使用 QdrantIngestOperator 将数据摄取到 Qdrant 实例中。

使用 Operator

QdrantIngestOperator 需要将 vectors 作为输入摄取到 Qdrant 中。使用 conn_id 参数指定用于连接 Qdrant 实例的 Qdrant 连接。vectors 还可以包含引用与 vectors 对应的原始文本的 metadata,这些 metadata 也可以摄取到数据库中。

以下是使用此 Operator 的示例

tests/system/qdrant/example_dag_qdrant.py

vectors = [[0.732, 0.611, 0.289, 0.421], [0.217, 0.526, 0.416, 0.981], [0.326, 0.483, 0.376, 0.136]]
ids: list[str | int] = [32, 21, "b626f6a9-b14d-4af9-b7c3-43d8deb719a6"]
payload = [{"meta": "data"}, {"meta": "data_2"}, {"meta": "data_3", "extra": "data"}]

QdrantIngestOperator(
    task_id="qdrant_ingest",
    collection_name="test_collection",
    vectors=vectors,
    ids=ids,
    payload=payload,
    batch_size=1,
)

此条目有帮助吗?