airflow.providers.pgvector.hooks.pgvector

类

PgVectorHook

扩展 PostgresHook，以便在 PostgreSQL 中使用 pgvector 扩展处理向量数据类型。

模块内容

class airflow.providers.pgvector.hooks.pgvector.PgVectorHook(*args, **kwargs)[source]

Bases: airflow.providers.postgres.hooks.postgres.PostgresHook

扩展 PostgresHook，以便在 PostgreSQL 中使用 pgvector 扩展处理向量数据类型。

create_table(table_name, columns, if_not_exists=True)[source]

在 Postgres 数据库中创建表。

参数:

  • table_name (str) – 要创建的表名。

  • columns (list[str]) – 表的列定义列表。

  • if_not_exists (bool) – 如果为 True，仅在表不存在时创建表。

create_extension(extension_name, if_not_exists=True)[source]

创建 PostgreSQL 扩展。

参数:

  • extension_name (str) – 要创建的扩展名称。

  • if_not_exists (bool) – 如果为 True，仅在扩展不存在时创建。

drop_table(table_name, if_exists=True)[source]

从 Postgres 数据库中删除表。

参数:

  • table_name (str) – 要删除的表名。

  • if_exists (bool) – 如果为 True，仅在表存在时删除。

truncate_table(table_name, restart_identity=True)[source]

截断表，删除所有行。

参数:

  • table_name (str) – 要截断的表名。

  • restart_identity (bool) – 如果为 True，并且表有序列，则重新启动该序列。

此条目是否有帮助？