PgVectorIngestOperator¶

使用 PgVectorIngestOperator 將向量嵌入以及其他資料列值儲存到支援向量的 PostgreSQL 表中。

使用 Operator¶

PgVectorIngestOperator 接受一個 SQL 查詢，該查詢可用於將資料和向量嵌入列資料插入到已安裝向量擴充套件的 Postgres 資料庫表中。

請確保您的 Postgres 資料庫中安裝了 pgvector 擴充套件。安裝說明請參閱 https://github.com/pgvector/pgvector#installation。

下面顯示了使用 Operator 攝取資料的示例

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]')",
    )