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 b57f0f10a..35f225123 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 @@ -86,6 +86,7 @@ public class BaseModelController { @Operation(summary = "删除基座模型") public CommonResult deleteTheModel(@Valid BaseModelPageReqVO pageReqVO) { baseModelService.deleteTheModel(pageReqVO); + baseModelTaskService.updateTheBaseModelState(); 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 6768e8e43..dadf9cfda 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 @@ -77,7 +77,7 @@ public interface BaseModelService { List getAllModels(); - void updetatebyIds(List differentModels); + void deletebyIds(List differentModels); List selectList(LambdaQueryWrapper select); 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 27dd2d1e8..1d6997549 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 @@ -196,8 +196,8 @@ public class BaseModelServiceImpl implements BaseModelService { } @Override - public void updetatebyIds(List differentModels) { - baseModelMapper.updateBatch(differentModels); + public void deletebyIds(List differentModels) { + baseModelMapper.deleteByIds(differentModels); } @Override 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 b947a2d4c..e3e495663 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 @@ -226,10 +226,7 @@ public class BaseModelTaskService { .filter(baseModel -> !remoteModelNames.contains(baseModel.getModelName())) .collect(Collectors.toList()); if (differentModels.size()>0) { - for (BaseModelDO baseModel : differentModels){ - baseModel.setDeleted(true); - } - baseModelService.updetatebyIds(differentModels); + baseModelService.deletebyIds(differentModels); } // 筛选出 remoteModelNames 中不在 differentModels 中的模型名称 @@ -249,7 +246,7 @@ public class BaseModelTaskService { } - @Scheduled(cron = "0 0/1 * * * ?") +// @Scheduled(cron = "0 0/1 * * * ?") public void refreshTheModelService() { try { // 获取所有基础模型列表