修改模型元数据

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

@@ -615,7 +615,7 @@ public class ModelsServiceImpl implements ModelsService {
 
             Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
             YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
-            String meta = JSON.toJSONString(metaMap);
+            String meta = JSON.toJSONString(modelMetaVo);
 
             DVCUtils.dvcInit(rootPath);
             // 配置远程S3地址
@@ -711,7 +711,7 @@ public class ModelsServiceImpl implements ModelsService {
 
             Map<String, Object> metaMap = JsonUtils.objectToMap(modelMetaVo);
             YamlUtils.generateYamlFile(metaMap, metaPath, "metadata");
-            String meta = JSON.toJSONString(metaMap);
+            String meta = JSON.toJSONString(modelMetaVo);
 
             // 配置远程S3地址
             String s3Path = "management-platform-files/" + ci4sUsername + "/model/" + modelsVo.getId() + "/" + repositoryName + "/" + branchName;

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/ModelMetaVo.java

@@ -2,6 +2,7 @@ package com.ruoyi.platform.vo;
 
 import com.fasterxml.jackson.databind.PropertyNamingStrategy;
 import com.fasterxml.jackson.databind.annotation.JsonNaming;
+import com.ruoyi.platform.domain.dependencydomain.ProjectDepency;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 
@@ -40,10 +41,10 @@ public class ModelMetaVo implements Serializable {
     private String modelSource;
 
     @ApiModelProperty(value = "训练数据集")
-    private List<String> trainDatasets;
+    private List<NewDatasetVo> trainDatasets;
 
     @ApiModelProperty(value = "测试数据集")
-    private List<String> testDatasets;
+    private List<NewDatasetVo> testDatasets;
 
     @ApiModelProperty(value = "参数")
     private HashMap<String, String> params;
@@ -80,5 +81,9 @@ public class ModelMetaVo implements Serializable {
 
     @ApiModelProperty(name = "relative_paths",value = "文件相对路径")
     private String relativePaths;
+
+    @ApiModelProperty(name = "project_depency",value = "项目依赖")
+    private ProjectDepency projectDepency;
+
 }