| 16 | from src.models.vec_model.vec_model import VectorizeModel |
| 17 | |
| 18 | class VecModelHandler(RequestHandler): |
| 19 | |
| 20 | def initialize(self, vec_model:VectorizeModel): |
| 21 | self.vec_model = vec_model |
| 22 | # request_body: |
| 23 | # { |
| 24 | # "query": "什么人不能吃花生" |
| 25 | # } |
| 26 | |
| 27 | # response_body = { |
| 28 | # "answer":[{ |
| 29 | # "query":"query", |
| 30 | # "answer":"answer", |
| 31 | # "score":"score" |
| 32 | # }] |
| 33 | # } |
| 34 | |
| 35 | async def post(self): |
| 36 | vec_result = self.vec_model.predict_vec_request(json_decode(self.request.body).get("query", "")) |
| 37 | response_body = {"vec_result": vec_result} |
| 38 | self.write(response_body) |
| 39 | |
| 40 | def StartVecModelHandler(request_config: dict, vec_model: VectorizeModel): |
| 41 | # 启动TestClass服务的进程 |
nothing calls this directly
no outgoing calls
no test coverage detected