diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java index a3cdf6cdc..695bd6d24 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java @@ -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 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); }