aboutsummaryrefslogtreecommitdiff
path: root/gn3/api
diff options
context:
space:
mode:
Diffstat (limited to 'gn3/api')
-rw-r--r--gn3/api/llm.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/gn3/api/llm.py b/gn3/api/llm.py
index 836a880..a617faf 100644
--- a/gn3/api/llm.py
+++ b/gn3/api/llm.py
@@ -75,12 +75,11 @@ def rate_queries(task_id):
PRIMARY KEY(task_id))"""
cursor.execute(create_table)
cursor.execute("""INSERT INTO Rating(user_id, query,
- answer, weight, task_id, created_at)
- VALUES(?, ?, ?, ?, ?, ?)
+ answer, weight, task_id)
+ VALUES(?, ?, ?, ?, ?)
ON CONFLICT(task_id) DO UPDATE SET
weight=excluded.weight
- """, (str(user_id), query, answer, weight, task_id,
- datetime.now(timezone.utc)))
+ """, (str(user_id), query, answer, weight, task_id))
return {
"message": "You have successfully rated this query.Thank you!"
}, 200