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 35f225123..09fd06928 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 @@ -84,8 +84,8 @@ public class BaseModelController { @DeleteMapping("/deleteTheModel") @Operation(summary = "删除基座模型") - public CommonResult deleteTheModel(@Valid BaseModelPageReqVO pageReqVO) { - baseModelService.deleteTheModel(pageReqVO); + public CommonResult deleteTheModel(@RequestParam("id") Long id) { + baseModelService.deleteTheModel(id); 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 dadf9cfda..9dee47ad2 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 @@ -85,5 +85,5 @@ public interface BaseModelService { BaseModelDO getById(Long fineTuningTask); - void deleteTheModel(BaseModelPageReqVO pageReqVO); + void deleteTheModel(Long id); } 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 025ad1c67..e953b93d0 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 @@ -229,10 +229,9 @@ public class BaseModelServiceImpl implements BaseModelService { } @Override - public void deleteTheModel(BaseModelPageReqVO pageReqVO) { + public void deleteTheModel(Long id) { try { - Map map = new HashMap<>(); - map.put("model_name",pageReqVO.getModelName()); + BaseModelDO pageReqVO = baseModelMapper.selectById(id); // String resStr = HttpUtils.del(llmBackendProperties.getDeleteTheModel()+"/"+pageReqVO.getModelName(), null);