From 8130bed63a505cf718481c0d95d3fcf7b037db96 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 15:11:45 +0800
Subject: [PATCH] refactor(ai): 优化代码结构和更新时间处理
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java | 19 ++++++++++++-------
1 files changed, 12 insertions(+), 7 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
index 4885a94..61c9102 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
@@ -71,6 +71,7 @@
if(ObjectUtil.isNull(aiProductRoleLink)){
throw new FebsException("产品AI陪练不存在");
}
+ String companyId = aiProductRoleLink.getCompanyId();
Date nowTime = new Date();
LambdaQueryWrapper<AiMemberTalk> query = Wrappers.lambdaQuery(AiMemberTalk.class);
@@ -79,7 +80,7 @@
query.last("limit 1");
AiMemberTalk aiMemberTalk = this.getByQuery(query);
if (ObjectUtil.isNull(aiMemberTalk)){
- aiMemberTalk = this.add(memberUuid,productId,nowTime);
+ aiMemberTalk = this.add(memberUuid,companyId,productId,nowTime);
}
List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.QUESTION.getCode());
@@ -91,7 +92,7 @@
apiMemberTalkVo.setType(1);
apiMemberTalkVo.setContext(aiResponse.getResContext());
if(aiResponse.getCode().equals("200")){
- AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, aiResponse.getResContext(), nowTime);
+ AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, aiResponse.getResContext(), nowTime);
apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
}else{
@@ -137,6 +138,7 @@
callback.accept(aiResponse);
return;
}
+ String companyId = aiProductRoleLink.getCompanyId();
// 构造AI请求
AiRequest aiRequest = new AiRequest();
@@ -153,14 +155,14 @@
query.last("limit 1");
AiMemberTalk aiMemberTalk = this.getByQuery(query);
if (ObjectUtil.isNull(aiMemberTalk)) {
- aiMemberTalk = this.add(memberUuid, productId, nowTime);
+ aiMemberTalk = this.add(memberUuid,companyId, productId, nowTime);
}
try {
if (aiResponse.getCode().equals("200")) {
// 如果是最终结果（包含报告）
if (aiResponse.getReport() != null) {
// 保存完整响应到数据库
- aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, aiResponse.getResContext(), nowTime);
+ aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, aiResponse.getResContext(), nowTime);
this.updateTimeUpdate(nowTime, aiMemberTalk.getId());
callback.accept(aiResponse);
@@ -216,6 +218,8 @@
throw new FebsException("产品AI陪练对话不存在");
}
+ String companyId = aiMemberTalk.getCompanyId();
+
LambdaQueryWrapper<AiProductRoleLink> productLinkQuery = Wrappers.lambdaQuery(AiProductRoleLink.class);
productLinkQuery.eq(AiProductRoleLink::getProductId,aiMemberTalk.getProductId());
productLinkQuery.last("limit 1");
@@ -233,7 +237,7 @@
memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
memberTalkItemQuery.last("limit 1");
AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getByQuery(memberTalkItemQuery);
- aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,reqContext,new Date());
+ aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),companyId,2,reqContext,new Date());
String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext);
log.info("format:{}",format);
@@ -252,7 +256,7 @@
if ("null".equals( context)){
context = aiResponse.getResContext();
}
- AiMemberTalkItem talkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 3, context, nowTime);
+ AiMemberTalkItem talkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 3, context, nowTime);
this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
@@ -264,9 +268,10 @@
return new FebsResponse().success().data(apiMemberTalkVo);
}
@Override
- public AiMemberTalk add(String memberUuid, String productId, Date nowTime) {
+ public AiMemberTalk add(String memberUuid,String companyId, String productId, Date nowTime) {
AiMemberTalk aiMemberTalk = new AiMemberTalk();
aiMemberTalk.setId(UUID.getSimpleUUIDString());
+ aiMemberTalk.setCompanyId(companyId);
aiMemberTalk.setCreatedTime(nowTime);
aiMemberTalk.setMemberId(memberUuid);
aiMemberTalk.setProductId(productId);
--
Gitblit v1.9.1