实验模型手动导出

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java

@@ -730,8 +730,7 @@ public class ModelsServiceImpl implements ModelsService {
 
                     HashMap<String, TrainTaskDepency> 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());