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> 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 getModelServiceByTaskId (Long taskId); + + List 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 queryCheckpoints(Long id) { + FineTuningTaskDO fineTuningTaskDO = fineTuningTaskMapper.selectById(id); + JSONObject jsonObject = JSON.parseObject(fineTuningTaskDO.getCheckPointData()); + if (jsonObject == null) { + return Collections.emptyList(); + } + Set strings = jsonObject.keySet(); + List res = new ArrayList<>(strings); + return res; + } + }