From 2a119c13e3218a623ffc164709bde99f56ec42f8 Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Thu, 23 Jan 2025 18:01:59 +0800
Subject: [PATCH] =?UTF-8?q?[update]=20=E5=9F=BA=E5=BA=A7=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BA/=E4=BF=AE=E6=94=B9=20=E5=90=8D=E7=A7=B0?=
 =?UTF-8?q?=E4=B8=8D=E8=83=BD=E9=87=8D=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../basemodel/BaseModelServiceImpl.java       | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
index 71a58a432..eaa764765 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
@@ -24,6 +24,7 @@ import java.util.Map;
 
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.BASE_MODEL_NOT_EXISTS;
+import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.LABEL_NAME_EXISTS;
 
 /**
  * 基座模型 Service 实现类
@@ -55,6 +56,7 @@ public class BaseModelServiceImpl implements BaseModelService {
 
     @Override
     public Long createBaseModel (BaseModelSaveReqVO createReqVO) {
+        validateBaseModelNameExists(createReqVO);
         // 插入
         BaseModelDO baseModel = BeanUtils.toBean(createReqVO, BaseModelDO.class);
         baseModelMapper.insert(baseModel);
@@ -62,8 +64,27 @@ public class BaseModelServiceImpl implements BaseModelService {
         return baseModel.getId();
     }
 
+    /**
+     * 校验模型名称是否重复
+     *
+     * @param reqVO 模型信息
+     */
+    private void validateBaseModelNameExists (BaseModelSaveReqVO reqVO) {
+        LambdaQueryWrapper<BaseModelDO> wrapper = new LambdaQueryWrapper<BaseModelDO>()
+                .eq(BaseModelDO::getModelName, reqVO.getModelName());
+
+        if (reqVO.getId() != null) {
+            wrapper.ne(BaseModelDO::getId, reqVO.getId());
+        }
+        List<BaseModelDO> baseModel = baseModelMapper.selectList(wrapper);
+        if (CollectionUtils.isNotEmpty(baseModel)) {
+            throw exception(LABEL_NAME_EXISTS);
+        }
+    }
+
     @Override
     public void updateBaseModel (BaseModelSaveReqVO updateReqVO) {
+        validateBaseModelNameExists(updateReqVO);
         // 校验存在
         validateBaseModelExists(updateReqVO.getId());
         // 更新