Add default timestamp and and primary key for Rating table.

1 files changed, 2 insertions, 2 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)