From e7258e38aeb46281fda474b8b7fcc3eee35edd9f Mon Sep 17 00:00:00 2001
From: Kujtim Hoxha <kujtimii.h@gmail.com>
Date: Sun, 23 Mar 2025 22:25:31 +0100
Subject: initial agent setup

---
 internal/db/sql/sessions.sql | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'internal/db/sql/sessions.sql')

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 *;
-- 
cgit v1.2.3

