diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java index 7e0ee8649..6912880ff 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java @@ -19,6 +19,7 @@ import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties import cn.iocoder.yudao.module.llm.service.application.ApplicationService; import cn.iocoder.yudao.module.llm.service.async.AsyncModelServiceService; import cn.iocoder.yudao.module.llm.service.basemodel.BaseModelService; +import cn.iocoder.yudao.module.llm.service.basemodel.BaseModelTaskService; import cn.iocoder.yudao.module.llm.service.http.ModelService; import cn.iocoder.yudao.module.llm.service.http.TrainHttpService; import cn.iocoder.yudao.module.llm.service.modelassesstaskauto.ModelAssessTaskAutoService; @@ -89,6 +90,8 @@ public class ModelServiceServiceImpl implements ModelServiceService { @Resource @Lazy private BaseModelService baseModelService; + @Resource + private BaseModelTaskService baseModelTaskService; @Override public Long createModelService (ModelServiceSaveReqVO createReqVO) { @@ -118,7 +121,7 @@ public class ModelServiceServiceImpl implements ModelServiceService { String replace = llmBackendProperties.getModelServiceApi().replace("apiKey", apikey); modelService.setApiUrl(replace); - modelService.setStatus(3); + modelService.setStatus(1); BaseModelDO byAigcId = baseModelService.getById(modelService.getFineTuningTask()); if (byAigcId != null){ modelService.setBaseModelName(byAigcId.getModelName());