【优化】角色分类列表 group

cherishsince
1 parent 78bb2d1d
Showing 2 changed files with 3 additions and 2 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/AiChatRoleMapper.java
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/model/AiChatRoleServiceImpl.java
@@ -50,8 +50,9 @@ public interface AiChatRoleMapper extends BaseMapperX&lt;AiChatRoleDO&gt; {
     default List<AiChatRoleDO> selectListGroupByCategory(Integer status) {
         return selectList(new LambdaQueryWrapperX<AiChatRoleDO>()
+                .select(AiChatRoleDO::getCategory)
                 .eq(AiChatRoleDO::getStatus, status)
-                .apply("GROUP BY category"));
+                .groupBy(AiChatRoleDO::getCategory));
     }
 }
@@ -132,7 +132,7 @@ public class AiChatRoleServiceImpl implements AiChatRoleService {
     @Override
     public List<String> getChatRoleCategoryList() {
         List<AiChatRoleDO> list = chatRoleMapper.selectListGroupByCategory(CommonStatusEnum.ENABLE.getStatus());
-        return convertList(list, AiChatRoleDO::getName);
+        return convertList(list, AiChatRoleDO::getCategory);
     }
 }