diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java index 193651e7..f3d631aa 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java @@ -730,8 +730,7 @@ public class ModelsServiceImpl implements ModelsService { HashMap map = new HashMap<>(); map.put("trainTask", modelsVo.getTrainTask()); - String t1 = JSON.toJSONString(map); - buildingModel = modelDependency1Dao.queryByTrainTask(t1); + buildingModel = modelDependency1Dao.queryByTrainTask(JSON.toJSONString(map)); } else { dvcUtils.moveFiles(sourcePath, modelPath); @@ -778,8 +777,20 @@ public class ModelsServiceImpl implements ModelsService { YamlUtils.generateYamlFile(params, rootPath, "params"); if (buildingModel != null) { - //更新模型依赖 - modelDependency1Dao.updateState(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion(), Constant.State_valid); + if (Constant.Source_Hand_Export.equals(modelsVo.getModelSource())) { + //保存模型依赖 + BeansUtils.copyPropertiesIgnoreNull(buildingModel, modelDependency); + modelDependency.setState(Constant.State_valid); + modelDependency.setModelName(modelsVo.getName()); + modelDependency.setVersion(modelsVo.getVersion()); + modelDependency.setRepoId(modelsVo.getId()); + modelDependency.setIdentifier(repositoryName); + modelDependency.setMeta(meta); + modelDependency1Dao.insert(modelDependency); + } else { + //更新模型依赖 + modelDependency1Dao.updateState(modelsVo.getId(), modelsVo.getIdentifier(), modelsVo.getVersion(), Constant.State_valid); + } } else { //保存模型依赖 modelDependency.setRepoId(modelsVo.getId());