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