feat(llm): 添加查询模型服务接口

- 在 ModelServiceController 中新增 queryCheckpoints 接口- 在 ModelServiceService 接口中添加 queryCheckpoints 方法
- 在 ModelServiceServiceImpl 中实现 queryCheckpoints 方法
- 使用 JSONUtil 解析 FineTuningTaskDO 中的 checkPointData
- 返回解析后的 checkpoint 列表
This commit is contained in:
sunxiqing 2025-03-03 17:20:34 +08:00
parent 105ff24b49
commit f5276dc2c6
3 changed files with 25 additions and 0 deletions

View File

@ -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')")

View File

@ -87,4 +87,6 @@ public interface ModelServiceService {
* @return 模型服务的 Map 集合
*/
Map<Long, String> getModelServiceByTaskId (Long taskId);
List<String> queryCheckpoints(Long id);
}

View File

@ -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;
}
}