From 5ea1abc297842faf1500aa159cbfbc1a1d16b80f Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Fri, 10 Oct 2025 10:36:15 +0800
Subject: [PATCH] feat(ai): 添加公司知识库和分类字段

---
 src/main/java/cc/mrbird/febs/mall/vo/MallMemberVo.java                       |    3 +++
 src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java                        |    8 ++++++++
 src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java |    1 +
 3 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java b/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
index 78aee0e..c3ad4d8 100644
--- a/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
+++ b/src/main/java/cc/mrbird/febs/ai/entity/AiCompany.java
@@ -17,4 +17,12 @@
      * 公司名称
      */
     private String code;
+    /**
+     * 知识库ID
+     */
+    private String knowledgeId;
+    /**
+     * 文档分类ID
+     */
+    private String categoryId;
 }
diff --git a/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
index 9f40d2d..53e8b07 100644
--- a/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/mall/service/impl/ApiMallMemberServiceImpl.java
@@ -289,6 +289,7 @@
         if (StrUtil.isNotEmpty(mallMember.getCompanyId())){
             AiCompany aiCompany = aiCompanyMapper.selectById(mallMember.getCompanyId());
             mallMemberVo.setCompanyName(aiCompany.getName());
+            mallMemberVo.setCompanyCode(aiCompany.getCode());
         }
 
         HappyMemberLevel happyMemberLevel = happyMemberLevelMapper.selectOne(
diff --git a/src/main/java/cc/mrbird/febs/mall/vo/MallMemberVo.java b/src/main/java/cc/mrbird/febs/mall/vo/MallMemberVo.java
index 83f7c34..c2c59f7 100644
--- a/src/main/java/cc/mrbird/febs/mall/vo/MallMemberVo.java
+++ b/src/main/java/cc/mrbird/febs/mall/vo/MallMemberVo.java
@@ -31,6 +31,9 @@
     @ApiModelProperty(value = "公司")
     private String companyName;
 
+    @ApiModelProperty(value = "公司编码")
+    private String companyCode;
+
     @ApiModelProperty(value = "手机号")
     private String phone;
 

--
Gitblit v1.9.1