feat generate 升级 attribute retrieve 迁移

app/core/config.py

@@ -19,11 +19,13 @@ class Settings(BaseSettings):
     LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
 
 
-DEBUG = False
+DEBUG = True
 if DEBUG:
-    LOGS_PATH = "logs/errors.log"
+    LOGS_PATH = "logs/"
+    CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv"
 else:
-    LOGS_PATH = "app/logs/errors.log"
+    LOGS_PATH = "app/logs/"
+    CATEGORY_PATH = "app/service/attribute/config/descriptor/category/category_dis.csv"
 
 # RABBITMQ_ENV = ""  # 生产环境
 RABBITMQ_ENV = "-dev"  # 开发环境
@@ -32,8 +34,7 @@ RABBITMQ_ENV = "-dev"  # 开发环境
 settings = Settings()
 
 # minio 配置
-MINIO_IP = "www.minio.aida.com.hk"
-MINIO_PORT = 9000
+MINIO_URL = "www.minio.aida.com.hk:9000"
 MINIO_ACCESS = 'vXKFLSJkYeEq2DrSZvkB'
 MINIO_SECRET = 'uKTZT3x7C43WvPN9QTc99DiRkwddWZrG9Uh3JVlR'
 MINIO_SECURE = True
@@ -51,15 +52,18 @@ RABBITMQ_PARAMS = {
     "virtual_host": "/"
 }
 
+# attribute service config
+ATT_TRITON_URL = "10.1.1.240:8020"
+
 # SR service config
 SR_MODEL_NAME = "super_resolution"
 SR_TRITON_URL = "10.1.1.240:10031"
 SR_MINIO_BUCKET = "aida-users"
-SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", "SuperResolution-local")
+SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", f"SuperResolution{RABBITMQ_ENV}")
 
 # GenerateImage service config
-GI_MODEL_NAME = '_stable_diffusion'
-GI_MODEL_URL = '10.1.1.240:7001'
+GI_MODEL_NAME = 'stable_diffusion_xl_lcm'
+GI_MODEL_URL = '10.1.1.150:8001'
 GI_MINIO_BUCKET = "aida-users"
 GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}")