EDB Postgres AI AI/ML - Installing the aidb tech preview

The preview release of aidb is distributed as a self-contained Docker container that runs PostgreSQL and includes all of the aidb dependencies.

Configuring and running the container image

If you haven't already, sign up for an EDB account and log in to the EDB container registry.

Log in to Docker with the username tech-preview and your EDB Repos 2.0 subscription token as your password:

docker login docker.enterprisedb.com -u tech-preview -p <your_EDB_repo_token>

Login Succeeded

Download the aidb container image:

docker pull docker.enterprisedb.com/tech-preview/aidb

...
Status: Downloaded newer image for docker.enterprisedb.com/tech-preview/aidb:latest
docker.enterprisedb.com/tech-preview/aidb:latest

Specify a password to use for Postgres in the environment variable PGPASSWORD. The tech preview container set up Postgres with this password and use it to connect to it. In bash or zsh set it as follows:

export PGPASSWORD=<your_password>

You can use the aidb extension with encoder LLMs in Open AI or with open encoder LLMs from HuggingFace. If you want to use Open AI you also must provide your API key for that in the OPENAI_API_KEY environment variable:

export OPENAI_API_KEY=<your_openai_key>

You can use the aidb extension with AI data stored in Postgres tables or on S3 compatible object storage. To work with object storage you need to specify the ACCESS_KEY and SECRET_KEY environment variables:

export ACCESS_KEY=<your_access_key>
export SECRET_KEY=<your_secret_key>

Start the aidb tech preview container with the following command. It makes the tech preview PostgreSQL database available on local port 15432:

docker run -d --name aidb \
       -e ACCESS_KEY=$ACCESS_KEY \
       -e SECRET_KEY=$SECRET_KEY \
       -e OPENAI_API_KEY=$OPENAI_API_KEY \
       -e POSTGRES_PASSWORD=$PGPASSWORD \
       -e PGDATA=/var/lib/postgresql/data/pgdata \
       -p 15432:5432 \
       docker.enterprisedb.com/tech-preview/aidb:latest

Connect to Postgres

If you haven't yet, install the Postgres command-line tools. If you're on a Mac, using Homebrew, you can install it as follows:

brew install libpq

Connect to the tech preview PostgreSQL running in the container. Note that this relies on setting the PGPASSWORD environment variable - if you're using a different terminal for this part, make sure you re-export the password:

psql -h localhost -p 15432 -U postgres postgres

psql (16.1, server 16.3 (Debian 16.3-1.pgdg120+1))
Type "help" for help.

postgres=#

Install the aidb extension:

create extension aidb cascade;

NOTICE:  installing required extension "plpython3u"
NOTICE:  installing required extension "vector"
CREATE EXTENSION
postgres=# 

\dx

                               List of installed extensions
    Name    | Version |   Schema   |                     Description
------------+---------+------------+------------------------------------------------------
 aidb       | 0.0.2   | public     | An extension to do the AIs
 plpgsql    | 1.0     | pg_catalog | PL/pgSQL procedural language
 plpython3u | 1.0     | pg_catalog | PL/Python3U untrusted procedural language
 vector     | 0.7.2   | public     | vector data type and ivfflat and hnsw access methods
(4 rows)