diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java index 1eb3c497b..752b0747c 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java @@ -16,6 +16,7 @@ import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties import cn.iocoder.yudao.module.llm.service.application.ApplicationService; import cn.iocoder.yudao.module.llm.service.async.AsyncModelServiceService; import cn.iocoder.yudao.module.llm.service.http.ModelService; +import cn.iocoder.yudao.module.llm.service.http.TrainHttpService; import cn.iocoder.yudao.module.llm.service.modelassesstaskauto.ModelAssessTaskAutoService; import cn.iocoder.yudao.module.llm.service.modelassesstaskmanual.ModelAssessTaskManualService; import cn.iocoder.yudao.module.llm.service.modelassesstaskmanualbackup.ModelAssessTaskManualBackupService; @@ -76,6 +77,9 @@ public class ModelServiceServiceImpl implements ModelServiceService { @Resource private ServerNameService serverNameService; + @Resource + private TrainHttpService trainHttpService; + @Override public Long createModelService (ModelServiceSaveReqVO createReqVO) { LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); @@ -145,6 +149,10 @@ public class ModelServiceServiceImpl implements ModelServiceService { validateModelServiceExists(id); // 校验使用 validateModelUse(id); + + ModelServiceDO modelServiceDO = modelServiceMapper.selectById(id); + ServerNameDO serverNameDO = serverNameService.getServerName(modelServiceDO.getGpuType()); + trainHttpService.modelUndeploy(new HashMap<>(), serverNameDO.getHost(),modelServiceDO.getJobId()); // 删除 modelServiceMapper.deleteById(id); }