From 88014bdb4c55851ffcaf76f21376073cc01de7b9 Mon Sep 17 00:00:00 2001
From: zhouchengrong <Zchengrong@yeah.net>
Date: Mon, 17 Jun 2024 17:14:10 +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 | 11 ++++++++---
 app/core/config.py                |  2 +-
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/app/api/api_attribute_retrieve.py b/app/api/api_attribute_retrieve.py
index ef3955f..7a14e9d 100644
--- a/app/api/api_attribute_retrieve.py
+++ b/app/api/api_attribute_retrieve.py
@@ -1,9 +1,11 @@
 import json
 import logging
 from fastapi import APIRouter, HTTPException
+
+from app.core.config import DEBUG
 from app.schemas.attribute_retrieve import *
 from app.schemas.response_template import ResponseModel
-from app.service.attribute.config import const
+from app.service.attribute.config import const, local_debug_const
 from app.service.attribute.service_att_recognition import AttributeRecognition
 from app.service.attribute.service_category_recognition import CategoryRecognition
 
@@ -16,13 +18,16 @@ logger = logging.getLogger()
 def attribute_recognition(request_item: list[AttributeRecognitionModel]):
     try:
         logger.info(f"attribute_recognition request item is : @@@@@@:{request_item}")
-        service = AttributeRecognition(const=const, request_data=request_item)
+        if DEBUG:
+            service = AttributeRecognition(const=local_debug_const, request_data=request_item)
+        else:
+            service = AttributeRecognition(const=const, request_data=request_item)
         data = service.get_result()
         logger.info(f"attribute_recognition response @@@@@@:{json.dumps(data, indent=4)}")
     except Exception as e:
         logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}")
         raise HTTPException(status_code=404, detail=str(e))
-    return ResponseModel(data=data)
+    return ResponseModel(data={"list": data})
 
 
 # 类别识别
diff --git a/app/core/config.py b/app/core/config.py
index b293cef..08c0998 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -19,7 +19,7 @@ class Settings(BaseSettings):
     LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
 
 
-DEBUG = False
+DEBUG = True
 if DEBUG:
     LOGS_PATH = "logs/"
     CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv"