DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(llm): 优化禁用模型功能

Browse Source 
- 在 ModelServiceController 中添加模型禁用前的检查应用逻辑
- 在 ModelServiceService 接口中新增 inspectTheApplication 方法
- 在 ModelServiceServiceImpl 中实现 inspectTheApplication 方法，增加应用中心模型使用情况的校验
This commit is contained in:
sunxiqing 2025-03-17 12:51:24 +08:00
parent b0239d5884
commit 720210d6b9
3 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelservice/ModelServiceController.java
View File

@ -162,6 +162,7 @@ public class ModelServiceController {
@Operation(summary = "禁用模型")
// @PreAuthorize("@ss.hasPermission('llm:base-model:update')")
public CommonResult<Boolean> disable(@Valid @RequestBody ModelServiceSaveReqVO updateReqVO) {
modelServiceService.inspectTheApplication(updateReqVO);
modelServiceService.disableTheModel(updateReqVO);
return success(true);
}

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceService.java
View File

@ -93,4 +93,6 @@ public interface ModelServiceService {
void startTheModel(ModelServiceSaveReqVO updateReqVO);
void disableTheModel(ModelServiceSaveReqVO updateReqVO);
void inspectTheApplication(ModelServiceSaveReqVO updateReqVO);
}

7
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
View File

@ -9,6 +9,7 @@ import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServicePageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServiceRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServiceSaveReqVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
@ -30,6 +31,7 @@ import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
@ -485,4 +487,9 @@ public class ModelServiceServiceImpl implements ModelServiceService {
}
}
@Override
public void inspectTheApplication(ModelServiceSaveReqVO updateReqVO) {
validateModelUsesInApplicationCenter(updateReqVO.getId(), updateReqVO.getServiceName());
}
}