diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java index 2ed72e120..a35b6cdc1 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java @@ -67,7 +67,7 @@ public class BaseModelController { // @PreAuthorize("@ss.hasPermission('llm:base-model:update')") public CommonResult active(@Valid @RequestBody BaseModelSaveReqVO updateReqVO) { // updateReqVO.setIsActive(1); - baseModelService.updateBaseModel(updateReqVO); + baseModelService.startTheModel(updateReqVO); return success(true); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java index 682a17c59..55a39555a 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java @@ -92,4 +92,6 @@ public interface BaseModelService { * @return 基座模型 */ List listBaseModels(); + + void startTheModel(BaseModelSaveReqVO updateReqVO); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java index 8fd014cfa..1d09a254f 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java @@ -110,18 +110,10 @@ public class BaseModelServiceImpl implements BaseModelService { public void updateBaseModel (BaseModelSaveReqVO updateReqVO) { validateBaseModelNameExists(updateReqVO); // 校验存在 - String moder = updateReqVO.getModelName(); - try { - Map map = new HashMap<>(); - map.put("model",moder); - String resStr = HttpUtils.post(llmBackendProperties.getDeployModel(), null,JSON.toJSONString(map)); - validateBaseModelExists(updateReqVO.getId()); - // 更新 -// BaseModelDO updateObj = BeanUtils.toBean(updateReqVO, BaseModelDO.class); -// baseModelMapper.updateById(updateObj); - }catch (Exception e){ - log.error("启动基础模型状态时发生异常: {}", e.getMessage(), e); - } + validateBaseModelExists(updateReqVO.getId()); + // 更新 + BaseModelDO updateObj = BeanUtils.toBean(updateReqVO, BaseModelDO.class); + baseModelMapper.updateById(updateObj); } @Override @@ -319,6 +311,19 @@ public class BaseModelServiceImpl implements BaseModelService { return selects; } + @Override + public void startTheModel(BaseModelSaveReqVO updateReqVO) { + try { + String moder = updateReqVO.getModelName(); + Map map = new HashMap<>(); + map.put("model",moder); + String resStr = HttpUtils.post(llmBackendProperties.getDeployModel(), null,JSON.toJSONString(map)); + log.info("启动基础模型状态:{}", resStr); + }catch (Exception e){ + log.error("启动基础模型状态时发生异常: {}", e.getMessage(), e); + } + } + /** * 校验模型使用情况