修改为无阈值参数

main
fanpt 1 year ago
parent 2ca7196d2c
commit c4f34445d5

@ -29,7 +29,7 @@ logger = logging.getLogger(__name__)
class QuestionRequest(BaseModel):
question: str
ScoreThreshold: float
class QuestionResponse(BaseModel):
@ -141,13 +141,14 @@ 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, request.ScoreThreshold)
result = match_query(qa_service, query, top_k, score_threshold)
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