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,9 +1,11 @@
import json
import logging
import time
from fastapi import APIRouter
from fastapi import APIRouter, HTTPException
from app.schemas.prompt_generation import PromptGenerationImageModel
from app.schemas.response_template import ResponseModel
from app.service.prompt_generation.chatgpt_for_translation import translate_to_en
router = APIRouter()
@@ -13,16 +15,10 @@ logger = logging.getLogger()
@router.post("/translateToEN")
def prompt_generation(request_data: PromptGenerationImageModel):
try:
logger.info(f"prompt_translate to English request data : @@@@@@:{request_data}")
code = 200
message = "OK!"
start_time = time.time()
logger.info(f"prompt_generation request item is : @@@@@@:{request_data}")
data = translate_to_en(request_data.text)
logger.info(f"prompt_generation Run time is @@@@@@:{time.time() - start_time}")
logger.info(f"prompt_generation response @@@@@@:{json.dumps(data, indent=4)}")
except Exception as e:
code = 400
message = str(e)
data = str(e)
logger.warning(f"prompt_generation 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)