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());