From ce6b890133c6a35eb39ed9766e37fdac4a4089f0 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Fri, 21 Feb 2025 11:04:37 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../llm/service/modelservice/ModelServiceServiceImpl.java | 8 ++++++++
 1 file changed, 8 insertions(+)

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<ModelServiceDO> 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);
     }