feat(llm): 添加查询模型服务接口
- 在 ModelServiceController 中新增 queryCheckpoints 接口- 在 ModelServiceService 接口中添加 queryCheckpoints 方法 - 在 ModelServiceServiceImpl 中实现 queryCheckpoints 方法 - 使用 JSONUtil 解析 FineTuningTaskDO 中的 checkPointData - 返回解析后的 checkpoint 列表
This commit is contained in:
parent
105ff24b49
commit
f5276dc2c6
@ -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')")
|
||||
|
@ -87,4 +87,6 @@ public interface ModelServiceService {
|
||||
* @return 模型服务的 Map 集合
|
||||
*/
|
||||
Map<Long, String> getModelServiceByTaskId (Long taskId);
|
||||
|
||||
List<String> queryCheckpoints(Long id);
|
||||
}
|
||||
|
@ -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;
|
||||
}
|
||||
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user