1.修改工况推理接口

app/api/comprehensive_data.py

@@ -30,6 +30,7 @@ from ..services.settlement_data import SettlementDataService
 from ..services.level_data import LevelDataService
 from ..services.original_data import OriginalDataService
 from ..services.comprehensive import ComprehensiveDataService
+from ..utils.get_operating_mode import OperatingModePredictor
 import logging
 router = APIRouter(prefix="/comprehensive_data", tags=["综合数据管理"])
 logger = logging.getLogger(__name__)
@@ -470,12 +471,14 @@ def get_settlement_by_linecode(
 
         settlement_service = SettlementDataService()
         result = settlement_service.get_settlement_by_linecode(db, linecode)
-
+        settlement_data = result['settlement_data']
+        predictor = OperatingModePredictor()
+        result_1d = predictor.predict(settlement_data)
         return DataResponse(
             code=ResponseCode.SUCCESS,
             message=f"查询成功，共获取{len(result['settlement_data'])}条沉降数据",
-            total=len(result['settlement_data']),
-            data=result['settlement_data']
+            total=len(result_1d),
+            data=result_1d
         )
 
     except Exception as e:
@@ -723,4 +726,5 @@ def get_checkpoint_by_point(request: LevelDataQueryRequest, db: Session = Depend
             message=f"{ResponseMessage.QUERY_FAILED}: {str(e)}",
             total=0,
             data=[]
-        )
+        )
+