From 1506f16162c71e8852e62ecbe2d0888548a37800 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 24 Mar 2026 10:26:41 +0800
Subject: [PATCH] feat(yinhe): 添加公司级别限制LOGO功能
---
src/main/java/cc/mrbird/febs/yinhe/service/impl/YhAiServiceImpl.java | 9 +++++++--
1 files changed, 7 insertions(+), 2 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 c8fed3a..0d31c71 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
@@ -86,6 +86,13 @@
yhMemberVo.setCompanyId(yhAiCompany.getId());
yhMemberVo.setCompanyName(yhAiCompany.getName());
yhMemberVo.setPhone(yhAiMember.getPhone());
+
+ String levelCode = yhAiCompany.getLevelCode();
+ YHSysCompanyLevel yhSysCompanyLevel = yhSysCompanyLevelMapper.selectOne(
+ Wrappers.lambdaQuery(YHSysCompanyLevel.class)
+ .eq(YHSysCompanyLevel::getCode, levelCode)
+ );
+ yhMemberVo.setLimitLogo(yhSysCompanyLevel.getLimitLogo());
}
return new FebsResponse().success().data(yhMemberVo);
}
@@ -162,7 +169,6 @@
LambdaQueryWrapper<YHAiAgent> query = Wrappers.lambdaQuery(YHAiAgent.class);
boolean defaultCompany = isDefaultCompany(companyId);
- log.info("开始查询智能体列表{}",defaultCompany);
if (!defaultCompany){
List<YHAiAgentLink> yhAiAgentLinks = yhAiAgentLinkMapper.selectList(
@@ -213,7 +219,6 @@
boolean defaultCompany = isDefaultCompany(companyId);
- log.info("开始查询智能体{}",defaultCompany);
if (!defaultCompany){
List<YHAiAgentLink> yhAiAgentLinks = yhAiAgentLinkMapper.selectList(
Wrappers.lambdaQuery(YHAiAgentLink.class)
--
Gitblit v1.9.1