MCPcopy Create free account
hub / github.com/ZBayes/basic_rag / VecModelHandler

Class VecModelHandler

src/server/handlers/vec_model_handler.py:18–38  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

16from src.models.vec_model.vec_model import VectorizeModel
17
18class 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
40def StartVecModelHandler(request_config: dict, vec_model: VectorizeModel):
41 # 启动TestClass服务的进程

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected