检查点

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中的模型名称

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<String> fileList = JSONArray.parseArray(checkFileList,String.class);