From 001da1169ce18233a46bf22abd352cf8fa979f2c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Fri, 23 Jan 2026 10:22:49 +0800
Subject: [PATCH] refactor(ai): 优化会员学习记录查询逻辑
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java | 65 ++++++++++++++------------------
1 files changed, 29 insertions(+), 36 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
index 40bba03..e0b3d3a 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkStreamServiceImpl.java
@@ -74,7 +74,7 @@
}
AiProduct aiProduct = aiProductService.getById(productId);
- Integer questionCount = aiProduct.getQuestionCount();
+ Integer talkCnt = aiProduct.getTalkCnt();
String companyId = aiProductRoleLink.getCompanyId();
@@ -85,19 +85,32 @@
query.eq(AiMemberTalk::getState,AiTypeEnum.AI_MEMBER_TALK_STATE_ING.getCode());
query.orderByDesc(AiMemberTalk::getCreatedTime);
query.last("limit 1");
- AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
- if (ObjectUtil.isNull(aiMemberTalk)){
- aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime,questionCount);
- }
ApiMemberTalkStreamVo apiMemberTalkVo = new ApiMemberTalkStreamVo();
+
+ AiMemberTalk aiMemberTalk = aiMemberTalkService.getByQuery(query);
String title = "加载时间过程，请重试！";
- AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
- if (ObjectUtil.isNotNull(question)){
- title = question.getTitle();
- AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
- apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
- aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+ if (ObjectUtil.isNull(aiMemberTalk)){
+ aiMemberTalk = aiMemberTalkService.add(memberUuid,companyId,productId,nowTime,talkCnt);
+ AiProductQuestion question = aiProductQuestionService.createQuestion(productId);
+ if (ObjectUtil.isNotNull(question)){
+ title = question.getTitle();
+ AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(),companyId, 1, title, nowTime);
+ apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
+ aiMemberTalkService.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+ }
+ }else{
+ AiMemberTalkItem oneByQuery = aiMemberTalkItemService.getOneByQuery(
+ Wrappers.lambdaQuery(AiMemberTalkItem.class)
+ .eq(AiMemberTalkItem::getMemberTalkId, aiMemberTalk.getId())
+ .eq(AiMemberTalkItem::getType, 1)
+ .orderByDesc(AiMemberTalkItem::getCreatedTime)
+ .last("limit 1")
+ );
+ if (ObjectUtil.isNotNull(oneByQuery)){
+ title = oneByQuery.getContext();
+ apiMemberTalkVo.setMemberTalkItemId(oneByQuery.getId());
+ }
}
apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
apiMemberTalkVo.setType(1);
@@ -111,8 +124,8 @@
public Flux<FebsResponse> startV2(ApiMemberTalkStreamV2Dto dto) {
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
- String memberTalkId = dto.getId();
- AiMemberTalk aiMemberTalk = this.getById(memberTalkId);
+ String memberTalkId = dto.getMemberTalkId();
+ AiMemberTalk aiMemberTalk = aiMemberTalkMapper.selectById(memberTalkId);
if (ObjectUtil.isNull(aiMemberTalk)){
throw new FebsException("对话不存在");
}
@@ -347,7 +360,7 @@
AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getOneByQuery(memberTalkItemQuery);
String question = aiMemberTalkItem.getContext();
- String promptHead = aiProductRole.getPromptHead();
+ String promptHead = aiProductRole.getModelId();
String answer = dto.getReqContext();
List<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();
@@ -446,28 +459,8 @@
this.updateMemberTalkUpdateTime(aiMemberTalk.getState(),aiMemberTalk.getDoneCnt(),aiMemberTalk.getId(), new Date());
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,companyId, 3, JSONUtil.toJsonStr(stringStringHashMap), new Date());
- memberTalkItemId = add.getId();
- }else{
- memberTalkItemId = dto.getMemberTalkItemId();
- AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getById(memberTalkItemId);
- String context = aiMemberTalkItem.getContext();
- HashMap<String, String> stringStringHashMap = JSONUtil.toBean(context, HashMap.class);
- stringStringHashMap.put(contentByCode,content);
- aiMemberTalkItemService.update(null,
- Wrappers.lambdaUpdate(AiMemberTalkItem.class)
- .set(AiMemberTalkItem::getContext,JSONUtil.toJsonStr(stringStringHashMap))
- .set(AiMemberTalkItem::getUpdatedTime,new Date())
- .set(AiMemberTalkItem::getRevision,aiMemberTalkItem.getRevision()+1)
- .eq(AiMemberTalkItem::getId,memberTalkItemId)
- );
- }
+ AiMemberTalkItem add = aiMemberTalkItemService.add(memberUuid, memberTalkId,companyId, 1, content, new Date());
+ String memberTalkItemId = add.getId();
HashMap<String, String> stringStringHashMap = new HashMap<>();
stringStringHashMap.put("memberTalkItemId",memberTalkItemId);
--
Gitblit v1.9.1