From dbaf34a56ef5d0e2662171a98af112819c6cb08a Mon Sep 17 00:00:00 2001
From: zchengrong <124802516+zchengrong@users.noreply.github.com>
Date: Wed, 23 Apr 2025 11:09:14 +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:=20=20=20=E7=8E=AF=E5=A2=83=E5=88=87=E6=8D=A2=20fix?=
 =?UTF-8?q?=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20docs=EF=BC=88?=
 =?UTF-8?q?=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refactor?=
 =?UTF-8?q?=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=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 | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/app/core/config.py b/app/core/config.py
index 3309d85..03be10a 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -36,7 +36,6 @@ RABBITMQ_ENV = "-prod"  # 生产环境
 
 JAVA_STREAM_API_URL = os.getenv("JAVA_STREAM_API_URL", "https://api.aida.com.hk/api/third/party/receiveDesignResults")
 
-
 settings = Settings()
 
 # minio 配置
@@ -100,7 +99,7 @@ OPENAI_MODEL_LIST = {"gpt-3.5-turbo-0613",
 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", f"SuperResolution{RABBITMQ_ENV}")
+SR_RABBITMQ_QUEUES = f"SuperResolution{RABBITMQ_ENV}"
 
 # GenerateImage service config
 FAST_GI_MODEL_URL = '10.1.1.243:10011'
@@ -112,21 +111,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}")
@@ -134,13 +132,13 @@ 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}")
+GPI_RABBITMQ_QUEUES = f"ToProductImage{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}")
+GRI_RABBITMQ_QUEUES = f"Relight{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'