DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(module-llm):优化模型卸载和删除逻辑- 在 ModelServiceServiceImpl 中增加日志记录，详细记录模型卸载和删除过程

Browse Source 
- 优化 TrainHttpService 中的 modelUndeploy 方法，添加异常处理并记录日志
This commit is contained in:
Liuyang 2025-02-26 11:25:26 +08:00
parent 49e5068089
commit ee52e777d6
2 changed files with 36 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
View File

@ -247,13 +247,32 @@ public class TrainHttpService {
}
public AigcModelDeployVO modelUndeploy (Map<String, String> headers, String url, Long deployId) {
// login(headers);
String modelDeploy = llmBackendProperties.getModelUndeploy();
String res = HttpUtils.post(url + modelDeploy + deployId, headers, "");
log.info(" modelDeploy:{}", res);
AigcModelDeployVO modelDeployRespVO = JSON.parseObject(res, AigcModelDeployVO.class);
log.info(" modelDeploy:{}", modelDeployRespVO);
return modelDeployRespVO;
try {
log.info("开始卸载模型部署，部署ID: {}", deployId);
// 构建请求 URL
String modelDeploy = llmBackendProperties.getModelUndeploy();
String requestUrl = url + modelDeploy + deployId;
log.debug("构建请求 URL: {}", requestUrl);
// 发起 HTTP POST 请求
log.debug("正在发起 HTTP POST 请求...");
String res = HttpUtils.post(requestUrl, headers, "");
log.info("HTTP 请求完成。响应内容: {}", res);
// 解析响应内容
log.debug("正在解析响应内容...");
AigcModelDeployVO modelDeployRespVO = JSON.parseObject(res, AigcModelDeployVO.class);
log.info("响应解析完成。模型卸载结果: {}", JSON.toJSONString(modelDeployRespVO));
// 返回结果
return modelDeployRespVO;
} catch (Exception e) {
log.error("卸载模型部署时发生异常。部署ID: {}", deployId, e);
throw new RuntimeException("模型卸载失败", e);
}
}
/**

12
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
View File

@ -147,14 +147,22 @@ public class ModelServiceServiceImpl implements ModelServiceService {
public void deleteModelService (Long id) {
// 校验存在
validateModelServiceExists(id);
// 校验使用
validateModelUse(id);
ModelServiceDO modelServiceDO = modelServiceMapper.selectById(id);
ServerNameDO serverNameDO = serverNameService.getServerName(modelServiceDO.getGpuType());
trainHttpService.modelUndeploy(new HashMap<>(), serverNameDO.getHost(),modelServiceDO.getJobId());
// 删除
// 发起模型卸载请求
log.info("正在发起模型卸载请求，服务ID: {}", id);
trainHttpService.modelUndeploy(new HashMap<>(), serverNameDO.getHost(), modelServiceDO.getJobId());
log.info("模型卸载请求完成，服务ID: {}", id);
// 删除模型服务
log.debug("正在删除模型服务，服务ID: {}", id);
modelServiceMapper.deleteById(id);
log.info("模型服务删除成功，服务ID: {}", id);
}
/**