From 6c340588edca370436c38f0b146e0edadcd32ff9 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 18 Mar 2026 16:22:15 +0800
Subject: [PATCH] fix(yinhe): 修复AI服务响应状态和排序问题

---
 src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java |   31 ++++++++++++++++++++-----------
 1 files changed, 20 insertions(+), 11 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java b/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
index abb9549..8e21916 100644
--- a/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java
@@ -90,7 +90,10 @@
 
     @Override
     public FebsResponse categoryList(YHCategoryDto dto) {
-        String companyId = YHLoginUserUtil.getLoginUser().getCompanyId();
+        String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
+        YhAiMember yhAiMember = yhAiMemberMapper.selectOne(Wrappers.lambdaQuery(YhAiMember.class).eq(YhAiMember::getMemberUuid, memberUuid));
+        String companyId = yhAiMember.getCompanyId();
+
         List<YHCategoryVo> yhCategoryVos = new ArrayList<>();
 
         LambdaQueryWrapper<YHAiAgentCategory> query = Wrappers.lambdaQuery(YHAiAgentCategory.class);
@@ -116,7 +119,9 @@
 
     @Override
     public FebsResponse agentList(YHAgentDto dto) {
-        String companyId = YHLoginUserUtil.getLoginUser().getCompanyId();
+        String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
+        YhAiMember yhAiMember = yhAiMemberMapper.selectOne(Wrappers.lambdaQuery(YhAiMember.class).eq(YhAiMember::getMemberUuid, memberUuid));
+        String companyId = yhAiMember.getCompanyId();
 
         ArrayList<YHAgentVo> yhAgentVos = new ArrayList<>();
         LambdaQueryWrapper<YHAiAgent> query = Wrappers.lambdaQuery(YHAiAgent.class);
@@ -440,7 +445,7 @@
                 vos.add(vo);
             }
         }
-        return new FebsResponse().data(vos);
+        return new FebsResponse().success().data(vos);
     }
 
     @Override
@@ -459,19 +464,21 @@
             yhLogoVo.setDesription(yhAiLogo.getDesription());
         }
 
-        return new FebsResponse().data(yhLogoVo);
+        return new FebsResponse().success().data(yhLogoVo);
     }
 
     @Override
     public FebsResponse loginBannerList() {
-        String companyId = YHLoginUserUtil.getLoginUser().getCompanyId();
+        String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
+        YhAiMember yhAiMember = yhAiMemberMapper.selectOne(Wrappers.lambdaQuery(YhAiMember.class).eq(YhAiMember::getMemberUuid, memberUuid));
+        String companyId = yhAiMember.getCompanyId();
+
         List<YHBannerVo> vos = new ArrayList<>();
         List<YHAiBanner> list = yhAiBannerMapper.selectList(
                 Wrappers.lambdaQuery(YHAiBanner.class)
                         .eq(YHAiBanner::getState, 1)
                         .eq(YHAiBanner::getCompanyId, companyId)
-                        .isNull(YHAiBanner::getCompanyId)
-                        .orderByDesc(YHAiBanner::getSort)
+                        .orderByAsc(YHAiBanner::getSort)
         );
         if (CollUtil.isNotEmpty(list)){
             for (YHAiBanner aiBanner : list){
@@ -482,19 +489,21 @@
                 vos.add(vo);
             }
         }
-        return new FebsResponse().data(vos);
+        return new FebsResponse().success().data(vos);
     }
 
     @Override
     public FebsResponse loginLogoInfo() {
-        String companyId = YHLoginUserUtil.getLoginUser().getCompanyId();
+        String memberUuid = YHLoginUserUtil.getLoginUser().getMemberUuid();
+        YhAiMember yhAiMember = yhAiMemberMapper.selectOne(Wrappers.lambdaQuery(YhAiMember.class).eq(YhAiMember::getMemberUuid, memberUuid));
+        String companyId = yhAiMember.getCompanyId();
+
         YHLogoVo yhLogoVo = new YHLogoVo();
 
         List<YHAiLogo> yhAiLogos = yhAiLogoMapper.selectList(
                 Wrappers.lambdaQuery(YHAiLogo.class)
                         .eq(YHAiLogo::getState, 1)
                         .eq(YHAiLogo::getCompanyId, companyId)
-                        .isNull(YHAiLogo::getCompanyId)
         );
         if (CollUtil.isNotEmpty(yhAiLogos)){
             YHAiLogo yhAiLogo = yhAiLogos.get(0);
@@ -503,6 +512,6 @@
             yhLogoVo.setDesription(yhAiLogo.getDesription());
         }
 
-        return new FebsResponse().data(yhLogoVo);
+        return new FebsResponse().success().data(yhLogoVo);
     }
 }

--
Gitblit v1.9.1