From bbb5810e934907f3ed2c936f1047f025da52b5c3 Mon Sep 17 00:00:00 2001 From: Alexander_Kabui Date: Wed, 17 Jan 2024 16:28:27 +0300 Subject: Refactor llms.process: Rename getGNQA to get_gnqa --- gn3/api/llm.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/gn3/api/llm.py b/gn3/api/llm.py index 5fcc1f5..9c277c1 100644 --- a/gn3/api/llm.py +++ b/gn3/api/llm.py @@ -4,7 +4,7 @@ from flask import jsonify, request, Blueprint, current_app -from gn3.llms.process import getGNQA +from gn3.llms.process import get_gnqa from gn3.llms.process import rate_document @@ -19,7 +19,7 @@ def gnqa(): try: auth_token = current_app.config.get("FAHAMU_AUTH_TOKEN") - task_id, answer, refs = getGNQA( + task_id, answer, refs = get_gnqa( query, auth_token) return jsonify({ @@ -38,6 +38,8 @@ 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, -- cgit v1.2.3