lizhibing 3 месяцев назад
Родитель
Сommit
51e6b6d1fa

+ 6 - 0
salesHelper-yuyan/src/main/java/com/yuyan/salesHelper/domain/setting/AiModelParam.java

@@ -96,4 +96,10 @@ public class AiModelParam {
     @JsonIgnore
     private Date updateTime;
 
+
+    /**
+     * 是否启用
+     */
+    private Boolean enabled;
+
 }

+ 2 - 0
salesHelper-yuyan/src/main/java/com/yuyan/salesHelper/setting/service/IAiModelParamService.java

@@ -31,4 +31,6 @@ public interface IAiModelParamService extends IService<AiModelParam> {
      */
     boolean updateParam(AiModelParam aiModelParam, SysUser sysUser);
 
+
+    
 }

+ 6 - 2
salesHelper-yuyan/src/main/java/com/yuyan/salesHelper/setting/service/impl/AiModelParamServiceImpl.java

@@ -1,5 +1,6 @@
 package com.yuyan.salesHelper.setting.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuyan.common.core.domain.entity.SysUser;
 import com.yuyan.salesHelper.domain.setting.AiModelParam;
@@ -14,13 +15,16 @@ import org.springframework.stereotype.Service;
  * @date 2024-04-02
  */
 @Service
-public class AiModelParamServiceImpl extends ServiceImpl<AiModelParamMapper, AiModelParam> implements IAiModelParamService {
+public class AiModelParamServiceImpl extends ServiceImpl<AiModelParamMapper, AiModelParam>
+        implements IAiModelParamService {
 
     private static final Long defaultId = 1L;
 
     @Override
     public AiModelParam getParam() {
-        AiModelParam result = getById(defaultId);
+        LambdaQueryWrapper<AiModelParam> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.eq(AiModelParam::getEnabled, true);
+        AiModelParam result = getOne(queryWrapper);
         if (result == null) {
             result = new AiModelParam();
             result.setId(defaultId);