服务迁移测试

app/api/api_design.py

@@ -4,6 +4,7 @@ import logging
 import requests
 from fastapi import APIRouter, HTTPException, BackgroundTasks
 
+from app.core.config import settings
 from app.schemas.design import DesignModel, ModelProgressModel, DesignStreamModel, SAMRequestModel
 from app.schemas.response_template import ResponseModel
 from app.service.design_fast.design_generate import design_generate, design_generate_v2
@@ -425,7 +426,7 @@ async def seg_anything(request_data: SAMRequestModel):
     """
     try:
         logger.info(f"seg_anything request item is : @@@@@@:{json.dumps(request_data.dict(), indent=4)}")
-        data = requests.post("http://10.1.1.240:10075/predict", json=request_data.dict())
+        data = requests.post(f"http://{settings.A6000_SERVICE_HOST}:10075/predict", json=request_data.dict())
         logger.info(f"seg_anything response @@@@@@:{json.dumps(json.loads(data.content), indent=4)}")
         return ResponseModel(data=json.loads(data.content))
     except Exception as e: