From 54efa67bc4bbc44fcf88fbeb7922d96947b6b004 Mon Sep 17 00:00:00 2001
From: ire <931903008@qq.com>
Date: Mon, 10 Feb 2025 13:47:46 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A3=80=E6=9F=A5=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../llm/service/conversation/ConversationServiceImpl.java      | 3 +++
 .../llm/service/finetuningtask/FineTuningTaskSyncService.java  | 2 ++
 2 files changed, 5 insertions(+)

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