DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat(llm): 添加 GPU 名称字段并实现相关逻辑

Browse Source 
- 在 FineTuningTaskRespVO 中添加 gpuName 字段，用于展示 GPU 名称
- 在 FineTuningTaskServiceImpl 中实现 GPU 名称的获取逻辑
- 通过 ServerNameService 获取 GPU 类型对应的 ServerNameDO，并设置 GPU 名称
This commit is contained in:
Liuyang 2025-02-25 18:19:33 +08:00
parent a8cca3b52e
commit c0f597f81c
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java
View File

@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO;
import com.alibaba.fastjson.JSONObject;
import io.swagger.v3.oas.annotations.media.Schema;
@ -146,4 +147,5 @@ public class FineTuningTaskRespVO {
private String cardServerName;
public String gpuName;
}

11
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
View File

@ -18,6 +18,7 @@ import cn.iocoder.yudao.module.llm.enums.FinetuningTaskStatusEnum;
import cn.iocoder.yudao.module.llm.service.async.AsyncFineTuningTaskService;
import cn.iocoder.yudao.module.llm.service.http.TrainHttpService;
import cn.iocoder.yudao.module.llm.service.modelservice.ModelServiceService;
import cn.iocoder.yudao.module.llm.service.servername.ServerNameService;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
@ -61,6 +62,9 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
@Resource
private ServerNameMapper serverNameMapper;
@Resource
private ServerNameService serverNameService;
@Resource
private ModelServiceService modelServiceService;
@ -222,6 +226,13 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
if (datasetDO != null) {
item.setDatasetName(datasetDO.getDatasetName());
}
if(item.getGpuType()!= null){
ServerNameDO server = serverNameService.getServerName(item.getGpuType());
if (server!=null){
item.setGpuName(server.getCardServerName());
}
}
});
}