memory service schema added

2026-04-04 05:49:57 -07:00
parent ebbeecfa1a
commit 5d51aa9895
3 changed files with 90 additions and 0 deletions
--- a/packages/memory-service/src/db/schema.js
+++ b/packages/memory-service/src/db/schema.js
@@ -0,0 +1,63 @@
+const schema = `
+  CREATE TABLE IF NOT EXISTS sessions (
+    id          INTEGER PRIMARY KEY AUTOINCREMENT,
+    external_id TEXT    UNIQUE NOT NULL,
+    created_at  INTEGER NOT NULL DEFAULT (unixepoch()),
+    updated_at  INTEGER NOT NULL DEFAULT (unixepoch()),
+    metadata    TEXT
+  );
+
+  CREATE TABLE IF NOT EXISTS episodes (
+    id            INTEGER PRIMARY KEY AUTOINCREMENT,
+    session_id    INTEGER NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
+    user_message  TEXT NOT NULL,
+    ai_response   TEXT NOT NULL,
+    created_at    INTEGER NOT NULL DEFAULT (unixepoch()),
+    token_count   INTEGER,
+    metadata      TEXT
+  );
+
+  CREATE TABLE IF NOT EXISTS entities (
+    id         INTEGER PRIMARY KEY AUTOINCREMENT,
+    name       TEXT    NOT NULL,
+    type       TEXT    NOT NULL,
+    notes      TEXT,
+    created_at INTEGER NOT NULL DEFAULT (unixepoch()),
+    updated_at INTEGER NOT NULL DEFAULT (unixepoch()),
+    metadata   TEXT,
+    UNIQUE(name, type)
+  );
+
+  CREATE TABLE IF NOT EXISTS relationships (
+    id         INTEGER PRIMARY KEY AUTOINCREMENT,
+    from_id    INTEGER NOT NULL REFERENCES entities(id) ON DELETE CASCADE,
+    to_id      INTEGER NOT NULL REFERENCES entities(id) ON DELETE CASCADE,
+    label      TEXT    NOT NULL,
+    created_at INTEGER NOT NULL DEFAULT (unixepoch()),
+    metadata   TEXT,
+    UNIQUE(from_id, to_id, label)
+  );
+
+  CREATE TABLE IF NOT EXISTS summaries (
+    id            INTEGER PRIMARY KEY AUTOINCREMENT,
+    session_id    INTEGER REFERENCES sessions(id) ON DELETE CASCADE,
+    content       TEXT    NOT NULL,
+    episode_range TEXT,
+    created_at    INTEGER NOT NULL DEFAULT (unixepoch()),
+    metadata      TEXT
+  );
+
+  CREATE INDEX IF NOT EXISTS idx_episodes_session 
+    ON episodes(session_id);
+  CREATE INDEX IF NOT EXISTS idx_episodes_created 
+    ON episodes(created_at);
+  CREATE INDEX IF NOT EXISTS idx_entities_type 
+    ON entities(type);
+  CREATE INDEX IF NOT EXISTS idx_summaries_session 
+    ON summaries(session_id);
+
+  CREATE VIRTUAL TABLE IF NOT EXISTS episodes_fts 
+    USING fts5(user_message, ai_response, content=episodes, content_rowid=id);
+`;
+
+module.exports = schema;