添加请求参数:ScoreThreshold

main
fanpt 1 year ago
parent 9d0fa92b76
commit 2ca7196d2c

@ -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

Loading…
Cancel
Save