08268b32b8
Add `facts` table to schema with columns for id, agent_id, content, category, confidence, source_session, and created_at. Implement store_facts() and get_top_facts() on MemoryStore using upsert-by-id and confidence-desc ordering. Facts extracted from conversations are now durable across sessions. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
90 lines
2.6 KiB
Rust
90 lines
2.6 KiB
Rust
//! Database schema definitions
|
|
|
|
/// Current schema version
|
|
pub const SCHEMA_VERSION: i32 = 1;
|
|
|
|
/// Schema creation SQL
|
|
pub const CREATE_SCHEMA: &str = r#"
|
|
-- Agents table
|
|
CREATE TABLE IF NOT EXISTS agents (
|
|
id TEXT PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
config TEXT NOT NULL,
|
|
created_at TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL
|
|
);
|
|
|
|
-- Sessions table
|
|
CREATE TABLE IF NOT EXISTS sessions (
|
|
id TEXT PRIMARY KEY,
|
|
agent_id TEXT NOT NULL,
|
|
created_at TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL,
|
|
FOREIGN KEY (agent_id) REFERENCES agents(id) ON DELETE CASCADE
|
|
);
|
|
|
|
-- Messages table
|
|
CREATE TABLE IF NOT EXISTS messages (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
session_id TEXT NOT NULL,
|
|
seq INTEGER NOT NULL,
|
|
content TEXT NOT NULL,
|
|
created_at TEXT NOT NULL,
|
|
FOREIGN KEY (session_id) REFERENCES sessions(id) ON DELETE CASCADE,
|
|
UNIQUE(session_id, seq)
|
|
);
|
|
|
|
-- KV Store table
|
|
CREATE TABLE IF NOT EXISTS kv_store (
|
|
agent_id TEXT NOT NULL,
|
|
key TEXT NOT NULL,
|
|
value TEXT NOT NULL,
|
|
updated_at TEXT NOT NULL,
|
|
PRIMARY KEY (agent_id, key),
|
|
FOREIGN KEY (agent_id) REFERENCES agents(id) ON DELETE CASCADE
|
|
);
|
|
|
|
-- Schema version table
|
|
CREATE TABLE IF NOT EXISTS schema_version (
|
|
version INTEGER PRIMARY KEY
|
|
);
|
|
|
|
-- Hand execution runs table
|
|
CREATE TABLE IF NOT EXISTS hand_runs (
|
|
id TEXT PRIMARY KEY,
|
|
hand_name TEXT NOT NULL,
|
|
trigger_source TEXT NOT NULL,
|
|
params TEXT NOT NULL,
|
|
status TEXT NOT NULL DEFAULT 'pending',
|
|
result TEXT,
|
|
error TEXT,
|
|
duration_ms INTEGER,
|
|
created_at TEXT NOT NULL,
|
|
started_at TEXT,
|
|
completed_at TEXT
|
|
);
|
|
|
|
-- Indexes
|
|
CREATE INDEX IF NOT EXISTS idx_sessions_agent ON sessions(agent_id);
|
|
CREATE INDEX IF NOT EXISTS idx_messages_session ON messages(session_id);
|
|
CREATE INDEX IF NOT EXISTS idx_kv_agent ON kv_store(agent_id);
|
|
CREATE INDEX IF NOT EXISTS idx_hand_runs_hand ON hand_runs(hand_name);
|
|
CREATE INDEX IF NOT EXISTS idx_hand_runs_status ON hand_runs(status);
|
|
CREATE INDEX IF NOT EXISTS idx_hand_runs_created ON hand_runs(created_at);
|
|
|
|
-- Structured facts table (extracted from conversations)
|
|
CREATE TABLE IF NOT EXISTS facts (
|
|
id TEXT PRIMARY KEY,
|
|
agent_id TEXT NOT NULL,
|
|
content TEXT NOT NULL,
|
|
category TEXT NOT NULL,
|
|
confidence REAL NOT NULL,
|
|
source_session TEXT,
|
|
created_at INTEGER NOT NULL,
|
|
FOREIGN KEY (agent_id) REFERENCES agents(id) ON DELETE CASCADE
|
|
);
|
|
CREATE INDEX IF NOT EXISTS idx_facts_agent ON facts(agent_id);
|
|
CREATE INDEX IF NOT EXISTS idx_facts_category ON facts(agent_id, category);
|
|
CREATE INDEX IF NOT EXISTS idx_facts_confidence ON facts(agent_id, confidence DESC);
|
|
"#;
|