From cda3401cd52a425fbff5ae42a9c796e1a3f707d5 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Tue, 21 Jan 2025 14:42:20 +0800
Subject: [PATCH] =?UTF-8?q?[update]=20=E5=88=A0=E9=99=A4=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E9=9B=86=E6=A0=A1=E9=AA=8C=E4=BA=BA=E5=B7=A5=E8=AF=84=E4=BC=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ModelAssessTaskManualServiceImpl.java            | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java
index 87a405e50..ca4a5d2c6 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java
@@ -321,6 +321,18 @@ public class ModelAssessTaskManualServiceImpl implements ModelAssessTaskManualSe
                 ModelAssessTaskManualDO::getId, ModelAssessTaskManualDO::getTaskName, modelServiceId);
     }
 
+    /**
+     * 根据 数据集id 获取 人工评估
+     *
+     * @param datasetId 数据集id
+     * @return 人工评估
+     */
+    @Override
+    public Map<Long, String> getModelAssessmentTaskByDatasetId (Long datasetId) {
+        return getModelAssessmentTaskMap((wrapper, id) -> wrapper.eq(ModelAssessTaskManualDO::getDataset, datasetId),
+                ModelAssessTaskManualDO::getId, ModelAssessTaskManualDO::getTaskName, datasetId);
+    }
+
     @Override
     public List<ManualEvaluationReportRespVO> gitViewTheReportDetails (Long id) {
         List<ManualEvaluationReportRespVO> manualEvaluationReportRespVOS = new ArrayList<>();