From 6034a3539b892ca2b552565f5e705895e7e351d8 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 13 Jun 2024 13:43:34 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20=E6=9B=B4=E6=96=B0=E5=93=8D=E5=BA=94?= =?UTF-8?q?=E6=A8=A1=E6=9D=BF=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/api_attribute_retrieve.py | 4 ++-- app/api/api_chat_robot.py | 2 +- app/api/api_design.py | 2 +- app/api/api_design_pre_processing.py | 2 +- app/api/api_generate_image.py | 12 ++++++------ app/api/api_prompt_generation.py | 2 +- app/api/api_super_resolution.py | 4 ++-- 7 files changed, 14 insertions(+), 14 deletions(-) diff --git a/app/api/api_attribute_retrieve.py b/app/api/api_attribute_retrieve.py index 5b18ec5..7dd4afb 100644 --- a/app/api/api_attribute_retrieve.py +++ b/app/api/api_attribute_retrieve.py @@ -25,7 +25,7 @@ def attribute_recognition(request_item: list[AttributeRecognitionModel]): message = e data = 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 data = e logger.warning(f"category_recognition Run Exception @@@@@@:{e}") - return {"code": code, "message": message, "data": data} + return {"code": code, "msg": message, "data": data} diff --git a/app/api/api_chat_robot.py b/app/api/api_chat_robot.py index d919902..a0158da 100644 --- a/app/api/api_chat_robot.py +++ b/app/api/api_chat_robot.py @@ -24,4 +24,4 @@ def chat_robot(request_data: ChatRobotModel): data = str(e) logger.warning(f"chat_robot Run Exception @@@@@@:{e}") logger.info({"code": code, "message": message, "data": data}) - return {"code": code, "message": message, "data": data} + return {"code": code, "msg": message, "data": data} diff --git a/app/api/api_design.py b/app/api/api_design.py index 79cf7e1..a224067 100644 --- a/app/api/api_design.py +++ b/app/api/api_design.py @@ -25,4 +25,4 @@ def design(request_data: DesignModel): data = str(e) logger.warning(f"design Run Exception @@@@@@:{e}") logger.info({"code": code, "message": message, "data": data}) - return {"code": code, "message": message, "data": data} \ No newline at end of file + return {"code": code, "msg": message, "data": data} \ No newline at end of file diff --git a/app/api/api_design_pre_processing.py b/app/api/api_design_pre_processing.py index 91bba6e..38d051a 100644 --- a/app/api/api_design_pre_processing.py +++ b/app/api/api_design_pre_processing.py @@ -26,4 +26,4 @@ def design_pre_processing(request_data: DesignPreProcessingModel): data = str(e) logger.warning(f"design Run Exception @@@@@@:{e}") logger.info({"code": code, "message": message, "data": data}) - return {"code": code, "message": message, "data": data} + return {"code": code, "msg": message, "data": data} diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index 9c979f1..1d2f3a9 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -23,13 +23,13 @@ def generate_image(request_item: GenerateImageModel, background_tasks: Backgroun code = 400 message = e logger.warning(e) - return {"code": code, "message": message} + return {"code": code, "msg": message} @router.get("/generate_cancel/{tasks_id}>") def generate_image(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''' @@ -47,13 +47,13 @@ def generate_single_logo(request_item: GenerateSingleLogoImageModel, background_ code = 400 message = e logger.warning(e) - return {"code": code, "message": message} + return {"code": code, "msg": message} @router.get("/generate_single_logo_cancel/{tasks_id}>") def generate_single_logo_image(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''' @@ -71,10 +71,10 @@ def generate_product_image(request_item: GenerateProductImageModel, background_t code = 400 message = e logger.warning(e) - return {"code": code, "message": message} + return {"code": code, "msg": message} @router.get("/generate_product_image_cancel_cancel/{tasks_id}>") def generate_single_logo_image(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']} diff --git a/app/api/api_prompt_generation.py b/app/api/api_prompt_generation.py index 8b930a1..a963bde 100644 --- a/app/api/api_prompt_generation.py +++ b/app/api/api_prompt_generation.py @@ -25,4 +25,4 @@ def prompt_generation(request_data: PromptGenerationImageModel): data = str(e) logger.warning(f"prompt_generation Run Exception @@@@@@:{e}") logger.info({"code": code, "message": message, "data": data}) - return {"code": code, "message": message, "data": data} + return {"code": code, "msg": message, "data": data} diff --git a/app/api/api_super_resolution.py b/app/api/api_super_resolution.py index 0d14ad7..03311d1 100644 --- a/app/api/api_super_resolution.py +++ b/app/api/api_super_resolution.py @@ -21,10 +21,10 @@ def super_resolution(request_item: SuperResolutionModel, background_tasks: Backg code = 400 message = e logger.warning(e) - return {"code": code, "message": message} + return {"code": code, "msg": message} @router.get("/sr_cancel/{tasks_id}>") def super_resolution(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']}