2 files changed, 13 insertions, 1 deletions
diff --git a/internal/db/sql/messages.sql b/internal/db/sql/messages.sql
index a59cebe7d..475b23a86 100644
--- a/internal/db/sql/messages.sql
+++ b/internal/db/sql/messages.sql
@@ -9,6 +9,12 @@ FROM messages
 WHERE session_id = ?
 ORDER BY created_at ASC;
 
+-- name: ListMessagesBySessionAfter :many
+SELECT *
+FROM messages
+WHERE session_id = ? AND created_at > ?
+ORDER BY created_at ASC;
+
 -- name: CreateMessage :one
 INSERT INTO messages (
     id,
diff --git a/internal/db/sql/sessions.sql b/internal/db/sql/sessions.sql
index f065b5f56..81abebd3c 100644
--- a/internal/db/sql/sessions.sql
+++ b/internal/db/sql/sessions.sql
@@ -7,6 +7,8 @@ INSERT INTO sessions (
     prompt_tokens,
     completion_tokens,
     cost,
+    summary,
+    summarized_at,
     updated_at,
     created_at
 ) VALUES (
@@ -17,6 +19,8 @@ INSERT INTO sessions (
     ?,
     ?,
     ?,
+    ?,
+    ?,
     strftime('%s', 'now'),
     strftime('%s', 'now')
 ) RETURNING *;
@@ -38,7 +42,9 @@ SET
     title = ?,
     prompt_tokens = ?,
     completion_tokens = ?,
-    cost = ?
+    cost = ?,
+    summary = ?,
+    summarized_at = ?
 WHERE id = ?
 RETURNING *;