From 31a13cdfd0a0785297aeccab1574500be10f9363 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Fri, 14 Mar 2025 11:20:20 +0800 Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E6=B3=A8=E9=87=8A=E6=8E=89?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E9=87=8D=E6=96=B0=E9=83=A8=E7=BD=B2=E7=9A=84?= =?UTF-8?q?=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 BaseModelTaskService 类中，注释掉了处理模型状态为 "stop" 时的重新部署逻辑 - 保留了原有的日志输出和错误处理逻辑 - 此修改可能是为了暂时禁用模型自动重新部署的功能，或者准备替换为新的部署策略 --- .../basemodel/BaseModelTaskService.java | 54 +++++++++---------- 1 file changed, 27 insertions(+), 27 deletions(-) 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); }