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 
- 将激活模型的逻辑从 updateBaseModel 方法中分离，新增 startTheModel 方法
- 优化了模型激活的错误处理和日志记录
- 移除了更新模型时不必要的远程调用逻辑
This commit is contained in:
sunxiqing 2025-08-11 14:01:08 +08:00
parent af72c50150
commit f2e02db09b
3 changed files with 20 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java
View File

@ -67,7 +67,7 @@ public class BaseModelController {
// @PreAuthorize("@ss.hasPermission('llm:base-model:update')")
public CommonResult<Boolean> active(@Valid @RequestBody BaseModelSaveReqVO updateReqVO) {
// updateReqVO.setIsActive(1);
baseModelService.updateBaseModel(updateReqVO);
baseModelService.startTheModel(updateReqVO);
return success(true);
}

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

@ -92,4 +92,6 @@ public interface BaseModelService {
* @return 基座模型
*/
List<BaseModelDO> listBaseModels();
void startTheModel(BaseModelSaveReqVO updateReqVO);
}

29
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
View File

@ -110,18 +110,10 @@ public class BaseModelServiceImpl implements BaseModelService {
public void updateBaseModel (BaseModelSaveReqVO updateReqVO) {
validateBaseModelNameExists(updateReqVO);
// 校验存在
String moder = updateReqVO.getModelName();
try {
Map<String,String> map = new HashMap<>();
map.put("model",moder);
String resStr = HttpUtils.post(llmBackendProperties.getDeployModel(), null,JSON.toJSONString(map));
validateBaseModelExists(updateReqVO.getId());
// 更新
// BaseModelDO updateObj = BeanUtils.toBean(updateReqVO, BaseModelDO.class);
// baseModelMapper.updateById(updateObj);
}catch (Exception e){
log.error("启动基础模型状态时发生异常: {}", e.getMessage(), e);
}
validateBaseModelExists(updateReqVO.getId());
// 更新
BaseModelDO updateObj = BeanUtils.toBean(updateReqVO, BaseModelDO.class);
baseModelMapper.updateById(updateObj);
}
@Override
@ -319,6 +311,19 @@ public class BaseModelServiceImpl implements BaseModelService {
return selects;
}
@Override
public void startTheModel(BaseModelSaveReqVO updateReqVO) {
try {
String moder = updateReqVO.getModelName();
Map<String,String> map = new HashMap<>();
map.put("model",moder);
String resStr = HttpUtils.post(llmBackendProperties.getDeployModel(), null,JSON.toJSONString(map));
log.info("启动基础模型状态:{}", resStr);
}catch (Exception e){
log.error("启动基础模型状态时发生异常: {}", e.getMessage(), e);
}
}
/**
* 校验模型使用情况