ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

update tesou.py

Browse Source
This commit is contained in:
ACBBZ
2024-03-25 08:38:50 +00:00
parent c3dc2966f0
commit ec10081276
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/blackbox/tesou.py
View File

@ -17,7 +17,7 @@ class Tesou(Blackbox):
# 用户输入的数据格式为：[{"id": "123", "prompt": "叉烧饭，帮我查询叉烧饭的介绍"}] # 用户输入的数据格式为：[{"id": "123", "prompt": "叉烧饭，帮我查询叉烧饭的介绍"}]
def processing(self, id, prompt) -> str: def processing(self, id, prompt) -> str:
url = 'http://120.196.116.194:48891/chat/' url = 'http://120.196.116.194:48891/'
message = { message = {
"user_id": id, "user_id": id,
@ -30,12 +30,10 @@ class Tesou(Blackbox):
async def fast_api_handler(self, request: Request) -> Response: async def fast_api_handler(self, request: Request) -> Response:
try: try:
data = await request.json() data = await request.json()
print(data)
except: except:
return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
user_id = data.get("id") user_id = data.get("id")
user_prompt = data.get("prompt") user_prompt = data.get("prompt")
print(user_id, user_prompt)
if user_prompt is None: if user_prompt is None:
return JSONResponse(content={"error": "question is required"}, status_code=status.HTTP_400_BAD_REQUEST) return JSONResponse(content={"error": "question is required"}, status_code=status.HTTP_400_BAD_REQUEST)
return JSONResponse(content={"Response": self.processing(user_id, user_prompt)}, status_code=status.HTTP_200_OK) return JSONResponse(content={"Response": self.processing(user_id, user_prompt)}, status_code=status.HTTP_200_OK)