From 4fb4d808b663f2d295218c7a97890b8756aed85f Mon Sep 17 00:00:00 2001
From: zhouchengrong <zchengrong@yeah.net>
Date: Tue, 2 Sep 2025 22:29:30 +0800
Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?=
 =?UTF-8?q?=EF=BC=89:=20fix=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20docs?=
 =?UTF-8?q?=EF=BC=88=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refac?=
 =?UTF-8?q?tor=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E6=B5=8B=E8=AF=95):?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/core/config.py | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/app/core/config.py b/app/core/config.py
index 6ec49ed..2abe29c 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -37,8 +37,8 @@ else:
     CHROMADB_PATH = "/chromadb/"
 
 
-# RABBITMQ_ENV = ""  # 生产环境
-RABBITMQ_ENV = "-dev"  # 开发环境
+RABBITMQ_ENV = "-prod"  # 生产环境#
+# RABBITMQ_ENV = "-dev"  # 开发环境
 # RABBITMQ_ENV = "-local"  # 本地测试环境
 
 JAVA_STREAM_API_URL = os.getenv("JAVA_STREAM_API_URL", "https://api.aida.com.hk/api/third/party/receiveDesignResults")
@@ -118,20 +118,20 @@ GI_MODEL_NAME = 'flux'
 GMV_MODEL_URL = '10.1.1.243:10081'
 GMV_MODEL_NAME = 'multi_view'
 
-GMV_RABBITMQ_QUEUES = os.getenv("GMV_RABBITMQ_QUEUES", f"GenerateMultiView{RABBITMQ_ENV}")
+GMV_RABBITMQ_QUEUES = f"GenerateMultiView{RABBITMQ_ENV}"
 
 GI_MINIO_BUCKET = "aida-users"
-GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}")
+GI_RABBITMQ_QUEUES = f"GenerateImage{RABBITMQ_ENV}"
 GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg"
 
 # SLOGAN service config
-SLOGAN_RABBITMQ_QUEUES = os.getenv("SLOGAN_RABBITMQ_QUEUES", f"Slogan{RABBITMQ_ENV}")
+SLOGAN_RABBITMQ_QUEUES = f"Slogan{RABBITMQ_ENV}"
 
 # Generate Single Logo service config
 GSL_MODEL_URL = '10.1.1.243:10041'
 GSL_MINIO_BUCKET = "aida-users"
 GSL_MODEL_NAME = 'stable_diffusion_xl_transparent'
-GEN_SINGLE_LOGO_RABBITMQ_QUEUES = os.getenv("GEN_SINGLE_LOGO_RABBITMQ_QUEUES", f"GenSingleLogo{RABBITMQ_ENV}")
+GEN_SINGLE_LOGO_RABBITMQ_QUEUES = f"GenSingleLogo{RABBITMQ_ENV}"
 
 # Generate Product service config
 # GPI_RABBITMQ_QUEUES = os.getenv("GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"ToProductImage{RABBITMQ_ENV}")
@@ -139,23 +139,23 @@ GEN_SINGLE_LOGO_RABBITMQ_QUEUES = os.getenv("GEN_SINGLE_LOGO_RABBITMQ_QUEUES", f
 # GPI_MODEL_URL = '10.1.1.243:10051'
 
 # Generate Product service config 旧版product img 模型
-GPI_RABBITMQ_QUEUES = os.getenv("GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"ToProductImage{RABBITMQ_ENV}")
-BATCH_GPI_RABBITMQ_QUEUES = os.getenv("BATCH_GEN_PRODUCT_IMAGE_RABBITMQ_QUEUES", f"BatchToProductImage{RABBITMQ_ENV}")
+GPI_RABBITMQ_QUEUES = f"ToProductImage{RABBITMQ_ENV}"
+BATCH_GPI_RABBITMQ_QUEUES = f"BatchToProductImage{RABBITMQ_ENV}"
 GPI_MODEL_NAME_OVERALL = 'diffusion_ensemble_all'
 GPI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_cnet'
 GPI_MODEL_URL = '10.1.1.243:10051'
 
 # Generate Single Logo service config
-GRI_RABBITMQ_QUEUES = os.getenv("GEN_RELIGHT_IMAGE_RABBITMQ_QUEUES", f"Relight{RABBITMQ_ENV}")
-BATCH_GRI_RABBITMQ_QUEUES = os.getenv("BATCH_GEN_RELIGHT_IMAGE_RABBITMQ_QUEUES", f"BatchRelight{RABBITMQ_ENV}")
+GRI_RABBITMQ_QUEUES = f"Relight{RABBITMQ_ENV}"
+BATCH_GRI_RABBITMQ_QUEUES = f"BatchRelight{RABBITMQ_ENV}"
 GRI_MODEL_NAME_OVERALL = 'diffusion_relight_ensemble'
 GRI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_relight'
 GRI_MODEL_URL = '10.1.1.240:10051'
 
 # Pose Transform service config
 
-PS_RABBITMQ_QUEUES = os.getenv("PS_RABBITMQ_QUEUES", f"PoseTransform{RABBITMQ_ENV}")
-BATCH_PS_RABBITMQ_QUEUES = os.getenv("BATCH_PS_RABBITMQ_QUEUES", f"BatchPoseTransform{RABBITMQ_ENV}")
+PS_RABBITMQ_QUEUES = f"PoseTransform{RABBITMQ_ENV}"
+BATCH_PS_RABBITMQ_QUEUES = f"BatchPoseTransform{RABBITMQ_ENV}"
 PT_MODEL_URL = '10.1.1.243:10061'
 
 # SEG service config
@@ -169,7 +169,7 @@ SEGMENTATION = {
 OLLAMA_URL = "http://10.1.1.240:11434/api/embeddings"
 
 # design batch
-BATCH_DESIGN_RABBITMQ_QUEUES = os.getenv("BATCH_DESIGN_RABBITMQ_QUEUES", f"DesignBatch{RABBITMQ_ENV}")
+BATCH_DESIGN_RABBITMQ_QUEUES = f"DesignBatch{RABBITMQ_ENV}"
 
 # DESIGN config
 DESIGN_MODEL_URL = '10.1.1.240:10000'