-
Notifications
You must be signed in to change notification settings - Fork 0
/
api_routes.py
90 lines (70 loc) · 2.95 KB
/
api_routes.py
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
import os
import subprocess
from datetime import datetime
import langcheck
import pytz
from dotenv import load_dotenv
from flask import Blueprint, jsonify, request
import database as db
from calculate_metrics import add_init_to_db, get_factual_consistency
from rag import RAG
api_routes_blueprint = Blueprint('api', __name__)
load_dotenv()
# Initialize the RAG system
rag_system = RAG()
@api_routes_blueprint.route('/api/chat', methods=['POST'])
@api_routes_blueprint.route('/api/chat_demo', methods=['POST'])
def chat():
user_message = request.get_json().get('message', '')
language = request.get_json().get('language', 'en')
if request.path == '/api/chat_demo':
# Get canned responses to speed up live demos
response_message, source = rag_system.query_demo(
user_message, language)
else:
response_message, source = rag_system.query(user_message, language)
# Compute the factual consistency score and add it along with the chat
# data to the db
factual_consistency_score, factual_consistency_explanation = get_factual_consistency(
response_message, source, language)
timestamp = datetime.now(
pytz.timezone('Asia/Tokyo')).strftime('%Y-%m-%d %H:%M:%S')
log_id = add_init_to_db(user_message, response_message, source, language,
factual_consistency_score,
factual_consistency_explanation, timestamp)
# Compute and log all the other metrics
subprocess.Popen(["python", "calculate_metrics.py", str(log_id)])
warning = factual_consistency_score < 0.5
return jsonify(response=response_message,
score=factual_consistency_score,
warning=warning,
source=source,
id=log_id)
@api_routes_blueprint.route('/api/ref_metric', methods=['POST'])
def get_reference_based_metric():
log_id = request.get_json().get('log_id', '')
reference_text = request.get_json().get('reference')
# Update the status before updating the record
db.update_chatlog_by_id({'status': 'new'}, log_id)
# Compute the metrics
subprocess.Popen([
"python", "calculate_reference_metrics.py",
str(log_id), reference_text
])
return jsonify(success=True)
@api_routes_blueprint.route('/api/logs', methods=['GET'])
def logs():
page = int(request.args.get('page', 1))
per_page = 10
offset = (page - 1) * per_page
return jsonify(logs=db.get_chatlogs_and_metrics(per_page, offset))
@api_routes_blueprint.route('/api/metrics/<log_id>', methods=['GET'])
def metrics_endpoint(log_id):
metrics_data = db.get_metrics_by_log_id(log_id)
if metrics_data is None:
return jsonify({"error": "No metrics available"}), 400
chatlog_data = db.get_chatlog_by_id(log_id)
if chatlog_data is None:
return jsonify({"error": "No chat logs available"}), 400
metrics_data['status'] = chatlog_data['status']
return jsonify(metrics_data)