From ff7a4a7e461acc689391b428a02905edbc0b8112 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Fri, 24 Jan 2025 15:40:41 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B=E8=B0=83?=
 =?UTF-8?q?=E4=BC=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../async/AsyncModelAccessManualService.java    | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
index f3051f775..d67a1f0ed 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncModelAccessManualService.java
@@ -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());