PostgreSQL Store

The store/postgres package provides a production-ready implementation of the composite store.Store interface using PostgreSQL with the bun ORM.

Setup

import (
    "github.com/uptrace/bun"
    "github.com/xraph/cortex/store/postgres"
)

db := connectDB() // your *bun.DB connection
pgStore := postgres.New(db)

// Run migrations
if err := pgStore.Migrate(ctx); err != nil {
    log.Fatal(err)
}

Tables

The PostgreSQL store creates the following tables:

JSONB columns

Complex nested structures are stored as JSONB columns:

• Skill tools and knowledge → tools JSONB, knowledge JSONB
• Trait dimensions and influences → dimensions JSONB, influences JSONB
• Behavior triggers and actions → triggers JSONB, actions JSONB
• Persona styles → cognitive_style JSONB, communication_style JSONB, perception JSONB
• All entities → metadata JSONB

Composite store interface

The PostgreSQL store implements all 8 sub-interfaces:

type Store interface {
    agent.Store      // 6 methods
    skill.Store      // 6 methods
    trait.Store      // 6 methods
    behavior.Store   // 6 methods
    persona.Store    // 6 methods
    run.Store        // 8 methods
    memory.Store     // 8 methods
    checkpoint.Store // 4 methods

    Migrate(ctx context.Context) error
    Ping(ctx context.Context) error
    Close() error
}

Embedded migrations

Migrations are embedded in the binary using embed.FS and run automatically when Migrate() is called. There is no need for external migration tools.

Usage with engine

eng, err := engine.New(
    engine.WithStore(pgStore),
)