feat 更新响应模板

fix
This commit is contained in:
zhouchengrong
2024-06-13 14:31:14 +08:00
parent 6034a3539b
commit b012b91613
11 changed files with 120 additions and 100 deletions

View File

@@ -1,8 +1,10 @@
import json
import logging
import time
from fastapi import APIRouter
from fastapi import APIRouter, HTTPException
from app.schemas.chat_robot import ChatRobotModel
from app.schemas.response_template import ResponseModel
from app.service.chat_robot.script.main import chat
router = APIRouter()
@@ -13,15 +15,9 @@ logger = logging.getLogger()
def chat_robot(request_data: ChatRobotModel):
try:
logger.info(f"chat_robot request item is : @@@@@@:{request_data}")
code = 200
message = "OK!"
start_time = time.time()
data = chat(post_data=request_data)
logger.info(f"chat_robot Run time is @@@@@@:{time.time() - start_time}")
logger.info(f"chat_robot response @@@@@@:{json.dumps(data, indent=4)}")
except Exception as e:
code = 400
message = str(e)
data = str(e)
logger.warning(f"chat_robot Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data})
return {"code": code, "msg": message, "data": data}
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)