Pipelines reference - Models

This section provides reference documentation for EDB Postgres AI AI Accelerator Pipelines Models. It includes information on the functions and views available in the aidb extension for working with Models.

Tables

aidb.model_providers

The aidb.model_providers table stores information about the model providers that have been created in the database.

Functions

aidb.register_model

Registers a a new model in the system by saving its name, provider and optional configuration.

Parameters

Example

SELECT aidb.register_model(
               name => 'my_t5'::text,
               provider => 't5_local'::character varying,
               config => '{"param1": "value1", "param2": "value2"}'::jsonb,
               credentials => '{"token": "abcd"}'::jsonb
           );

or equivalently, using default values:

SELECT aidb.register_model('my_t5', 't5_local');

aidb.list_registered_models

Returns a list of all registered models and their configured options.

Parameters

None

Returns

Example

SELECT * FROM aidb.list_registered_models();

 name  |  provider  |    options
-------+------------+---------------
 bert  | bert_local | {"config={}"}
 clip  | clip_local | {"config={}"}
 t5    | t5_local   | {"config={}"}

aidb.get_registered_model

Returns the configuration for a registered model.

Parameters

Returns

Example

SELECT * FROM aidb.get_registered_model('t5');

 name | provider |    options
------+----------+---------------
 t5   | t5_local | {"config={}"}
(1 row)

aidb.delete_registered_model

Deletes a registered model.

Parameters

Example

SELECT aidb.delete_registered_model('t5');

     delete_registered_model
---------------------------------
 (t5,t5_local,"{""config={}""}")
(1 row)

Returns

aidb.encode_text

Encodes text using a registered model, generating a vector representation of a given text input.

Parameters

aidb.encode_text_batch

Encodes a batch of text using a registered model, generating a vector representation of a given text inputs.

Parameters

aidb.decode_text

Decodes text using a registered model, generating a vector representation of a given text input.

Parameters

Returns

aidb.decode_text_batch

Decodes a batch of text using a registered model, generating a representation of a given text input.

Parameters

Returns

aidb.encode_image

Encodes an image using a registered model, generating a vector representation of a given image input.

Parameters

Returns