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 21:27:50 +08:00
parent f558be937e
commit 93e79f4f48
2 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -135,11 +135,12 @@ public class ConversationServiceImpl implements ConversationService {
chatReqVO.setSystemPrompt(application.getPrompt());
}
}
if (Objects.equals(1, chatReqVO.getModelType())){
/* if (Objects.equals(1, chatReqVO.getModelType())){
return publicModelChat(chatReqVO);
}else {
return privateModelChat(chatReqVO);
}
}*/
return publicModelChat(chatReqVO);
}
@Override
@ -167,6 +168,15 @@ public class ConversationServiceImpl implements ConversationService {
throw exception(BASE_MODEL_NOT_EXISTS);
}
model = baseModelDO.getAigcModelName();
}else if (Objects.equals(0, chatReqVO.getModelType())) {
// 自定义模型
ModelServiceDO modelServiceDO = modelServiceMapper.selectById(chatReqVO.getModelId());
if (modelServiceDO == null) {
throw exception(MODEL_SERVICE_NOT_EXISTS);
}
model = modelServiceDO.getBaseModelName();
}else {
throw exception(BASE_MODEL_NOT_EXISTS);
}
List<ModelCompletionsReqVO.ModelCompletionsMessage> messages = new ArrayList<>();

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

@ -36,7 +36,9 @@ public class FineTuningTaskSyncService {
if(Objects.equals(fineTuningTaskDO.getStatus(), FinetuningTaskStatusEnum.TRAINING.getStatus())
|| Objects.equals(fineTuningTaskDO.getStatus(), FinetuningTaskStatusEnum.WAITING.getStatus())){
AigcFineTuningDetailRespVO resp = trainHttpService.finetuningDetail(new HashMap<>(), fineTuningTaskDO.getJobId());
if (resp == null){
continue;
}
FineTuningTaskDO updateObj = new FineTuningTaskDO();
Integer status = FineTuningTaskStatusConstants.getStatus(resp.getTrainStatus());
if(status != null){