PostgreSQL Store

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

Setup01

import (
    "github.com/xraph/grove"
    "github.com/xraph/grove/drivers/pgdriver"
    "github.com/xraph/sentinel/store/postgres"
)

db, err := grove.Open(pgdriver.Open("postgres://user:pass@localhost:5432/sentinel?sslmode=disable"))
if err != nil {
    log.Fatal(err)
}

pgStore := postgres.New(db)

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

Tables02

The PostgreSQL store creates the following tables:

JSONB columns03

Complex nested structures are stored as JSONB columns:

• Case scorers and tags → scorers JSONB, tags JSONB

• Result scorer outputs and run trace → scorer_results JSONB, run_trace JSONB

• Dimension scores → dimension_scores JSONB

• All entities → metadata JSONB

Internals04

Composite store interface05

The PostgreSQL store implements all 5 sub-interfaces:

type Store interface {
    suite.Store         // 6 methods
    testcase.Store      // 7 methods
    evalrun.Store       // 5 methods
    baseline.Store      // 5 methods
    promptversion.Store // 5 methods

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

Embedded migrations06

Migrations are registered programmatically via grove's migration orchestrator and run automatically when Migrate() is called. There are 5 migration steps:

1. create_suites — Creates sentinel_suites

2. create_cases — Creates sentinel_cases

3. create_runs_and_results — Creates sentinel_runs and sentinel_results

4. create_baselines — Creates sentinel_baselines

5. create_prompt_versions — Creates sentinel_prompt_versions

Other store backends07

All backends implement the same store.Store interface identically.

Usage with engine08

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