From 2ca7196d2c9e6ea6bfd5caa82f9a842ce04bbcdf Mon Sep 17 00:00:00 2001
From: fanpt <320622572@qq.com>
Date: Thu, 21 Mar 2024 15:19:16 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AF=B7=E6=B1=82=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0=EF=BC=9AScoreThreshold?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fast_api.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/fast_api.py b/fast_api.py
index f5215bd..9eb0b3b 100644
--- a/fast_api.py
+++ b/fast_api.py
@@ -29,6 +29,7 @@ logger = logging.getLogger(__name__)
 
 class QuestionRequest(BaseModel):
     question: str
+    ScoreThreshold: float
 
 
 class QuestionResponse(BaseModel):
@@ -140,14 +141,13 @@ def match_question(request: QuestionRequest):
         newest = recent_knowledge_bases[-1]
 
         top_k = 3
-        score_threshold = 0.1
 
         device = None
         qa_service = QAService(newest, device)
 
-        result = match_query(qa_service, query, top_k, score_threshold)
+        result = match_query(qa_service, query, top_k, request.ScoreThreshold)
 
-        response = QuestionResponse(code=200, msg="success", data=result)
+        response = QuestionResponse(code=200, msg="success", data=[result])
         stop_time = time.time()
         duration = stop_time - start_time