PostgreSQL

PostgreSQL-backed session store for persistent conversation history. No extensions required.

Installation

go get github.com/joakimcarlsson/ai/memory/postgres

Setup

import pgsess "github.com/joakimcarlsson/ai/memory/postgres"

sessionStore, err := pgsess.SessionStore(ctx,
    "postgres://user:pass@localhost:5432/mydb?sslmode=disable",
)
if err != nil {
    log.Fatal(err)
}

myAgent := agent.New(llmClient,
    agent.WithSession("conv-1", sessionStore),
)

Tables and indexes are created automatically on first use.

Schema

CREATE TABLE sessions (
    id TEXT PRIMARY KEY,
    created_at TIMESTAMPTZ DEFAULT NOW()
);

CREATE TABLE messages (
    id TEXT PRIMARY KEY,
    session_id TEXT NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
    role TEXT NOT NULL,
    parts JSONB NOT NULL,
    model TEXT,
    created_at BIGINT NOT NULL
);

CREATE INDEX messages_session_idx ON messages(session_id, created_at);

Options

Option	Description
`pgsess.WithIDGenerator(fn)`	Custom ID generator for message records. Default: UUID v4

store, err := pgsess.SessionStore(ctx, connString,
    pgsess.WithIDGenerator(func() string {
        return myCustomID()
    }),
)

Full example

package main

import (
    "context"
    "fmt"
    "log"
    "os"

    "github.com/joakimcarlsson/ai/agent"
    "github.com/joakimcarlsson/ai/memory"
    pgvectormem "github.com/joakimcarlsson/ai/memory/pgvector"
    pgsess "github.com/joakimcarlsson/ai/memory/postgres"
    embopenai "github.com/joakimcarlsson/ai/embeddings/openai"
    llmopenai "github.com/joakimcarlsson/ai/llm/openai"
    "github.com/joakimcarlsson/ai/model"
)

func main() {
    ctx := context.Background()
    connString := "postgres://postgres:password@localhost:5432/example?sslmode=disable"

    embedder := embopenai.NewEmbedding(
        embopenai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
        embopenai.WithModel(model.OpenAIEmbeddingModels[model.TextEmbedding3Small]),
    )

    llmClient := llmopenai.NewLLM(
        llmopenai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
        llmopenai.WithModel(model.OpenAIModels[model.GPT4o]),
    )

    sessionStore, err := pgsess.SessionStore(ctx, connString)
    if err != nil {
        log.Fatal(err)
    }

    memoryStore, err := pgvectormem.MemoryStore(ctx, connString, embedder)
    if err != nil {
        log.Fatal(err)
    }

    myAgent := agent.New(llmClient,
        agent.WithSystemPrompt("You are a personal assistant with memory."),
        agent.WithSession("conv-1", sessionStore),
        agent.WithMemory("alice", memoryStore,
            memory.AutoExtract(),
            memory.AutoDedup(),
        ),
    )

    response, err := myAgent.Chat(ctx, "Hi! My name is Alice and I love Italian food.")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(response.Content)
}