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

refactor(llm): 调整模型服务状态和刷新逻辑- 修改 refreshTheModelService 方法的定时调度时间，从每分钟执行改为每 5 分钟执行- 更新模型服务状态判断逻辑，

Browse Source
This commit is contained in:
sunxiqing 2025-03-14 18:01:52 +08:00
parent 60ff7ace82
commit 3f3c6d0bb1
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/basemodel/BaseModelTaskService.java
View File

@ -246,7 +246,7 @@ public class BaseModelTaskService {
}
// @Scheduled(cron = "0 0/1 * * * ?")
@Scheduled(cron = "0 0/5 * * * ?")
public void refreshTheModelService() {
try {
// 获取所有基础模型列表
@ -267,7 +267,7 @@ public class BaseModelTaskService {
.collect(Collectors.toList());
//如果状态启动中没起来将状态变为停止
List<ModelServiceDO> differentModels = modelServiceDOS.stream()
.filter(baseModel -> !remoteModelNames.contains(baseModel.getBaseModelName()) && (baseModel.getStatus() == 4 || baseModel.getStatus() == 2))
.filter(baseModel -> !remoteModelNames.contains(baseModel.getBaseModelName()) && (baseModel.getStatus() == 2 || baseModel.getStatus() == 1))
.collect(Collectors.toList());
for (ModelServiceDO baseModel : differentModels){
baseModel.setStatus(3);
@ -301,7 +301,7 @@ public class BaseModelTaskService {
modelServiceMapper.updateById(localModel);
} else {
String string1 = pedestalModelVo.getHost() + "/v1/chat/completions";
localModel.setStatus(4);
localModel.setStatus(2);
localModel.setModelUrl(string1);
// localModel.setApiUrl(string1);
localModel.setJobId((long) pedestalModelVo.getId());