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