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
|
"""API for data used to generate menus"""
# pylint: skip-file
from flask import jsonify, request, Blueprint, current_app
from gn3.llms.process import get_gnqa
from gn3.llms.process import rate_document
GnQNA = Blueprint("GnQNA", __name__)
@GnQNA.route("/gnqna", methods=["POST"])
def gnqa():
query = request.json.get("querygnqa", "")
if not query:
return jsonify({"error": "querygnqa is missing in the request"}), 400
try:
auth_token = current_app.config.get("FAHAMU_AUTH_TOKEN")
task_id, answer, refs = get_gnqa(
query, auth_token)
return jsonify({
"task_id": task_id,
"query": query,
"answer": answer,
"references": refs
})
except Exception as error:
return jsonify({"query": query, "error": "Internal server error"}), 500
@GnQNA.route("/rating/<task_id>/<doc_id>/<int:rating>", methods=["POST"])
def rating(task_id, doc_id, rating):
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
|
