aboutsummaryrefslogtreecommitdiff
path: root/gn3/api/llm.py
blob: 5ad58cb878a09e3d90c1123214727ae4a6eeac2b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
"""Api endpoints for gnqa"""
import json
import sqlite3
from authlib.integrations.flask_oauth2.errors import _HTTPException

from flask import Blueprint
from flask import current_app
from flask import jsonify
from flask import request

from gn3.llms.process import get_gnqa
from gn3.llms.errors import LLMError
from gn3.auth.authorisation.oauth2.resource_server import require_oauth
from gn3.auth.authorisation.errors import AuthorisationError
from gn3.auth import db

gnqa = Blueprint("gnqa", __name__)


@gnqa.route("/search", methods=["POST"])
def search():
    """Main gnqa endpoint"""
    query = request.json.get("querygnqa", "")
    if not query:
        return jsonify({"error": "querygnqa is missing in the request"}), 400
    try:
        fahamu_token = current_app.config.get("FAHAMU_AUTH_TOKEN")
        if not fahamu_token:
            raise LLMError(
                "Request failed:an LLM authorisation token  is required ",
                query=query)
        task_id, answer, refs = get_gnqa(
            query, fahamu_token, current_app.config.get("DATA_DIR"))
        response = {
            "task_id": task_id,
            "query": query,
            "answer": answer,
            "references": refs
        }
        try:
            with (db.connection(current_app.config["LLM_DB_PATH"]) as conn,
                  require_oauth.acquire("profile user") as token):
                cursor = conn.cursor()
                cursor.execute("""CREATE TABLE IF NOT EXISTS
                history(user_id TEXT NOT NULL,
                task_id TEXT NOT NULL,
                query  TEXT NOT NULL,
                results  TEXT,
                PRIMARY KEY(task_id)) WITHOUT ROWID""")
                cursor.execute(
                    """INSERT INTO history(user_id,task_id,query,results)
                    VALUES(?,?,?,?)
                    """, (str(token.user.user_id), str(task_id["task_id"]),
                          query,
                          json.dumps(response))
                )
            return response
        except _HTTPException as httpe:
            raise AuthorisationError("Authentication is required.") from httpe
        except sqlite3.Error as error:
            raise sqlite3.OperationalError(*error.args) from error
    except LLMError as error:
        raise LLMError(f"request failed for query {str(error.args[-1])}",
                       query=query) from error


@gnqa.route("/rating/<task_id>", methods=["POST"])
@require_oauth("profile")
def rate_queries(task_id):
    """Endpoint for rating qnqa query and answer"""
    try:
        with (require_oauth.acquire("profile") as token,
              db.connection(current_app.config["LLM_DB_PATH"]) as conn):

            results = request.json
            user_id, query, answer, weight = (token.user.user_id,
                                              results.get("query"),
                                              results.get("answer"),
                                              results.get("weight", 0))
            cursor = conn.cursor()
            create_table = """CREATE TABLE IF NOT EXISTS Rating(
                  user_id TEXT 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
            """, (str(user_id), query, answer, weight, task_id))
        return {
            "message":
            "You have successfully rated this query:Thank you!!"
        }, 200
    except sqlite3.Error as error:
        raise sqlite3.OperationalError(*error.args) from error
    except _HTTPException as httpe:
        raise AuthorisationError("Authentication is required") from httpe


@gnqa.route("/history", methods=["GET"])
@require_oauth("profile user")
def fetch_prev_history():
    """ api method to fetch search query records"""
    try:
        with (require_oauth.acquire("profile user") as token,
              db.connection(current_app.config["LLM_DB_PATH"]) as conn):
            cursor = conn.cursor()
            if request.args.get("search_term"):
                cursor.execute(
                    """SELECT results from history Where task_id=? and user_id=?""",
                    (request.args.get("search_term"),
                     str(token.user.user_id),))
                return dict(cursor.fetchone())
            cursor.execute(
                """SELECT task_id,query from history WHERE user_id=?""",
                (str(token.user.user_id),))
            return jsonify([dict(item) for item in cursor.fetchall()])
    except sqlite3.Error as error:
        raise sqlite3.OperationalError(*error.args) from error
    except _HTTPException as httpe:
        raise AuthorisationError("Authorization is required") from httpe