refactor(llm): 重构基座模型删除逻辑
- 修改了 BaseModelController 中删除模型的方法,增加更新模型状态的逻辑 -重命名了 BaseModelService 接口中的方法,提高代码可读性 - 更新了 BaseModelServiceImpl 中的实现,使用更合适的方法 - 简化了 BaseModelTaskService 中的删除逻辑 - 注释掉了定时任务,可能需要根据实际情况重新启用
This commit is contained in:
parent
5700df8687
commit
60ff7ace82
@ -86,6 +86,7 @@ public class BaseModelController {
|
||||
@Operation(summary = "删除基座模型")
|
||||
public CommonResult<Boolean> deleteTheModel(@Valid BaseModelPageReqVO pageReqVO) {
|
||||
baseModelService.deleteTheModel(pageReqVO);
|
||||
baseModelTaskService.updateTheBaseModelState();
|
||||
return success(true);
|
||||
}
|
||||
|
||||
|
@ -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);
|
||||
|
||||
|
@ -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
|
||||
|
@ -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 {
|
||||
// 获取所有基础模型列表
|
||||
|
Loading…
x
Reference in New Issue
Block a user