1 files changed, 3 insertions, 10 deletions
diff --git a/internal/db/sql/messages.sql b/internal/db/sql/messages.sql
index 0674e62c1..64571158f 100644
--- a/internal/db/sql/messages.sql
+++ b/internal/db/sql/messages.sql
@@ -14,25 +14,18 @@ INSERT INTO messages (
     id,
     session_id,
     role,
-    finished,
-    content,
-    tool_calls,
-    tool_results,
+    parts,
     created_at,
     updated_at
 ) VALUES (
-    ?, ?, ?, ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
+    ?, ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
 )
 RETURNING *;
 
 -- name: UpdateMessage :exec
 UPDATE messages
 SET
-    content = ?,
-    thinking = ?,
-    tool_calls = ?,
-    tool_results = ?,
-    finished = ?,
+    parts = ?,
     updated_at = strftime('%s', 'now')
 WHERE id = ?;