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 
- 移除了不必要的日志输出
- 修改了模型状态更新逻辑
- 优化了模型 URL 的构建方式
- 调整了模型状态码的处理方式
This commit is contained in:
sunxiqing 2025-08-08 16:49:27 +08:00
parent 5b889eec3c
commit 627e031fcb
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -220,7 +220,7 @@ public class BaseModelTaskService {
// 调用接口获取远程模型状态
log.info("开始请求：{}", llmBackendProperties.getAListOfAvailableModels());
String resStr = HttpUtils.get(llmBackendProperties.getAListOfAvailableModels(), null);
log.info("获取aicg模型列表返回数据内容: {}", resStr);
// log.info("获取aicg模型列表返回数据内容: {}", resStr);
List<AListOfPedestalModelsVO> aListOfPedestalModelsVOS = JSONArray.parseArray(resStr, AListOfPedestalModelsVO.class);
//获取所有名称
@ -279,7 +279,7 @@ public class BaseModelTaskService {
// 调用接口获取远程模型状态
String resStr = HttpUtils.post(llmBackendProperties.getBaseModelStatus(), null, "");
log.info("获取aicg模型列表返回数据内容: {}", resStr);
// log.info("获取aicg模型列表返回数据内容: {}", resStr);
JSONArray objects = JSONArray.parseArray(resStr);
String s = JSON.toJSONString(objects);
@ -326,7 +326,7 @@ public class BaseModelTaskService {
String string1 = pedestalModelVo.getHost() + "/v1/chat/completions";
localModel.setStatus(2);
localModel.setModelUrl(string1);
localModel.setNumber(2);
// localModel.setNumber(localModel.getNumber()+1);
localModel.setJobId((long) pedestalModelVo.getId());
modelServiceMapper.updateById(localModel);
BaseModelDO baseModelDO = baseModelMapper.selectById(localModel.getFineTuningTask());
@ -365,10 +365,10 @@ public class BaseModelTaskService {
if (localModels != null && localModels.size() > 0) {
ModelServiceDO localModel = localModels.get(0);
// 如果状态不是 "running"，更新本地状态
if (localModel.getNumber()!=1) {
if (localModel.getNumber()==20) {
localModel.setStatus(3);
}else {
localModel.setNumber(2);
localModel.setNumber(localModel.getNumber()+1);
}
modelServiceMapper.updateById(localModel);
}