Add default timestamp and and primary key for Rating table.

2 files changed, 3 insertions, 3 deletions

diff --git a/gn3/api/llm.py b/gn3/api/llm.py
index 5901ef5..952a5b9 100644
--- a/gn3/api/llm.py
+++ b/gn3/api/llm.py
@@ -71,8 +71,8 @@ def rate_queries(task_id):
               answer TEXT NOT NULL,
               weight INTEGER NOT NULL DEFAULT 0,
               task_id TEXT NOT NULL UNIQUE,
-              created_at TIMESTAMP
-              )"""
+              created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+              PRIMARY KEY(task_id))"""
         cursor.execute(create_table)
         cursor.execute("""INSERT INTO Rating(user_id, query,
         answer, weight, task_id, created_at)
diff --git a/sql/update/llm_db_update.sql b/sql/update/llm_db_update.sql
index 6608a90..a4eb848 100644
--- a/sql/update/llm_db_update.sql
+++ b/sql/update/llm_db_update.sql
@@ -34,4 +34,4 @@ CREATE TABLE IF NOT EXISTS history (
 CREATE INDEX IF NOT EXISTS idx_tbl_history_cols_task_id_user_id
 ON history (task_id, user_id);
 
-ALTER TABLE Rating ADD COLUMN created_at TIMESTAMP;
+ALTER TABLE Rating ADD COLUMN created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP;