From daffbe1733db90b18da66fdc7fd91170e40e6850 Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Fri, 20 Jun 2025 11:09:12 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BF=AE=E6=94=B9=E8=B5=84?=
 =?UTF-8?q?=E6=BA=90=E7=89=88=E6=9C=AC=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../model/NewModelFromGitController.java          |  6 ++++++
 .../com/ruoyi/platform/service/ModelsService.java |  2 ++
 .../platform/service/impl/ModelsServiceImpl.java  | 15 +++++++++++++++
 3 files changed, 23 insertions(+)

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/model/NewModelFromGitController.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/model/NewModelFromGitController.java
index 4bb8f03b..0d6dcbde 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/model/NewModelFromGitController.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/controller/model/NewModelFromGitController.java
@@ -38,6 +38,12 @@ public class NewModelFromGitController extends BaseController {
         return AjaxResult.success(this.modelsService.newCreateVersion(modelsVo));
     }
 
+    @PutMapping("/updateVersionDesc")
+    @ApiOperation("修改版本描述")
+    public AjaxResult updateVersionDesc(@RequestBody ModelsVo modelsVo) throws Exception {
+        return AjaxResult.success(this.modelsService.updateVersionDesc(modelsVo));
+    }
+
     @CrossOrigin(origins = "*", allowedHeaders = "*")
     @PostMapping("/upload")
     @ApiOperation(value = "上传模型", notes = "根据模型id上传模型文件，并将信息存入数据库。")
diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java
index d8b47626..034f0b19 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ModelsService.java
@@ -93,6 +93,8 @@ public interface ModelsService {
 
     String newCreateVersion(ModelsVo modelsVo) throws Exception;
 
+    String updateVersionDesc(ModelsVo modelsVo) throws Exception;
+
     List<Map<String, String>> uploadModelLocal(MultipartFile[] files, String uuid) throws Exception;
 
     ResponseEntity<InputStreamResource> downloadAllModelFilesNew(String name, String identifier, Integer id, String version, Boolean isPublic) throws IOException, Exception;
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 6df28844..a4118adf 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
@@ -859,6 +859,21 @@ public class ModelsServiceImpl implements ModelsService {
         return "新增模型版本成功";
     }
 
+    @Override
+    public String updateVersionDesc(ModelsVo modelsVo) throws Exception {
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        String ci4sUsername = Boolean.TRUE.equals(modelsVo.getIsPublic()) ? Constant.Item_Public : loginUser.getUsername();
+        String gitLinkUsername = loginUser.getSysUser().getUserName();
+        String gitLinkPassword = decrypt(loginUser.getSysUser().getOriginPassword());
+        String repositoryName = modelsVo.getIdentifier();
+        String branchName = modelsVo.getVersion();
+        String relatePath = ci4sUsername + "/datasets/" + modelsVo.getId() + "/" + repositoryName + "/" + branchName;
+        String rootPath = localPath + relatePath;
+        YamlUtils.generateYamlFile(JsonUtils.objectToMap(modelsVo), rootPath, "dataset");
+        dvcUtils.pushNewBranchToRemote(rootPath, gitLinkUsername, gitLinkPassword, branchName);
+        return "更新模型版本成功";
+    }
+
 
     @Override
     public List<Map<String, String>> uploadModelLocal(MultipartFile[] files, String uuid) throws Exception {