From a45ab3e62df6e1dfcc6fff03916369fbdaf68ab8 Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Thu, 29 Aug 2024 11:05:39 +0300
Subject: Add default timestamp and and primary key for Rating table.

---
 gn3/api/llm.py               | 4 ++--
 sql/update/llm_db_update.sql | 2 +-
 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;
-- 
cgit v1.2.3