airflow.providers.pgvector.hooks.pgvector

Classes

PgVectorHook

Extend PostgresHook for working with PostgreSQL and pgvector extension for vector data types.

Module Contents

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

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

Extend PostgresHook for working with PostgreSQL and pgvector extension for vector data types.

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

Create a table in the Postgres database.

Parameters:

  • table_name (str) – The name of the table to create.

  • columns (list[str]) – A list of column definitions for the table.

  • if_not_exists (bool) – If True, only create the table if it does not already exist.

create_extension(extension_name, if_not_exists=True)[source]

Create a PostgreSQL extension.

Parameters:

  • extension_name (str) – The name of the extension to create.

  • if_not_exists (bool) – If True, only create the extension if it does not already exist.

drop_table(table_name, if_exists=True)[source]

Drop a table from the Postgres database.

Parameters:

  • table_name (str) – The name of the table to drop.

  • if_exists (bool) – If True, only drop the table if it exists.

truncate_table(table_name, restart_identity=True)[source]

Truncate a table, removing all rows.

Parameters:

  • table_name (str) – The name of the table to truncate.

  • restart_identity (bool) – If True, restart the serial sequence if the table has one.

Was this entry helpful?