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'