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:
llm_db_path = current_app.config["LLM_DB_PATH"]
with (require_oauth.acquire("profile") as token,
db.connection(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:
llm_db_path = current_app.config["LLM_DB_PATH"]
with (require_oauth.acquire("profile user") as token,
db.connection(llm_db_path) as conn):
cursor = conn.cursor()
if request.args.get("search_term"):
query = """SELECT results from history Where task_id=? and user_id=?"""
cursor.execute(query, (request.args.get(
"search_term"), str(token.user.user_id),))
return dict(cursor.fetchone())
query = """SELECT task_id,query from history WHERE user_id=?"""
cursor.execute(query, (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
|
