PgVectorIngestOperator¶

使用 PgVectorIngestOperator 將向量嵌入與其他資料欄位值一起儲存到支援向量的 PostgreSQL 資料表中。

使用運算子¶

PgVectorIngestOperator 接受 SQL 查詢，該查詢可用於在具有向量擴充功能的 Postgres 資料庫表中插入資料以及向量嵌入欄位資料。

請確保您的 Postgres 資料庫上已安裝 pgvector 擴充功能。有關安裝說明，請參閱 https://github.com/pgvector/pgvector#installation。

下面顯示了使用運算子來擷取資料的範例

tests/system/pgvector/example_pgvector.py[原始碼]

    pgvector_ingest = PgVectorIngestOperator(
        task_id="pgvector_ingest",
        conn_id=POSTGRES_CONN_ID,
        sql=f"INSERT INTO {TABLE_NAME} (name, value, vector_column) "
        f"VALUES ('John Doe', 123, '[1.0, 2.0, 3.0]')",
    )