diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java index 182a7dcc5..9ffffccee 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java @@ -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); }