diff --git a/business-css/frontend/src/api/algorithml/index.ts b/business-css/frontend/src/api/algorithml/index.ts
index d2a0311..261ebd4 100644
--- a/business-css/frontend/src/api/algorithml/index.ts
+++ b/business-css/frontend/src/api/algorithml/index.ts
@@ -43,3 +43,10 @@ export function trainStatus(taskId:any){
     method: 'get'
   });
 }
+
+export function trainCapacityCheck(){
+    return request({
+      url:'/train/capacity/check' ,
+      method: 'get',
+    });
+  }