000001_initial.up.sql 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. -- Sessions
  2. CREATE TABLE IF NOT EXISTS sessions (
  3. id TEXT PRIMARY KEY,
  4. parent_session_id TEXT,
  5. title TEXT NOT NULL,
  6. message_count INTEGER NOT NULL DEFAULT 0 CHECK (message_count >= 0),
  7. prompt_tokens INTEGER NOT NULL DEFAULT 0 CHECK (prompt_tokens >= 0),
  8. completion_tokens INTEGER NOT NULL DEFAULT 0 CHECK (completion_tokens>= 0),
  9. cost REAL NOT NULL DEFAULT 0.0 CHECK (cost >= 0.0),
  10. updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  11. created_at INTEGER NOT NULL -- Unix timestamp in milliseconds
  12. );
  13. CREATE TRIGGER IF NOT EXISTS update_sessions_updated_at
  14. AFTER UPDATE ON sessions
  15. BEGIN
  16. UPDATE sessions SET updated_at = strftime('%s', 'now')
  17. WHERE id = new.id;
  18. END;
  19. -- Messages
  20. CREATE TABLE IF NOT EXISTS messages (
  21. id TEXT PRIMARY KEY,
  22. session_id TEXT NOT NULL,
  23. role TEXT NOT NULL,
  24. parts TEXT NOT NULL default '[]',
  25. model TEXT,
  26. created_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  27. updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  28. finished_at INTEGER, -- Unix timestamp in milliseconds
  29. FOREIGN KEY (session_id) REFERENCES sessions (id) ON DELETE CASCADE
  30. );
  31. CREATE INDEX IF NOT EXISTS idx_messages_session_id ON messages (session_id);
  32. CREATE TRIGGER IF NOT EXISTS update_messages_updated_at
  33. AFTER UPDATE ON messages
  34. BEGIN
  35. UPDATE messages SET updated_at = strftime('%s', 'now')
  36. WHERE id = new.id;
  37. END;
  38. CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_insert
  39. AFTER INSERT ON messages
  40. BEGIN
  41. UPDATE sessions SET
  42. message_count = message_count + 1
  43. WHERE id = new.session_id;
  44. END;
  45. CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_delete
  46. AFTER DELETE ON messages
  47. BEGIN
  48. UPDATE sessions SET
  49. message_count = message_count - 1
  50. WHERE id = old.session_id;
  51. END;