airflow.providers.pgvector.hooks.pgvector

类

PgVectorHook

扩展 PostgresHook 以处理 PostgreSQL 和 pgvector 扩展，支持向量数据类型。

模块内容

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

基类： airflow.providers.postgres.hooks.postgres.PostgresHook

扩展 PostgresHook 以处理 PostgreSQL 和 pgvector 扩展，支持向量数据类型。

create_table(table_name, columns, if_not_exists=True)[源码]

在 Postgres 数据库中创建一个表。

参数:

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

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

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

create_extension(extension_name, if_not_exists=True)[源码]

创建一个 PostgreSQL 扩展。

参数:

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

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

drop_table(table_name, if_exists=True)[源码]

从 Postgres 数据库中删除一个表。

参数:

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

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

truncate_table(table_name, restart_identity=True)[源码]

截断表，移除所有行。

参数:

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

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

本条目有帮助吗？