refactor(llm): 重构基座模型删除逻辑

- 修改了 BaseModelController 中删除模型的方法,增加更新模型状态的逻辑
-重命名了 BaseModelService 接口中的方法,提高代码可读性
- 更新了 BaseModelServiceImpl 中的实现,使用更合适的方法
- 简化了 BaseModelTaskService 中的删除逻辑
- 注释掉了定时任务,可能需要根据实际情况重新启用
This commit is contained in:
sunxiqing 2025-03-14 17:53:38 +08:00
parent 5700df8687
commit 60ff7ace82
4 changed files with 6 additions and 8 deletions

View File

@ -86,6 +86,7 @@ public class BaseModelController {
@Operation(summary = "删除基座模型")
public CommonResult<Boolean> deleteTheModel(@Valid BaseModelPageReqVO pageReqVO) {
baseModelService.deleteTheModel(pageReqVO);
baseModelTaskService.updateTheBaseModelState();
return success(true);
}

View File

@ -77,7 +77,7 @@ public interface BaseModelService {
List<BaseModelDO> getAllModels();
void updetatebyIds(List<BaseModelDO> differentModels);
void deletebyIds(List<BaseModelDO> differentModels);
List<BaseModelDO> selectList(LambdaQueryWrapper<BaseModelDO> select);

View File

@ -196,8 +196,8 @@ public class BaseModelServiceImpl implements BaseModelService {
}
@Override
public void updetatebyIds(List<BaseModelDO> differentModels) {
baseModelMapper.updateBatch(differentModels);
public void deletebyIds(List<BaseModelDO> differentModels) {
baseModelMapper.deleteByIds(differentModels);
}
@Override

View File

@ -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 {
// 获取所有基础模型列表