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,
)

此條目有幫助嗎?