tycoding 10 mēneši atpakaļ
vecāks
revīzija
5aae6ca2ca

+ 9 - 3
langchat-biz/src/main/java/cn/tycoding/langchat/biz/service/impl/AigcModelServiceImpl.java

@@ -40,20 +40,26 @@ public class AigcModelServiceImpl extends ServiceImpl<AigcModelMapper, AigcModel
 
     @Override
     public List<AigcModel> getChatModels() {
-        return baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
+        List<AigcModel> list = baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
                 .eq(AigcModel::getType, ModelTypeEnum.CHAT.name()));
+        list.forEach(this::hide);
+        return list;
     }
 
     @Override
     public List<AigcModel> getImageModels() {
-        return baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
+        List<AigcModel> list = baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
                 .eq(AigcModel::getType, ModelTypeEnum.TEXT_IMAGE.name()));
+        list.forEach(this::hide);
+        return list;
     }
 
     @Override
     public List<AigcModel> getEmbeddingModels() {
-        return baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
+        List<AigcModel> list = baseMapper.selectList(Wrappers.<AigcModel>lambdaQuery()
                 .eq(AigcModel::getType, ModelTypeEnum.EMBEDDING.name()));
+        list.forEach(this::hide);
+        return list;
     }
 
     @Override