diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java index 4db799f3e..634652e95 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelServiceService.java @@ -130,11 +130,11 @@ public class AsyncModelServiceService { if (aigcRespVO.isSuccess()){ updateObj.setStatus(0); }else { - updateObj.setStatus(3); +// updateObj.setStatus(3); } modelServiceMapper.updateById(updateObj); }catch(Exception e){ - updateObj.setStatus(3); +// updateObj.setStatus(3); modelServiceMapper.updateById(updateObj); }; } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java index 429da3054..a0842e722 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceTaskSyncService.java @@ -62,7 +62,7 @@ public class ModelServiceTaskSyncService { // 使用 TypeReference 解析 JSON 字符串为 List try { - String query = "?filter={\"model_id\":"+modelServiceDO.getJobId()+"}"; + String query = "?filter={\"deploy_name\":"+modelServiceDO.getBaseModelName()+"}"; String res = trainHttpService.modelTableQuery(new HashMap<>(), "model_deploy",query); log.info("获取 aigc model_deploy 表数据 info {}",res); ObjectMapper mapper = new ObjectMapper();