diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java index 564c50993..a8a35e153 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java @@ -201,6 +201,9 @@ public class ConversationServiceImpl implements ConversationService { message.setContent(chatReqVO.getPrompt()); messages.add(message); + + //如果知识库id不为null,先去调用知识库 + ModelCompletionsReqVO modelCompletionsReqVO = new ModelCompletionsReqVO(); modelCompletionsReqVO.setMessages(messages); // baseModel 的 aigcModelName 为aigc中的模型名称 diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java index ee494e37c..64b652ff6 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java @@ -96,6 +96,8 @@ public class FineTuningTaskSyncService { try { //获取检查点信息 + //todo 模型工厂的功能有问题,暂时写死 + jobModelName = "Qwen2.5-0.5B-Instruct-147"; String checkFileList = trainHttpService.getCheckFileList(jobModelName); String checkpoint = ""; List fileList = JSONArray.parseArray(checkFileList,String.class);