PgVectorIngestOperator¶

使用 PgVectorIngestOperator 将向量嵌入以及其他数据列值存储到支持向量的 PostgreSQL 表中。

使用操作符¶

PgVectorIngestOperator 接受一个 SQL 查询，该查询可用于将数据以及向量嵌入列数据插入到具有向量扩展的 Postgres 数据库表中。

确保在您的 Postgres 数据库上安装了 pgvector 扩展。有关安装说明，请参阅 https://github.com/pgvector/pgvector#installation。

下面显示了一个使用操作符提取数据的示例

tests/system/providers/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]')",
    )