From 76ccaeb9c5831ebcf53ae71d57086b72c1967a25 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Fri, 7 Feb 2025 11:21:42 +0800
Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E8=B0=83=E6=95=B4=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E8=AF=84=E4=BC=B0=E4=BB=BB=E5=8A=A1=E6=9F=A5=E8=AF=A2?=
 =?UTF-8?q?=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了 ModelAssessTaskAutoServiceImpl 和 ModelAssessTaskManualServiceImpl 中对状态的限制查询
- 在 ModelServiceServiceImpl 中添加了 TODO 注释，计划后续增加对所有进行中的任务进行校验
---
 .../modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java    | 3 ++-
 .../ModelAssessTaskManualServiceImpl.java                      | 2 +-
 .../llm/service/modelservice/ModelServiceServiceImpl.java      | 1 +
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java
index 5919a1f30..3eb53356c 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java
@@ -208,6 +208,7 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
 
         PageResult<ModelAssessTaskAutoDO> modelAssessTaskAutoDOPageResult = modelAssessTaskAutoMapper.selectPage(pageReqVO);
         PageResult<ModelAssessTaskAutoRespVO> result = BeanUtils.toBean(modelAssessTaskAutoDOPageResult, ModelAssessTaskAutoRespVO.class);
+
         if (CollectionUtils.isNotEmpty(result.getList())) {
             // 我的模型获取校验
             List<Long> serviceIds = modelAssessTaskAutoDOPageResult.getList().stream().
@@ -313,7 +314,7 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
 
         LambdaQueryWrapper<ModelAssessTaskAutoDO> queryWrapper = new LambdaQueryWrapper<>();
         queryWrapper = condition.apply(queryWrapper, param);
-        queryWrapper.in(ModelAssessTaskAutoDO::getStatus, 2, 3);
+//        queryWrapper.in(ModelAssessTaskAutoDO::getStatus, 2, 3);
 
         List<ModelAssessTaskAutoDO> modelAssessTaskAutoDOS = modelAssessTaskAutoMapper.selectList(queryWrapper);
         if (CollectionUtils.isEmpty(modelAssessTaskAutoDOS)) {
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 3d010d855..9f354ab09 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
@@ -290,7 +290,7 @@ public class ModelAssessTaskManualServiceImpl implements ModelAssessTaskManualSe
 
         LambdaQueryWrapper<ModelAssessTaskManualDO> queryWrapper = new LambdaQueryWrapper<>();
         queryWrapper = condition.apply(queryWrapper, param);
-        queryWrapper.in(ModelAssessTaskManualDO::getStatus, 2, 3);
+//        queryWrapper.in(ModelAssessTaskManualDO::getStatus, 2, 3);
 
         List<ModelAssessTaskManualDO> modelAssessTaskManualDOS = modelAssessTaskManualMapper.selectList(queryWrapper);
         if (CollectionUtils.isEmpty(modelAssessTaskManualDOS)) {
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
index a229c0a7f..632e0f6af 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
@@ -155,6 +155,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
 
         // 校验 基座模型 是否在 应用中心 有使用
         validateModelUsesInApplicationCenter(id, modelName);
+        // TODO: 2.7 因为使用完成再删除，页面会显现为空，所以暂时全部校验
         // 校验 基座模型 是否在 模型评估任务-人工 有使用 （是否为进行中）
         validateModelUsesInModelAssessTaskManual(id, modelName);
         // 校验 基座模型 是否在 模型评估任务-自动 有使用 （是否为进行中、推理中）