aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 更新响应模板

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-06-13 13:43:34 +08:00
parent 49807a3bf3
commit 6034a3539b
7 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/api_attribute_retrieve.py
View File

@@ -25,7 +25,7 @@ def attribute_recognition(request_item: list[AttributeRecognitionModel]):
message = e message = e
data = e data = e
logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}") logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}")
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}
# 类别识别 # 类别识别
@@ -42,4 +42,4 @@ def category_recognition(request_item: list[CategoryRecognitionModel]):
message = e message = e
data = e data = e
logger.warning(f"category_recognition Run Exception @@@@@@:{e}") logger.warning(f"category_recognition Run Exception @@@@@@:{e}")
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}

2
app/api/api_chat_robot.py
View File

@@ -24,4 +24,4 @@ def chat_robot(request_data: ChatRobotModel):
data = str(e) data = str(e)
logger.warning(f"chat_robot Run Exception @@@@@@:{e}") logger.warning(f"chat_robot Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data}) logger.info({"code": code, "message": message, "data": data})
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}

2
app/api/api_design.py
View File

@@ -25,4 +25,4 @@ def design(request_data: DesignModel):
data = str(e) data = str(e)
logger.warning(f"design Run Exception @@@@@@:{e}") logger.warning(f"design Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data}) logger.info({"code": code, "message": message, "data": data})
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}

2
app/api/api_design_pre_processing.py
View File

@@ -26,4 +26,4 @@ def design_pre_processing(request_data: DesignPreProcessingModel):
data = str(e) data = str(e)
logger.warning(f"design Run Exception @@@@@@:{e}") logger.warning(f"design Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data}) logger.info({"code": code, "message": message, "data": data})
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}

12
app/api/api_generate_image.py
View File

@@ -23,13 +23,13 @@ def generate_image(request_item: GenerateImageModel, background_tasks: Backgroun
code = 400 code = 400
message = e message = e
logger.warning(e) logger.warning(e)
return {"code": code, "message": message} return {"code": code, "msg": message}
@router.get("/generate_cancel/{tasks_id}>") @router.get("/generate_cancel/{tasks_id}>")
def generate_image(tasks_id): def generate_image(tasks_id):
result = generate_image_infer_cancel(tasks_id) result = generate_image_infer_cancel(tasks_id)
return {"code": 200, "message": "OK!", "data": result['data']} return {"code": 200, "msg": "OK!", "data": result['data']}
'''single logo''' '''single logo'''
@@ -47,13 +47,13 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_
code = 400 code = 400
message = e message = e
logger.warning(e) logger.warning(e)
return {"code": code, "message": message} return {"code": code, "msg": message}
@router.get("/generate_single_logo_cancel/{tasks_id}>") @router.get("/generate_single_logo_cancel/{tasks_id}>")
def generate_single_logo_image(tasks_id): def generate_single_logo_image(tasks_id):
result = generate_single_logo_cancel(tasks_id) result = generate_single_logo_cancel(tasks_id)
return {"code": 200, "message": "OK!", "data": result['data']} return {"code": 200, "msg": "OK!", "data": result['data']}
'''product image''' '''product image'''
@@ -71,10 +71,10 @@ def generate_product_image(request_item: GenerateProductImageModel, background_t
code = 400 code = 400
message = e message = e
logger.warning(e) logger.warning(e)
return {"code": code, "message": message} return {"code": code, "msg": message}
@router.get("/generate_product_image_cancel_cancel/{tasks_id}>") @router.get("/generate_product_image_cancel_cancel/{tasks_id}>")
def generate_single_logo_image(tasks_id): def generate_single_logo_image(tasks_id):
result = generate_product_image_cancel(tasks_id) result = generate_product_image_cancel(tasks_id)
return {"code": 200, "message": "OK!", "data": result['data']} return {"code": 200, "msg": "OK!", "data": result['data']}

2
app/api/api_prompt_generation.py
View File

@@ -25,4 +25,4 @@ def prompt_generation(request_data: PromptGenerationImageModel):
data = str(e) data = str(e)
logger.warning(f"prompt_generation Run Exception @@@@@@:{e}") logger.warning(f"prompt_generation Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data}) logger.info({"code": code, "message": message, "data": data})
return {"code": code, "message": message, "data": data} return {"code": code, "msg": message, "data": data}

4
app/api/api_super_resolution.py
View File

@@ -21,10 +21,10 @@ def super_resolution(request_item: SuperResolutionModel, background_tasks: Backg
code = 400 code = 400
message = e message = e
logger.warning(e) logger.warning(e)
return {"code": code, "message": message} return {"code": code, "msg": message}
@router.get("/sr_cancel/{tasks_id}>") @router.get("/sr_cancel/{tasks_id}>")
def super_resolution(tasks_id): def super_resolution(tasks_id):
result = infer_cancel(tasks_id) result = infer_cancel(tasks_id)
return {"code": 200, "message": result['message'], "data": result['data']} return {"code": 200, "msg": result['message'], "data": result['data']}