From f5276dc2c6256dac1dc35c2cc29701098823d94c Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Mon, 3 Mar 2025 17:20:34 +0800
Subject: [PATCH] =?UTF-8?q?feat(llm):=20=E6=B7=BB=E5=8A=A0=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E6=A8=A1=E5=9E=8B=E6=9C=8D=E5=8A=A1=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 ModelServiceController 中新增 queryCheckpoints 接口- 在 ModelServiceService 接口中添加 queryCheckpoints 方法
- 在 ModelServiceServiceImpl 中实现 queryCheckpoints 方法
- 使用 JSONUtil 解析 FineTuningTaskDO 中的 checkPointData
- 返回解析后的 checkpoint 列表
---
 .../modelservice/ModelServiceController.java     |  7 +++++++
 .../modelservice/ModelServiceService.java        |  2 ++
 .../modelservice/ModelServiceServiceImpl.java    | 16 ++++++++++++++++
 3 files changed, 25 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelservice/ModelServiceController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelservice/ModelServiceController.java
index 439d1085d..8999daabc 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelservice/ModelServiceController.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelservice/ModelServiceController.java
@@ -60,6 +60,13 @@ public class ModelServiceController {
         return success(modelServiceService.createModelService(createReqVO));
     }
 
+    @GetMapping("/query-checkpoints")
+    @Operation(summary = "查询模型服务")
+    public CommonResult<List<String>> queryCheckpoints(@RequestParam("id") Long id){
+
+        return success(modelServiceService.queryCheckpoints(id));
+    }
+
     @PutMapping("/update")
     @Operation(summary = "更新模型服务")
 //    @PreAuthorize("@ss.hasPermission('llm:model-service:update')")
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceService.java
index cf2371640..43db9ae26 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceService.java
@@ -87,4 +87,6 @@ public interface ModelServiceService {
      * @return 模型服务的 Map 集合
      */
     Map<Long, String> getModelServiceByTaskId (Long taskId);
+
+    List<String> queryCheckpoints(Long id);
 }
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 2bab4ae3e..47d7e8d3c 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
@@ -1,5 +1,6 @@
 package cn.iocoder.yudao.module.llm.service.modelservice;
 
+import cn.hutool.json.JSONUtil;
 import cn.iocoder.yudao.framework.common.exception.ErrorCode;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
 import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
@@ -21,8 +22,11 @@ import cn.iocoder.yudao.module.llm.service.modelassesstaskauto.ModelAssessTaskAu
 import cn.iocoder.yudao.module.llm.service.modelassesstaskmanual.ModelAssessTaskManualService;
 import cn.iocoder.yudao.module.llm.service.modelassesstaskmanualbackup.ModelAssessTaskManualBackupService;
 import cn.iocoder.yudao.module.llm.service.servername.ServerNameService;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
 import org.springframework.validation.annotation.Validated;
@@ -413,4 +417,16 @@ public class ModelServiceServiceImpl implements ModelServiceService {
         return Collections.emptyMap();
     }
 
+    @Override
+    public List<String> queryCheckpoints(Long id) {
+        FineTuningTaskDO fineTuningTaskDO = fineTuningTaskMapper.selectById(id);
+        JSONObject jsonObject = JSON.parseObject(fineTuningTaskDO.getCheckPointData());
+        if (jsonObject == null) {
+            return Collections.emptyList();
+        }
+        Set<String> strings = jsonObject.keySet();
+        List<String> res = new ArrayList<>(strings);
+        return res;
+    }
+
 }