1 files changed, 5 insertions, 3 deletions
diff --git a/internal/db/sql/sessions.sql b/internal/db/sql/sessions.sql
index e90e5e328..2be8b7ccc 100644
--- a/internal/db/sql/sessions.sql
+++ b/internal/db/sql/sessions.sql
@@ -1,10 +1,10 @@
--- sqlfluff:dialect:sqlite
 -- name: CreateSession :one
 INSERT INTO sessions (
     id,
     title,
     message_count,
-    tokens,
+    prompt_tokens,
+    completion_tokens,
     cost,
     updated_at,
     created_at
@@ -14,6 +14,7 @@ INSERT INTO sessions (
     ?,
     ?,
     ?,
+    ?,
     strftime('%s', 'now'),
     strftime('%s', 'now')
 ) RETURNING *;
@@ -32,7 +33,8 @@ ORDER BY created_at DESC;
 UPDATE sessions
 SET
     title = ?,
-    tokens = ?,
+    prompt_tokens = ?,
+    completion_tokens = ?,
     cost = ?
 WHERE id = ?
 RETURNING *;