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

refactor(llm): 注释掉模型重新部署的代码

Browse Source 
- 在 BaseModelTaskService 类中，注释掉了处理模型状态为 "stop" 时的重新部署逻辑
- 保留了原有的日志输出和错误处理逻辑
- 此修改可能是为了暂时禁用模型自动重新部署的功能，或者准备替换为新的部署策略
This commit is contained in:
Liuyang 2025-03-14 11:20:20 +08:00
parent 0a79f153ef
commit 31a13cdfd0
1 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java
View File

@ -99,33 +99,33 @@ public class BaseModelTaskService {
String status = latestRecord.getStatus();
log.info("最新模型部署记录状态: {}", status);
// 如果模型状态为 "stop"，则重新部署
if ("stop".equals(status)) {
log.info("模型状态为 'stop'，正在重新部署模型...");
// 构建模型部署请求
AigcModelDeploySaveReq aigcModelDeploySaveReq = new AigcModelDeploySaveReq(
baseModelDO.getAigcModelName(), "gpu");
log.info("模型部署请求参数: {}", JSON.toJSONString(aigcModelDeploySaveReq));
// 发起模型部署请求
ModelDeployRespVO modelDeployRespVO = trainHttpService.modelDeploy(
new HashMap<>(), serverName.getHost(), aigcModelDeploySaveReq);
log.info("模型部署请求完成。响应内容: {}", JSON.toJSONString(modelDeployRespVO));
// 更新基础模型信息
if (!"error".equals(modelDeployRespVO.getMessage())) {
log.info("模型部署成功。正在更新基础模型信息...");
BaseModelSaveReqVO baseModelSaveReqVO = new BaseModelSaveReqVO();
baseModelSaveReqVO.setId(baseModelDO.getId());
baseModelSaveReqVO.setModelId(modelDeployRespVO.getId());
baseModelSaveReqVO.setChatUrl(modelDeployRespVO.getPort() + DEFAULT_MODEL_URL_SUFFIX);
baseModelService.updateBaseModel(baseModelSaveReqVO);
log.info("基础模型信息更新完成。模型ID: {}", baseModelDO.getId());
} else {
log.error("模型部署失败。模型ID: {}", baseModelDO.getId());
}
}
// // 如果模型状态为 "stop"，则重新部署
// if ("stop".equals(status)) {
// log.info("模型状态为 'stop'，正在重新部署模型...");
//
// // 构建模型部署请求
// AigcModelDeploySaveReq aigcModelDeploySaveReq = new AigcModelDeploySaveReq(
// baseModelDO.getAigcModelName(), "gpu");
// log.info("模型部署请求参数: {}", JSON.toJSONString(aigcModelDeploySaveReq));
//
// // 发起模型部署请求
// ModelDeployRespVO modelDeployRespVO = trainHttpService.modelDeploy(
// new HashMap<>(), serverName.getHost(), aigcModelDeploySaveReq);
// log.info("模型部署请求完成。响应内容: {}", JSON.toJSONString(modelDeployRespVO));
//
// // 更新基础模型信息
// if (!"error".equals(modelDeployRespVO.getMessage())) {
// log.info("模型部署成功。正在更新基础模型信息...");
// BaseModelSaveReqVO baseModelSaveReqVO = new BaseModelSaveReqVO();
// baseModelSaveReqVO.setId(baseModelDO.getId());
// baseModelSaveReqVO.setModelId(modelDeployRespVO.getId());
// baseModelSaveReqVO.setChatUrl(modelDeployRespVO.getPort() + DEFAULT_MODEL_URL_SUFFIX);
// baseModelService.updateBaseModel(baseModelSaveReqVO);
// log.info("基础模型信息更新完成。模型ID: {}", baseModelDO.getId());
// } else {
// log.error("模型部署失败。模型ID: {}", baseModelDO.getId());
// }
// }
} else {
log.warn("未找到模型部署记录。模型ID: {}", modelId);
}