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

refactor(llm): 更新模型服务状态码使用统一的常量- 将模型服务状态码0、2、3 替换为统一的常量

Browse Source 
-优化代码可维护性，减少硬编码
This commit is contained in:
sunxiqing 2025-03-14 20:30:55 +08:00
parent 6d09621d24
commit a0e4979bd1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -118,7 +118,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
String replace = llmBackendProperties.getModelServiceApi().replace("apiKey", apikey);
modelService.setApiUrl(replace);
modelService.setStatus(0);
modelService.setStatus(3);
BaseModelDO byAigcId = baseModelService.getById(modelService.getFineTuningTask());
if (byAigcId != null){
modelService.setBaseModelName(byAigcId.getModelName());
@ -450,7 +450,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
map.put("model",byAigcId.getModelName());
String resStr = HttpUtils.post(llmBackendProperties.getDeployModel(), null,JSON.toJSONString(map));
ModelServiceDO updateObj = BeanUtils.toBean(updateReqVO, ModelServiceDO.class);
updateObj.setStatus(2);
updateObj.setStatus(3);
modelServiceMapper.updateById(updateObj);
}catch (Exception e){
log.error("启动基础模型状态时发生异常: {}", e.getMessage(), e);
@ -465,7 +465,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
String resStr = HttpUtils.post(llmBackendProperties.getDeleteModel()+"?deploy_id="+byAigcId.getModelId(), null, "");
log.info(" unActive:{}", resStr);
ModelServiceDO updateObj = BeanUtils.toBean(updateReqVO, ModelServiceDO.class);
updateObj.setStatus(0);
updateObj.setStatus(3);
modelServiceMapper.updateById(updateObj);
// 禁用模型
}catch (Exception e){