From 906dd7b3c4c94cba4e293c54b48a3a87746fa459 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Thu, 9 Jan 2025 17:24:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=B0=83=E4=BC=98=E8=AF=A6?=
 =?UTF-8?q?=E6=83=85=20=E8=8E=B7=E5=8F=96=E8=AE=AD=E7=BB=83=E6=97=A5?=
 =?UTF-8?q?=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../admin/finetuningtask/vo/FineTuningTaskRespVO.java      | 4 ++++
 .../service/finetuningtask/FineTuningTaskServiceImpl.java  | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java
index 8c1b0a17a..00ff06d4d 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java
@@ -1,5 +1,6 @@
 package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo;
 
+import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO;
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.*;
 import java.util.*;
@@ -134,4 +135,7 @@ public class FineTuningTaskRespVO {
     @Schema(description = "数据集名称")
     private String datasetName;
 
+    @Schema(description = "模型广场信息")
+    private AigcFineTuningDetailRespVO detailRespVO;
+
 }
\ No newline at end of file
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
index eceb561e6..aaea244c3 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
@@ -13,6 +13,8 @@ import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper;
 import cn.iocoder.yudao.module.llm.dal.mysql.finetuningtask.FineTuningTaskMapper;
 import cn.iocoder.yudao.module.llm.enums.FinetuningTaskStatusEnum;
 import cn.iocoder.yudao.module.llm.service.async.AsyncFineTuningTaskService;
+import cn.iocoder.yudao.module.llm.service.http.TrainHttpService;
+import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
@@ -21,6 +23,7 @@ import org.springframework.validation.annotation.Validated;
 
 import javax.annotation.Resource;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -46,6 +49,8 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
     private DatasetMapper datasetMapper;
     @Resource
     private AsyncFineTuningTaskService asyncFineTuningTaskService;
+    @Resource
+    TrainHttpService trainHttpService;
 
     @Override
     public Long createFineTuningTask(FineTuningTaskSaveReqVO createReqVO) {
@@ -116,6 +121,8 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
             BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId);
             result.setBaseModelName(baseModelDO.getModelName());
         }
+        AigcFineTuningDetailRespVO aigcFineTuningDetailRespVO = trainHttpService.finetuningDetail(new HashMap<>(), fineTuningTaskDO.getJobId());
+        result.setDetailRespVO(aigcFineTuningDetailRespVO);
         return result;
     }
     @Override