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

修复模型调优

Browse Source
This commit is contained in:
limin 2025-01-24 15:40:41 +08:00
parent 56e4ba790f
commit ff7a4a7e46
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
View File

@ -194,8 +194,25 @@ public class AsyncModelAccessManualService {
List<DatasetQuestionRespVO> datasetQuestionList = datasetQuestionService.getDatasetQuestionList(task.getDataset());
try {
if (!CollectionUtils.isAnyEmpty(datasetQuestionList)){
String modelName;
if (task.getModelType()==1){
BaseModelDO baseModelDO = baseModelMapper.selectById(task.getModelService());
if (baseModelDO != null){
modelName = baseModelDO.getModelName();
} else {
modelName = "";
}
} else {
ModelServiceDO modelServiceDO = modelServiceMapper.selectById(task.getModelService());
if (modelServiceDO != null){
modelName = modelServiceDO.getBaseModelName();
} else {
modelName = "";
}
}
datasetQuestionList.forEach(item -> {
ModelCompletionsReqVO modelCompletionsReqVO = new ModelCompletionsReqVO();
modelCompletionsReqVO.setModel(modelName);
ModelCompletionsReqVO.ModelCompletionsMessage message = new ModelCompletionsReqVO.ModelCompletionsMessage();
if (item.getQuestion() != null){
message.setContent(item.getQuestion());