PostgreSQL

Use PostgreSQL for agent session storage and persistence.

Kern supports using PostgreSQL as a database with the PostgresDb class.

Usage

1from kern.agent import Agent
2from kern.db.postgres import PostgresDb
3
4db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai" # Replace with your own connection string
5
6# Setup your Database
7db = PostgresDb(db_url=db_url)
8
9# Setup your Agent with the Database
10agent = Agent(db=db)

Run Postgres (with PgVector)

Install docker desktop and run PgVector on port 5532 using:

1docker run -d \
2 -e POSTGRES_DB=ai \
3 -e POSTGRES_USER=ai \
4 -e POSTGRES_PASSWORD=ai \
5 -e PGDATA=/var/lib/postgresql/data/pgdata \
6 -v pgvolume:/var/lib/postgresql/data \
7 -p 5532:5432 \
8 --name pgvector \
9 agnohq/pgvector:16

Params

ParameterTypeDefaultDescription
idOptional[str]-The ID of the database instance. UUID by default.
db_urlOptional[str]-The database URL to connect to.
db_engineOptional[Engine]-The SQLAlchemy database engine to use.
db_schemaOptional[str]-The database schema to use.
session_tableOptional[str]-Name of the table to store Agent, Team and Workflow sessions.
memory_tableOptional[str]-Name of the table to store memories.
metrics_tableOptional[str]-Name of the table to store metrics.
eval_tableOptional[str]-Name of the table to store evaluation runs data.
knowledge_tableOptional[str]-Name of the table to store knowledge content.
traces_tableOptional[str]-Name of the table to store traces.
spans_tableOptional[str]-Name of the table to store spans.