000001_initial.up.sql 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  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. -- Files
  20. CREATE TABLE IF NOT EXISTS files (
  21. id TEXT PRIMARY KEY,
  22. session_id TEXT NOT NULL,
  23. path TEXT NOT NULL,
  24. content TEXT NOT NULL,
  25. version TEXT NOT NULL,
  26. created_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  27. updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  28. FOREIGN KEY (session_id) REFERENCES sessions (id) ON DELETE CASCADE
  29. );
  30. CREATE INDEX IF NOT EXISTS idx_files_session_id ON files (session_id);
  31. CREATE INDEX IF NOT EXISTS idx_files_path ON files (path);
  32. CREATE TRIGGER IF NOT EXISTS update_files_updated_at
  33. AFTER UPDATE ON files
  34. BEGIN
  35. UPDATE files SET updated_at = strftime('%s', 'now')
  36. WHERE id = new.id;
  37. END;
  38. -- Messages
  39. CREATE TABLE IF NOT EXISTS messages (
  40. id TEXT PRIMARY KEY,
  41. session_id TEXT NOT NULL,
  42. role TEXT NOT NULL,
  43. parts TEXT NOT NULL default '[]',
  44. model TEXT,
  45. created_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  46. updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds
  47. finished_at INTEGER, -- Unix timestamp in milliseconds
  48. FOREIGN KEY (session_id) REFERENCES sessions (id) ON DELETE CASCADE
  49. );
  50. CREATE INDEX IF NOT EXISTS idx_messages_session_id ON messages (session_id);
  51. CREATE TRIGGER IF NOT EXISTS update_messages_updated_at
  52. AFTER UPDATE ON messages
  53. BEGIN
  54. UPDATE messages SET updated_at = strftime('%s', 'now')
  55. WHERE id = new.id;
  56. END;
  57. CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_insert
  58. AFTER INSERT ON messages
  59. BEGIN
  60. UPDATE sessions SET
  61. message_count = message_count + 1
  62. WHERE id = new.session_id;
  63. END;
  64. CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_delete
  65. AFTER DELETE ON messages
  66. BEGIN
  67. UPDATE sessions SET
  68. message_count = message_count - 1
  69. WHERE id = old.session_id;
  70. END;