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

基础模型api

Browse Source
This commit is contained in:
ire 2025-02-19 15:32:50 +08:00
parent 956742141b
commit b14e76b763
4 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java
View File

@ -54,4 +54,8 @@ public class BaseModelSaveReqVO {
private Long modelId;
private String chatUrl;
private Long gpuId;
private String gpuName;
}

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java
View File

@ -79,4 +79,8 @@ public class BaseModelDO extends BaseDO {
private String chatUrl;
private Long gpuId;
private String gpuName;
}

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
View File

@ -26,6 +26,7 @@ import cn.iocoder.yudao.module.llm.service.datarefluxdata.DataRefluxDataService;
import cn.iocoder.yudao.module.llm.service.http.ModelService;
import cn.iocoder.yudao.module.llm.service.http.vo.*;
import cn.iocoder.yudao.module.llm.service.prompttemplates.PromptTemplatesService;
import cn.iocoder.yudao.module.llm.service.servername.ServerNameService;
import com.alibaba.excel.util.StringUtils;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
@ -170,6 +171,7 @@ public class ConversationServiceImpl implements ConversationService {
if (baseModelDO == null) {
throw exception(BASE_MODEL_NOT_EXISTS);
}
selfModelUrl = baseModelDO.getChatUrl() + "/v1/chat/completions";
model = baseModelDO.getAigcModelName();
}else if (Objects.equals(0, chatReqVO.getModelType())) {
// 自定义模型

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

@ -127,7 +127,7 @@ public class FineTuningTaskSyncService {
try {
//获取检查点信息
//todo 模型工厂的功能有问题，暂时写死
jobModelName = "Qwen2.5-0.5B-Instruct-147";
// jobModelName = "Qwen2.5-0.5B-Instruct-147";
String checkFileList = trainHttpService.getCheckFileList(hostUrl,jobModelName);
List<String> checkpoints = new ArrayList<>();
List<String> fileUrls = new ArrayList<>();