1 files changed, 35 insertions, 12 deletions
diff --git a/gn3/api/llm.py b/gn3/api/llm.py
index 98ff4e2..aff01cb 100644
--- a/gn3/api/llm.py
+++ b/gn3/api/llm.py
@@ -11,9 +11,13 @@ from gn3.llms.process import get_gnqa
 from gn3.llms.process import rate_document
 from gn3.llms.process import get_user_queries
 from gn3.llms.process import fetch_query_results
+from gn3.auth.authorisation.oauth2.resource_server import require_oauth
+from gn3.auth import db
+
 
 from redis import Redis
 import json
+import sqlite3
 from datetime import timedelta
 
 GnQNA = Blueprint("GnQNA", __name__)
@@ -58,19 +62,38 @@ def gnqa():
         return jsonify({"query": query, "error": f"Request failed-{str(error)}"}), 500
 
 
-@GnQNA.route("/rating/<task_id>/<doc_id>/<int:rating>", methods=["POST"])
-def rating(task_id, doc_id, rating):
+@GnQNA.route("/rating/<task_id>", methods=["POST"])
+@require_oauth("profile")
+def rating(task_id):
     try:
-        results = rate_document(task_id, doc_id, rating,
-                                current_app.config.get("FAHAMU_AUTH_TOKEN"))
-
-        return jsonify({
-            **results,
-            "doc_id": doc_id,
-            "task_id": task_id,
-        }),
-    except Exception as error:
-        return jsonify({"error": str(error), doc_id: doc_id}), 500
+        with require_oauth.acquire("profile") as the_token:
+            user = the_token.user.user_id
+            results = request.json
+            user_id, query, answer, weight = (the_token.user.user_id,
+                                              results.get("query"),
+                                              results.get("answer"),
+                                              results.get("weight", 0))
+            with db.connection(current_app.config["GNQA_DB"]) as conn:
+                cursor = conn.cursor()
+                create_table = """CREATE TABLE IF NOT EXISTS Rating(
+                      user_id INTEGER NOT NULL,
+                      query TEXT NOT NULL,
+                      answer TEXT NOT NULL,
+                      weight INTEGER NOT NULL DEFAULT 0,
+                      task_id TEXT NOT NULL UNIQUE
+                      )"""
+                cursor.execute(create_table)
+                cursor.execute("""INSERT INTO Rating(user_id,query,answer,weight,task_id)
+                VALUES(?,?,?,?,?)
+                ON CONFLICT(task_id) DO UPDATE SET
+                weight=excluded.weight
+                """, (user_id, query, answer, weight, task_id))
+                return {
+                    "message": "success",
+                    "status": 0
+                }, 200
+    except sqlite3.Error as error:
+        raise error
 
 
 @GnQNA.route("/history/<query>", methods=["GET"])