From fe70ca5db711b458714bfbe4a6eda6cf7d8a0421 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 13:58:46 +0800
Subject: [PATCH] feat(member): 添加会员公司ID并更新相关服务

---
 src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java |   12 ++++++++----
 1 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
index ff0df55..3c1e6f8 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
@@ -66,6 +66,8 @@
             throw new FebsException("产品AI陪练不存在");
         }
 
+        String companyId = aiProductRoleLink.getCompanyId();
+
         Date nowTime = new Date();
         LambdaQueryWrapper<AiMemberTalk> query = Wrappers.lambdaQuery(AiMemberTalk.class);
         query.eq(AiMemberTalk::getMemberId,memberUuid);
@@ -73,7 +75,7 @@
         query.last("limit 1");
         AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
         if (ObjectUtil.isNull(aiMemberTalk)){
-            aiMemberTalk = aiMemberTalkService.add(memberUuid,productId,nowTime);
+            aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime);
         }
 
         ApiMemberTalkStreamVo apiMemberTalkVo = new ApiMemberTalkStreamVo();
@@ -81,7 +83,7 @@
         AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
         if (ObjectUtil.isNotNull(question)){
             title = question.getTitle();
-            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, title, nowTime);
+            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
             apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
             aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
         }
@@ -128,7 +130,8 @@
         if (ObjectUtil.isNull(aiMemberTalk)){
             throw new FebsException("对话不存在");
         }
-        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,content,new Date());
+        String companyId = aiMemberTalk.getCompanyId();
+        aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),companyId,2,content,new Date());
         return new FebsResponse().success();
     }
 
@@ -274,13 +277,14 @@
         if (ObjectUtil.isNull(aiMemberTalk)){
             throw new FebsException("对话不存在");
         }
+        String companyId = aiMemberTalk.getCompanyId();
         Integer type = dto.getType();
         String memberTalkItemId;
         String contentByCode = AiTalkOutputEnum.HIGH_LIGHT.getCodeByType(type);
         if(StrUtil.isEmpty(dto.getMemberTalkItemId())){
             HashMap<String, String> stringStringHashMap = new HashMap<>();
             stringStringHashMap.put(contentByCode,content);
-            AiMemberTalkItem add = aiMemberTalkItemService.add(memberUuid, memberTalkId, 3, JSONUtil.toJsonStr(stringStringHashMap), new Date());
+            AiMemberTalkItem add = aiMemberTalkItemService.add(memberUuid, memberTalkId,companyId, 3, JSONUtil.toJsonStr(stringStringHashMap), new Date());
             memberTalkItemId = add.getId();
         }else{
             memberTalkItemId = dto.getMemberTalkItemId();

--
Gitblit v1.9.1